From 6bed7da50a41526663933d28fbf0e22d529fcfd3 Mon Sep 17 00:00:00 2001 From: Azure SDK Bot <53356347+azure-sdk@users.noreply.github.com> Date: Thu, 14 Mar 2024 23:41:23 -0400 Subject: [PATCH] [AutoRelease] t2-compute-2024-03-14-25862(can only be merged by SDK owner) (#34770) * code and test * changelog --------- Co-authored-by: azure-sdk Co-authored-by: msyyc <70930885+msyyc@users.noreply.github.com> --- sdk/compute/azure-mgmt-compute/CHANGELOG.md | 6 +- sdk/compute/azure-mgmt-compute/_meta.json | 8 +- sdk/compute/azure-mgmt-compute/assets.json | 2 +- .../compute/_compute_management_client.py | 22 +- .../azure/mgmt/compute/_configuration.py | 5 +- .../azure/mgmt/compute/_serialization.py | 90 +- .../compute/aio/_compute_management_client.py | 22 +- .../azure/mgmt/compute/aio/_configuration.py | 5 +- .../v2015_06_15/_compute_management_client.py | 26 +- .../compute/v2015_06_15/_configuration.py | 7 +- .../mgmt/compute/v2015_06_15/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2015_06_15/aio/_configuration.py | 7 +- .../v2015_06_15/aio/operations/_operations.py | 1958 ++-- .../compute/v2015_06_15/models/_models_py3.py | 150 +- .../v2015_06_15/operations/_operations.py | 2040 ++--- .../v2016_03_30/_compute_management_client.py | 26 +- .../compute/v2016_03_30/_configuration.py | 7 +- .../mgmt/compute/v2016_03_30/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2016_03_30/aio/_configuration.py | 7 +- .../v2016_03_30/aio/operations/_operations.py | 2004 ++--- .../compute/v2016_03_30/models/_models_py3.py | 160 +- .../v2016_03_30/operations/_operations.py | 2090 ++--- .../_compute_management_client.py | 26 +- .../v2016_04_30_preview/_configuration.py | 7 +- .../v2016_04_30_preview/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../v2016_04_30_preview/aio/_configuration.py | 7 +- .../aio/operations/_operations.py | 2982 ++----- .../_compute_management_client_enums.py | 3 +- .../v2016_04_30_preview/models/_models_py3.py | 204 +- .../operations/_operations.py | 3082 ++----- .../v2017_03_30/_compute_management_client.py | 26 +- .../compute/v2017_03_30/_configuration.py | 7 +- .../mgmt/compute/v2017_03_30/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2017_03_30/aio/_configuration.py | 7 +- .../v2017_03_30/aio/operations/_operations.py | 3579 +++----- .../_compute_management_client_enums.py | 3 +- .../compute/v2017_03_30/models/_models_py3.py | 274 +- .../v2017_03_30/operations/_operations.py | 3709 +++----- .../v2017_09_01/_compute_management_client.py | 26 +- .../compute/v2017_09_01/_configuration.py | 7 +- .../mgmt/compute/v2017_09_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2017_09_01/aio/_configuration.py | 7 +- .../v2017_09_01/aio/operations/_operations.py | 28 +- .../compute/v2017_09_01/models/_models_py3.py | 2 +- .../v2017_09_01/operations/_operations.py | 28 +- .../v2017_12_01/_compute_management_client.py | 26 +- .../compute/v2017_12_01/_configuration.py | 7 +- .../mgmt/compute/v2017_12_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2017_12_01/aio/_configuration.py | 7 +- .../v2017_12_01/aio/operations/_operations.py | 3523 +++----- .../compute/v2017_12_01/models/_models_py3.py | 331 +- .../v2017_12_01/operations/_operations.py | 3667 +++----- .../v2018_04_01/_compute_management_client.py | 26 +- .../compute/v2018_04_01/_configuration.py | 7 +- .../mgmt/compute/v2018_04_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2018_04_01/aio/_configuration.py | 7 +- .../v2018_04_01/aio/operations/_operations.py | 4284 +++------ .../_compute_management_client_enums.py | 3 +- .../compute/v2018_04_01/models/_models_py3.py | 368 +- .../v2018_04_01/operations/_operations.py | 4448 +++------- .../v2018_06_01/_compute_management_client.py | 26 +- .../compute/v2018_06_01/_configuration.py | 7 +- .../mgmt/compute/v2018_06_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2018_06_01/aio/_configuration.py | 7 +- .../v2018_06_01/aio/operations/_operations.py | 4874 +++------- .../_compute_management_client_enums.py | 3 +- .../compute/v2018_06_01/models/_models_py3.py | 422 +- .../v2018_06_01/operations/_operations.py | 5052 ++++------- .../v2018_09_30/_compute_management_client.py | 26 +- .../compute/v2018_09_30/_configuration.py | 7 +- .../mgmt/compute/v2018_09_30/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2018_09_30/aio/_configuration.py | 7 +- .../v2018_09_30/aio/operations/_operations.py | 678 +- .../_compute_management_client_enums.py | 3 +- .../compute/v2018_09_30/models/_models_py3.py | 28 +- .../v2018_09_30/operations/_operations.py | 682 +- .../v2018_10_01/_compute_management_client.py | 26 +- .../compute/v2018_10_01/_configuration.py | 7 +- .../mgmt/compute/v2018_10_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2018_10_01/aio/_configuration.py | 7 +- .../v2018_10_01/aio/operations/_operations.py | 3742 +++----- .../compute/v2018_10_01/models/_models_py3.py | 368 +- .../v2018_10_01/operations/_operations.py | 3904 +++----- .../v2019_03_01/_compute_management_client.py | 26 +- .../compute/v2019_03_01/_configuration.py | 7 +- .../mgmt/compute/v2019_03_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2019_03_01/aio/_configuration.py | 7 +- .../v2019_03_01/aio/operations/_operations.py | 5554 ++++-------- .../_compute_management_client_enums.py | 15 +- .../compute/v2019_03_01/models/_models_py3.py | 462 +- .../v2019_03_01/operations/_operations.py | 5764 ++++-------- .../v2019_04_01/_compute_management_client.py | 26 +- .../compute/v2019_04_01/_configuration.py | 7 +- .../mgmt/compute/v2019_04_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2019_04_01/aio/_configuration.py | 7 +- .../v2019_04_01/aio/operations/_operations.py | 28 +- .../compute/v2019_04_01/models/_models_py3.py | 2 +- .../v2019_04_01/operations/_operations.py | 28 +- .../v2019_07_01/_compute_management_client.py | 26 +- .../compute/v2019_07_01/_configuration.py | 7 +- .../mgmt/compute/v2019_07_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2019_07_01/aio/_configuration.py | 7 +- .../v2019_07_01/aio/operations/_operations.py | 6440 ++++---------- .../_compute_management_client_enums.py | 15 +- .../compute/v2019_07_01/models/_models_py3.py | 542 +- .../v2019_07_01/operations/_operations.py | 6676 ++++---------- .../v2019_11_01/_compute_management_client.py | 26 +- .../compute/v2019_11_01/_configuration.py | 7 +- .../mgmt/compute/v2019_11_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2019_11_01/aio/_configuration.py | 7 +- .../v2019_11_01/aio/operations/_operations.py | 914 +- .../_compute_management_client_enums.py | 17 +- .../compute/v2019_11_01/models/_models_py3.py | 32 +- .../v2019_11_01/operations/_operations.py | 926 +- .../v2019_12_01/_compute_management_client.py | 26 +- .../compute/v2019_12_01/_configuration.py | 7 +- .../mgmt/compute/v2019_12_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2019_12_01/aio/_configuration.py | 7 +- .../v2019_12_01/aio/operations/_operations.py | 5794 ++++-------- .../compute/v2019_12_01/models/_models_py3.py | 537 +- .../v2019_12_01/operations/_operations.py | 6032 ++++--------- .../v2020_05_01/_compute_management_client.py | 26 +- .../compute/v2020_05_01/_configuration.py | 7 +- .../mgmt/compute/v2020_05_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2020_05_01/aio/_configuration.py | 7 +- .../v2020_05_01/aio/operations/_operations.py | 1173 +-- .../_compute_management_client_enums.py | 23 +- .../compute/v2020_05_01/models/_models_py3.py | 36 +- .../v2020_05_01/operations/_operations.py | 1191 +-- .../v2020_06_01/_compute_management_client.py | 26 +- .../compute/v2020_06_01/_configuration.py | 7 +- .../mgmt/compute/v2020_06_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2020_06_01/aio/_configuration.py | 7 +- .../v2020_06_01/aio/operations/_operations.py | 5202 ++++------- .../compute/v2020_06_01/models/_models_py3.py | 403 +- .../v2020_06_01/operations/_operations.py | 5434 ++++-------- .../v2020_06_30/_compute_management_client.py | 26 +- .../compute/v2020_06_30/_configuration.py | 7 +- .../mgmt/compute/v2020_06_30/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2020_06_30/aio/_configuration.py | 7 +- .../v2020_06_30/aio/operations/_operations.py | 1201 +-- .../_compute_management_client_enums.py | 27 +- .../compute/v2020_06_30/models/_models_py3.py | 38 +- .../v2020_06_30/operations/_operations.py | 1221 +-- .../v2020_09_30/_compute_management_client.py | 26 +- .../compute/v2020_09_30/_configuration.py | 7 +- .../mgmt/compute/v2020_09_30/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2020_09_30/aio/_configuration.py | 7 +- .../v2020_09_30/aio/operations/_operations.py | 2696 ++---- .../_compute_management_client_enums.py | 27 +- .../compute/v2020_09_30/models/_models_py3.py | 200 +- .../v2020_09_30/operations/_operations.py | 2764 ++---- .../_compute_management_client.py | 26 +- .../v2020_10_01_preview/_configuration.py | 7 +- .../v2020_10_01_preview/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../v2020_10_01_preview/aio/_configuration.py | 7 +- .../aio/operations/_operations.py | 970 +- .../v2020_10_01_preview/models/_models_py3.py | 14 +- .../operations/_operations.py | 985 +-- .../v2020_12_01/_compute_management_client.py | 26 +- .../compute/v2020_12_01/_configuration.py | 7 +- .../mgmt/compute/v2020_12_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2020_12_01/aio/_configuration.py | 7 +- .../v2020_12_01/aio/operations/_operations.py | 6893 +++++---------- .../_compute_management_client_enums.py | 33 +- .../compute/v2020_12_01/models/_models_py3.py | 447 +- .../v2020_12_01/operations/_operations.py | 7173 +++++---------- .../v2021_03_01/_compute_management_client.py | 26 +- .../compute/v2021_03_01/_configuration.py | 7 +- .../mgmt/compute/v2021_03_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_03_01/aio/_configuration.py | 7 +- .../v2021_03_01/aio/operations/_operations.py | 6711 ++++---------- .../_compute_management_client_enums.py | 1 - .../compute/v2021_03_01/models/_models_py3.py | 342 +- .../v2021_03_01/operations/_operations.py | 7007 +++++---------- .../v2021_04_01/_compute_management_client.py | 26 +- .../compute/v2021_04_01/_configuration.py | 7 +- .../mgmt/compute/v2021_04_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_04_01/aio/_configuration.py | 7 +- .../v2021_04_01/aio/operations/_operations.py | 7537 +++++----------- .../_compute_management_client_enums.py | 44 +- .../compute/v2021_04_01/models/_models_py3.py | 370 +- .../v2021_04_01/operations/_operations.py | 7851 +++++------------ .../v2021_07_01/_compute_management_client.py | 26 +- .../compute/v2021_07_01/_configuration.py | 7 +- .../mgmt/compute/v2021_07_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_07_01/aio/_configuration.py | 7 +- .../v2021_07_01/aio/operations/_operations.py | 7410 +++++----------- .../_compute_management_client_enums.py | 1 - .../compute/v2021_07_01/models/_models_py3.py | 516 +- .../v2021_07_01/operations/_operations.py | 7732 +++++----------- .../v2021_08_01/_compute_management_client.py | 26 +- .../compute/v2021_08_01/_configuration.py | 7 +- .../mgmt/compute/v2021_08_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_08_01/aio/_configuration.py | 7 +- .../v2021_08_01/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 53 +- .../compute/v2021_08_01/models/_models_py3.py | 42 +- .../v2021_08_01/operations/_operations.py | 1537 +--- .../v2021_10_01/_compute_management_client.py | 26 +- .../compute/v2021_10_01/_configuration.py | 7 +- .../mgmt/compute/v2021_10_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_10_01/aio/_configuration.py | 7 +- .../v2021_10_01/aio/operations/_operations.py | 1158 +-- .../compute/v2021_10_01/models/_models_py3.py | 135 +- .../v2021_10_01/operations/_operations.py | 1192 +-- .../v2021_11_01/_compute_management_client.py | 26 +- .../compute/v2021_11_01/_configuration.py | 7 +- .../mgmt/compute/v2021_11_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_11_01/aio/_configuration.py | 7 +- .../v2021_11_01/aio/operations/_operations.py | 6069 ++++--------- .../_compute_management_client_enums.py | 1 - .../compute/v2021_11_01/models/_models_py3.py | 336 +- .../v2021_11_01/operations/_operations.py | 6349 ++++--------- .../v2021_12_01/_compute_management_client.py | 26 +- .../compute/v2021_12_01/_configuration.py | 7 +- .../mgmt/compute/v2021_12_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2021_12_01/aio/_configuration.py | 7 +- .../v2021_12_01/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 57 +- .../compute/v2021_12_01/models/_models_py3.py | 42 +- .../v2021_12_01/operations/_operations.py | 1537 +--- .../v2022_01_03/_compute_management_client.py | 26 +- .../compute/v2022_01_03/_configuration.py | 7 +- .../mgmt/compute/v2022_01_03/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_01_03/aio/_configuration.py | 7 +- .../v2022_01_03/aio/operations/_operations.py | 1406 +-- .../compute/v2022_01_03/models/_models_py3.py | 204 +- .../v2022_01_03/operations/_operations.py | 1452 +-- .../v2022_03_01/_compute_management_client.py | 26 +- .../compute/v2022_03_01/_configuration.py | 7 +- .../mgmt/compute/v2022_03_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_03_01/aio/_configuration.py | 7 +- .../v2022_03_01/aio/operations/_operations.py | 6087 ++++--------- .../_compute_management_client_enums.py | 1 - .../compute/v2022_03_01/models/_models_py3.py | 338 +- .../v2022_03_01/operations/_operations.py | 6371 ++++--------- .../v2022_03_02/_compute_management_client.py | 26 +- .../compute/v2022_03_02/_configuration.py | 7 +- .../mgmt/compute/v2022_03_02/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_03_02/aio/_configuration.py | 7 +- .../v2022_03_02/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 61 +- .../compute/v2022_03_02/models/_models_py3.py | 46 +- .../v2022_03_02/operations/_operations.py | 1537 +--- .../v2022_03_03/_compute_management_client.py | 26 +- .../compute/v2022_03_03/_configuration.py | 7 +- .../mgmt/compute/v2022_03_03/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_03_03/aio/_configuration.py | 7 +- .../v2022_03_03/aio/operations/_operations.py | 1406 +-- .../compute/v2022_03_03/models/_models_py3.py | 220 +- .../v2022_03_03/operations/_operations.py | 1452 +-- .../v2022_04_04/_compute_management_client.py | 26 +- .../compute/v2022_04_04/_configuration.py | 7 +- .../mgmt/compute/v2022_04_04/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_04_04/aio/_configuration.py | 7 +- .../v2022_04_04/aio/operations/_operations.py | 1041 +-- .../compute/v2022_04_04/models/_models_py3.py | 26 +- .../v2022_04_04/operations/_operations.py | 1077 +-- .../v2022_07_02/_compute_management_client.py | 26 +- .../compute/v2022_07_02/_configuration.py | 7 +- .../mgmt/compute/v2022_07_02/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_07_02/aio/_configuration.py | 7 +- .../v2022_07_02/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 61 +- .../compute/v2022_07_02/models/_models_py3.py | 46 +- .../v2022_07_02/operations/_operations.py | 1537 +--- .../v2022_08_01/_compute_management_client.py | 26 +- .../compute/v2022_08_01/_configuration.py | 7 +- .../mgmt/compute/v2022_08_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_08_01/aio/_configuration.py | 7 +- .../v2022_08_01/aio/operations/_operations.py | 6087 ++++--------- .../_compute_management_client_enums.py | 1 - .../compute/v2022_08_01/models/_models_py3.py | 344 +- .../v2022_08_01/operations/_operations.py | 6371 ++++--------- .../v2022_08_03/_compute_management_client.py | 26 +- .../compute/v2022_08_03/_configuration.py | 7 +- .../mgmt/compute/v2022_08_03/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_08_03/aio/_configuration.py | 7 +- .../v2022_08_03/aio/operations/_operations.py | 1406 +-- .../compute/v2022_08_03/models/_models_py3.py | 184 +- .../v2022_08_03/operations/_operations.py | 1452 +-- .../v2022_09_04/_compute_management_client.py | 26 +- .../compute/v2022_09_04/_configuration.py | 7 +- .../mgmt/compute/v2022_09_04/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_09_04/aio/_configuration.py | 7 +- .../v2022_09_04/aio/operations/_operations.py | 1041 +-- .../compute/v2022_09_04/models/_models_py3.py | 26 +- .../v2022_09_04/operations/_operations.py | 1077 +-- .../v2022_11_01/_compute_management_client.py | 26 +- .../compute/v2022_11_01/_configuration.py | 7 +- .../mgmt/compute/v2022_11_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2022_11_01/aio/_configuration.py | 7 +- .../v2022_11_01/aio/operations/_operations.py | 6087 ++++--------- .../_compute_management_client_enums.py | 1 - .../compute/v2022_11_01/models/_models_py3.py | 365 +- .../v2022_11_01/operations/_operations.py | 6371 ++++--------- .../v2023_01_02/_compute_management_client.py | 26 +- .../compute/v2023_01_02/_configuration.py | 7 +- .../mgmt/compute/v2023_01_02/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_01_02/aio/_configuration.py | 7 +- .../v2023_01_02/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 61 +- .../compute/v2023_01_02/models/_models_py3.py | 46 +- .../v2023_01_02/operations/_operations.py | 1537 +--- .../v2023_03_01/_compute_management_client.py | 26 +- .../compute/v2023_03_01/_configuration.py | 7 +- .../mgmt/compute/v2023_03_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_03_01/aio/_configuration.py | 7 +- .../v2023_03_01/aio/operations/_operations.py | 6148 ++++--------- .../_compute_management_client_enums.py | 5 +- .../compute/v2023_03_01/models/_models_py3.py | 367 +- .../v2023_03_01/operations/_operations.py | 6434 ++++---------- .../v2023_04_02/_compute_management_client.py | 26 +- .../compute/v2023_04_02/_configuration.py | 7 +- .../mgmt/compute/v2023_04_02/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_04_02/aio/_configuration.py | 7 +- .../v2023_04_02/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 61 +- .../compute/v2023_04_02/models/_models_py3.py | 46 +- .../v2023_04_02/operations/_operations.py | 1537 +--- .../v2023_07_01/_compute_management_client.py | 26 +- .../compute/v2023_07_01/_configuration.py | 7 +- .../mgmt/compute/v2023_07_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_07_01/aio/_configuration.py | 7 +- .../v2023_07_01/aio/operations/_operations.py | 6148 ++++--------- .../_compute_management_client_enums.py | 5 +- .../compute/v2023_07_01/models/_models_py3.py | 369 +- .../v2023_07_01/operations/_operations.py | 6434 ++++---------- .../v2023_07_03/_compute_management_client.py | 26 +- .../compute/v2023_07_03/_configuration.py | 7 +- .../mgmt/compute/v2023_07_03/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_07_03/aio/_configuration.py | 7 +- .../v2023_07_03/aio/operations/_operations.py | 1406 +-- .../compute/v2023_07_03/models/_models_py3.py | 184 +- .../v2023_07_03/operations/_operations.py | 1452 +-- .../v2023_09_01/_compute_management_client.py | 26 +- .../compute/v2023_09_01/_configuration.py | 7 +- .../mgmt/compute/v2023_09_01/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_09_01/aio/_configuration.py | 7 +- .../v2023_09_01/aio/operations/_operations.py | 6417 ++++---------- .../_compute_management_client_enums.py | 5 +- .../compute/v2023_09_01/models/_models_py3.py | 373 +- .../v2023_09_01/operations/_operations.py | 6711 +++++--------- .../v2023_10_02/_compute_management_client.py | 26 +- .../compute/v2023_10_02/_configuration.py | 7 +- .../mgmt/compute/v2023_10_02/_metadata.json | 10 +- .../aio/_compute_management_client.py | 28 +- .../compute/v2023_10_02/aio/_configuration.py | 7 +- .../v2023_10_02/aio/operations/_operations.py | 1503 +--- .../_compute_management_client_enums.py | 63 +- .../compute/v2023_10_02/models/_models_py3.py | 46 +- .../v2023_10_02/operations/_operations.py | 1537 +--- .../availability_set_create.py | 3 + ...availability_set_delete_maximum_set_gen.py | 1 + ...availability_set_delete_minimum_set_gen.py | 1 + .../availability_set_get_maximum_set_gen.py | 1 + .../availability_set_get_minimum_set_gen.py | 1 + ...et_list_available_sizes_maximum_set_gen.py | 1 + ...et_list_available_sizes_minimum_set_gen.py | 1 + .../availability_set_list_by_subscription.py | 1 + .../availability_set_list_maximum_set_gen.py | 1 + .../availability_set_list_minimum_set_gen.py | 1 + ...availability_set_update_maximum_set_gen.py | 3 + .../capacity_reservation_create_or_update.py | 3 + ...city_reservation_delete_maximum_set_gen.py | 1 + ...city_reservation_delete_minimum_set_gen.py | 1 + .../capacity_reservation_get.py | 1 + ...city_reservation_group_create_or_update.py | 3 + ...eservation_group_delete_maximum_set_gen.py | 1 + ...eservation_group_delete_minimum_set_gen.py | 1 + .../capacity_reservation_group_get.py | 1 + ...eservation_group_list_by_resource_group.py | 1 + ..._reservation_group_list_by_subscription.py | 1 + ...eservation_group_update_maximum_set_gen.py | 3 + ...y_reservation_list_by_reservation_group.py | 1 + ...city_reservation_update_maximum_set_gen.py | 3 + .../operations_list_maximum_set_gen.py | 1 + .../operations_list_minimum_set_gen.py | 1 + .../usage_list_maximum_set_gen.py | 1 + .../usage_list_minimum_set_gen.py | 1 + ...tual_machine_sizes_list_maximum_set_gen.py | 1 + ...tual_machine_sizes_list_minimum_set_gen.py | 1 + .../dedicated_host_create_or_update.py | 3 + .../dedicated_host_delete_maximum_set_gen.py | 1 + .../dedicated_host_delete_minimum_set_gen.py | 1 + .../dedicated_host_get.py | 1 + .../dedicated_host_group_create_or_update.py | 3 + ...t_group_create_or_update_with_ultra_ssd.py | 3 + ...cated_host_group_delete_maximum_set_gen.py | 1 + ...cated_host_group_delete_minimum_set_gen.py | 1 + .../dedicated_host_group_get.py | 1 + ..._ultra_ssd_enabled_dedicated_host_group.py | 1 + ..._list_by_resource_group_maximum_set_gen.py | 1 + ..._list_by_resource_group_minimum_set_gen.py | 1 + ...up_list_by_subscription_maximum_set_gen.py | 1 + ...up_list_by_subscription_minimum_set_gen.py | 1 + ...cated_host_group_update_maximum_set_gen.py | 3 + .../dedicated_host_list_available_sizes.py | 1 + ...host_list_by_host_group_maximum_set_gen.py | 1 + ...host_list_by_host_group_minimum_set_gen.py | 1 + .../dedicated_host_redeploy.py | 1 + .../dedicated_host_restart.py | 1 + .../dedicated_host_update_maximum_set_gen.py | 3 + .../dedicated_host_update_resize.py | 3 + ...ge_create_data_disk_from_ablob_included.py | 3 + ...e_data_disk_from_amanaged_disk_included.py | 3 + ...reate_data_disk_from_asnapshot_included.py | 3 + .../image_examples/image_create_from_ablob.py | 3 + ...ate_from_ablob_with_disk_encryption_set.py | 3 + .../image_create_from_amanaged_disk.py | 3 + ..._amanaged_disk_with_disk_encryption_set.py | 3 + .../image_create_from_asnapshot.py | 3 + ...from_asnapshot_with_disk_encryption_set.py | 3 + .../image_examples/image_create_from_avm.py | 3 + .../image_examples/image_get.py | 1 + .../image_list_by_resource_group.py | 1 + .../image_list_by_subscription.py | 1 + .../image_examples/image_update.py | 3 + .../images_delete_maximum_set_gen.py | 1 + .../images_delete_minimum_set_gen.py | 1 + .../log_analytics_request_rate_by_interval.py | 3 + .../log_analytics_throttled_requests.py | 3 + ...ximity_placement_group_create_or_update.py | 3 + .../proximity_placement_group_delete.py | 1 + .../proximity_placement_group_get.py | 1 + ..._placement_group_list_by_resource_group.py | 1 + ...ty_placement_group_list_by_subscription.py | 1 + .../proximity_placement_group_patch.py | 3 + ...store_point_collection_create_or_update.py | 3 + ..._create_or_update_for_cross_region_copy.py | 3 + ...point_collection_delete_maximum_set_gen.py | 1 + ...point_collection_delete_minimum_set_gen.py | 1 + .../restore_point_collection_get.py | 1 + ...ction_get_with_contained_restore_points.py | 1 + ...point_collection_list_by_resource_group.py | 1 + ...e_point_collection_list_by_subscription.py | 1 + ...point_collection_update_maximum_set_gen.py | 3 + .../restore_point_copy_between_regions.py | 3 + .../restore_point_create.py | 3 + .../restore_point_delete_maximum_set_gen.py | 1 + .../restore_point_delete_minimum_set_gen.py | 1 + .../restore_point_get.py | 1 + .../restore_point_get_with_instance_view.py | 1 + .../run_command_examples/run_command_get.py | 1 + .../run_command_examples/run_command_list.py | 1 + .../virtual_machine_run_command.py | 3 + ...al_machine_run_command_create_or_update.py | 3 + .../virtual_machine_run_command_delete.py | 1 + .../virtual_machine_run_command_get.py | 1 + .../virtual_machine_run_command_list.py | 1 + .../virtual_machine_run_command_update.py | 3 + ...irtual_machine_scale_set_vm_run_command.py | 3 + ...ale_set_vm_run_command_create_or_update.py | 3 + ...machine_scale_set_vm_run_command_delete.py | 1 + ...al_machine_scale_set_vm_run_command_get.py | 1 + ...l_machine_scale_set_vm_run_command_list.py | 1 + ...machine_scale_set_vm_run_command_update.py | 3 + .../ssh_public_key_create.py | 3 + .../ssh_public_key_delete_maximum_set_gen.py | 1 + .../ssh_public_key_delete_minimum_set_gen.py | 1 + .../ssh_public_key_generate_key_pair.py | 1 + ...nerate_key_pair_encryption_with_ed25519.py | 1 + ...y_generate_key_pair_encryption_with_rsa.py | 1 + .../ssh_public_key_get.py | 1 + ..._list_by_resource_group_maximum_set_gen.py | 1 + ..._list_by_resource_group_minimum_set_gen.py | 1 + ...ey_list_by_subscription_maximum_set_gen.py | 1 + ...ey_list_by_subscription_minimum_set_gen.py | 1 + .../ssh_public_key_update_maximum_set_gen.py | 3 + .../virtual_machine_assess_patches.py | 1 + ...ttach_detach_data_disks_maximum_set_gen.py | 3 + ...ttach_detach_data_disks_minimum_set_gen.py | 3 + ...virtual_machine_capture_maximum_set_gen.py | 3 + ...virtual_machine_capture_minimum_set_gen.py | 3 + ...onvert_to_managed_disks_maximum_set_gen.py | 1 + ...onvert_to_managed_disks_minimum_set_gen.py | 1 + ..._from_an_unmanaged_generalized_os_image.py | 3 + ...ne_create_from_acommunity_gallery_image.py | 3 + ...rtual_machine_create_from_acustom_image.py | 3 + ...e_create_from_ageneralized_shared_image.py | 3 + ...chine_create_from_ashared_gallery_image.py | 3 + ...e_create_from_aspecialized_shared_image.py | 3 + ...l_machine_create_in_an_availability_set.py | 3 + ...customer_assigned_platform_fault_domain.py | 3 + ..._vm_with_automatic_by_platform_settings.py | 3 + ...etting_assessment_mode_of_image_default.py | 3 + ...ith_patch_setting_mode_of_image_default.py | 3 + ..._setting_modes_of_automatic_by_platform.py | 3 + ...age_vm_with_unmanaged_os_and_data_disks.py | 3 + ..._vm_with_automatic_by_platform_settings.py | 3 + ...etting_assessment_mode_of_image_default.py | 3 + ...h_patch_setting_mode_of_automatic_by_os.py | 3 + ...y_platform_and_enable_hot_patching_true.py | 3 + ...ws_vm_with_patch_setting_mode_of_manual.py | 3 + ..._setting_modes_of_automatic_by_platform.py | 3 + ...rtual_machine_create_with_adiff_os_disk.py | 3 + ...using_diff_disk_placement_as_cache_disk.py | 3 + ...ng_diff_disk_placement_as_resource_disk.py | 3 + ...ine_create_with_amarketplace_image_plan.py | 3 + ...machine_create_with_application_profile.py | 3 + ...al_machine_create_with_boot_diagnostics.py | 3 + ...achine_create_with_capacity_reservation.py | 3 + ...achine_create_with_disk_controller_type.py | 3 + ...reate_with_disk_encryption_set_resource.py | 3 + ...al_machine_create_with_empty_data_disks.py | 3 + ..._machine_create_with_encryption_at_host.py | 3 + ...machine_create_with_encryption_identity.py | 3 + ...hine_create_with_extensions_time_budget.py | 3 + ...machine_create_with_hibernation_enabled.py | 3 + ...ne_create_with_managed_boot_diagnostics.py | 3 + ...te_with_network_interface_configuration.py | 3 + ...rk_interface_configuration_dns_settings.py | 3 + ...ine_create_with_password_authentication.py | 3 + ...ual_machine_create_with_premium_storage.py | 3 + ...achine_create_with_proxy_agent_settings.py | 3 + ...ne_create_with_scheduled_events_profile.py | 3 + ...eate_with_security_type_confidential_vm.py | 3 + ...fidential_vm_with_customer_managed_keys.py | 3 + ..._confidential_vm_with_non_persisted_tpm.py | 3 + ..._machine_create_with_ssh_authentication.py | 3 + ...rtual_machine_create_with_uefi_settings.py | 3 + .../virtual_machine_create_with_user_data.py | 3 + ..._machine_create_with_vm_size_properties.py | 3 + ...tual_machine_deallocate_maximum_set_gen.py | 1 + ...tual_machine_deallocate_minimum_set_gen.py | 1 + .../virtual_machine_delete_force.py | 1 + ...ension_create_or_update_maximum_set_gen.py | 3 + ...ension_create_or_update_minimum_set_gen.py | 3 + ...achine_extension_delete_maximum_set_gen.py | 1 + ...achine_extension_delete_minimum_set_gen.py | 1 + ...l_machine_extension_get_maximum_set_gen.py | 1 + ...l_machine_extension_get_minimum_set_gen.py | 1 + ..._machine_extension_list_maximum_set_gen.py | 1 + ..._machine_extension_list_minimum_set_gen.py | 1 + .../virtual_machine_extension_update.py | 3 + .../virtual_machine_generalize.py | 1 + .../virtual_machine_get.py | 1 + ...get_auto_placed_on_dedicated_host_group.py | 1 + .../virtual_machine_get_instance_view.py | 1 + ...iew_auto_placed_on_dedicated_host_group.py | 1 + ...l_machine_get_with_disk_controller_type.py | 1 + ...ual_machine_get_with_vm_size_properties.py | 1 + .../virtual_machine_install_patches.py | 3 + ...irtual_machine_list_all_maximum_set_gen.py | 1 + ...irtual_machine_list_all_minimum_set_gen.py | 1 + ...virtual_machine_list_available_vm_sizes.py | 1 + ...achine_list_by_subscription_by_location.py | 1 + .../virtual_machine_list_maximum_set_gen.py | 1 + .../virtual_machine_list_minimum_set_gen.py | 1 + ...ine_perform_maintenance_maximum_set_gen.py | 1 + ...ine_perform_maintenance_minimum_set_gen.py | 1 + ...rtual_machine_power_off_maximum_set_gen.py | 1 + ...rtual_machine_power_off_minimum_set_gen.py | 1 + .../virtual_machine_reapply.py | 1 + ...irtual_machine_redeploy_maximum_set_gen.py | 1 + ...irtual_machine_redeploy_minimum_set_gen.py | 1 + .../virtual_machine_reimage.py | 1 + ...rtual_machine_reimage_non_ephemeral_vms.py | 1 + ...virtual_machine_restart_maximum_set_gen.py | 1 + ...virtual_machine_restart_minimum_set_gen.py | 1 + ..._machine_retrieve_boot_diagnostics_data.py | 1 + .../virtual_machine_simulate_eviction.py | 1 + .../virtual_machine_start_maximum_set_gen.py | 1 + .../virtual_machine_start_minimum_set_gen.py | 1 + ...data_disk_using_to_be_detached_property.py | 3 + ...l_machine_update_force_detach_data_disk.py | 3 + ...ine_extension_image_get_maximum_set_gen.py | 1 + ...ine_extension_image_get_minimum_set_gen.py | 1 + ...ension_image_list_types_maximum_set_gen.py | 1 + ...ension_image_list_types_minimum_set_gen.py | 1 + ...ion_image_list_versions_maximum_set_gen.py | 1 + ...ion_image_list_versions_minimum_set_gen.py | 1 + ...rtual_machine_image_get_maximum_set_gen.py | 1 + ...rtual_machine_image_get_minimum_set_gen.py | 1 + ...tual_machine_image_list_maximum_set_gen.py | 1 + ...tual_machine_image_list_minimum_set_gen.py | 1 + ...chine_image_list_offers_maximum_set_gen.py | 1 + ...chine_image_list_offers_minimum_set_gen.py | 1 + ...e_image_list_publishers_maximum_set_gen.py | 1 + ...e_image_list_publishers_minimum_set_gen.py | 1 + ...machine_image_list_skus_maximum_set_gen.py | 1 + ...machine_image_list_skus_minimum_set_gen.py | 1 + ...ne_images_edge_zone_get_maximum_set_gen.py | 1 + ...ne_images_edge_zone_get_minimum_set_gen.py | 1 + ..._zone_list_by_edge_zone_maximum_set_gen.py | 1 + ..._zone_list_by_edge_zone_minimum_set_gen.py | 1 + ...e_images_edge_zone_list_maximum_set_gen.py | 1 + ...e_images_edge_zone_list_minimum_set_gen.py | 1 + ...s_edge_zone_list_offers_maximum_set_gen.py | 1 + ...s_edge_zone_list_offers_minimum_set_gen.py | 1 + ...ge_zone_list_publishers_maximum_set_gen.py | 1 + ...ge_zone_list_publishers_minimum_set_gen.py | 1 + ...ges_edge_zone_list_skus_maximum_set_gen.py | 1 + ...ges_edge_zone_list_skus_minimum_set_gen.py | 1 + ...chine_scale_set_approve_rolling_upgrade.py | 1 + ..._single_placement_group_maximum_set_gen.py | 3 + ..._diff_os_disk_using_diff_disk_placement.py | 3 + ..._from_an_unmanaged_generalized_os_image.py | 3 + ...ine_scale_set_create_from_acustom_image.py | 3 + ...t_create_from_ageneralized_shared_image.py | 3 + ...t_create_from_aspecialized_shared_image.py | 3 + ...le_tcp_state_tracking_network_interface.py | 3 + ...create_from_with_fpga_network_interface.py | 3 + ...ith_network_interface_with_dns_settings.py | 3 + ..._platform_image_with_unmanaged_os_disks.py | 3 + ...set_create_with_amarketplace_image_plan.py | 3 + ...ale_set_create_with_application_profile.py | 3 + ...scale_set_create_with_automatic_repairs.py | 3 + ...t_create_with_azure_application_gateway.py | 3 + ...ale_set_create_with_azure_load_balancer.py | 3 + ..._scale_set_create_with_boot_diagnostics.py | 3 + ...le_set_create_with_capacity_reservation.py | 3 + ...hine_scale_set_create_with_diff_os_disk.py | 3 + ...le_set_create_with_disk_controller_type.py | 3 + ...reate_with_disk_encryption_set_resource.py | 3 + ...create_with_empty_data_disks_on_each_vm.py | 3 + ...cale_set_create_with_encryption_at_host.py | 3 + ...th_extensions_suppress_failures_enabled.py | 3 + ..._set_create_with_extensions_time_budget.py | 3 + ...et_create_with_managed_boot_diagnostics.py | 3 + ...e_with_os_image_scheduled_event_enabled.py | 3 + ...set_create_with_password_authentication.py | 3 + ...e_scale_set_create_with_premium_storage.py | 3 + ...ale_set_create_with_priority_mix_policy.py | 7 +- ..._with_protected_settings_from_key_vault.py | 3 + ...le_set_create_with_proxy_agent_settings.py | 3 + ...reate_with_resilient_vm_creation_policy.py | 3 + ...reate_with_resilient_vm_deletion_policy.py | 3 + ...e_scale_set_create_with_scale_in_policy.py | 3 + ..._create_with_security_posture_reference.py | 3 + ...eate_with_security_type_confidential_vm.py | 3 + ..._confidential_vm_with_non_persisted_tpm.py | 3 + ..._create_with_service_artifact_reference.py | 3 + ...ale_set_create_with_spot_restore_policy.py | 3 + ...cale_set_create_with_ssh_authentication.py | 3 + ..._with_terminate_scheduled_event_enabled.py | 3 + ...ine_scale_set_create_with_uefi_settings.py | 3 + ...machine_scale_set_create_with_user_data.py | 3 + ...cale_set_create_with_vm_size_properties.py | 3 + ..._set_create_with_vms_in_different_zones.py | 3 + ...ne_scale_set_deallocate_maximum_set_gen.py | 1 + ...ne_scale_set_deallocate_minimum_set_gen.py | 1 + .../virtual_machine_scale_set_delete_force.py | 1 + ...ension_create_or_update_maximum_set_gen.py | 3 + ...le_set_extension_delete_maximum_set_gen.py | 1 + ...le_set_extension_delete_minimum_set_gen.py | 1 + ...scale_set_extension_get_maximum_set_gen.py | 1 + ...scale_set_extension_get_minimum_set_gen.py | 1 + ...cale_set_extension_list_maximum_set_gen.py | 1 + ...cale_set_extension_list_minimum_set_gen.py | 1 + ...ine_scale_set_extension_rolling_upgrade.py | 1 + ...le_set_extension_update_maximum_set_gen.py | 3 + ...form_update_domain_walk_maximum_set_gen.py | 1 + ...form_update_domain_walk_minimum_set_gen.py | 1 + .../virtual_machine_scale_set_get.py | 1 + ...get_auto_placed_on_dedicated_host_group.py | 1 + ...e_set_get_instance_view_maximum_set_gen.py | 1 + ...e_set_get_instance_view_minimum_set_gen.py | 1 + ..._get_os_upgrade_history_maximum_set_gen.py | 1 + ..._get_os_upgrade_history_minimum_set_gen.py | 1 + ...scale_set_get_with_disk_controller_type.py | 1 + ...al_machine_scale_set_get_with_user_data.py | 1 + ...hine_scale_set_list_all_maximum_set_gen.py | 1 + ...hine_scale_set_list_all_minimum_set_gen.py | 1 + ...le_set_list_by_subscription_by_location.py | 1 + ..._machine_scale_set_list_maximum_set_gen.py | 1 + ..._machine_scale_set_list_minimum_set_gen.py | 1 + ...ine_scale_set_list_skus_maximum_set_gen.py | 1 + ...ine_scale_set_list_skus_minimum_set_gen.py | 1 + ...set_perform_maintenance_maximum_set_gen.py | 1 + ...set_perform_maintenance_minimum_set_gen.py | 1 + ...ine_scale_set_power_off_maximum_set_gen.py | 1 + ...ine_scale_set_power_off_minimum_set_gen.py | 1 + ...chine_scale_set_reapply_maximum_set_gen.py | 1 + ...chine_scale_set_reapply_minimum_set_gen.py | 1 + ...hine_scale_set_redeploy_maximum_set_gen.py | 1 + ...hine_scale_set_redeploy_minimum_set_gen.py | 1 + ...e_scale_set_reimage_all_maximum_set_gen.py | 1 + ...e_scale_set_reimage_all_minimum_set_gen.py | 1 + ...chine_scale_set_reimage_maximum_set_gen.py | 1 + ...chine_scale_set_reimage_minimum_set_gen.py | 1 + ...chine_scale_set_restart_maximum_set_gen.py | 1 + ...chine_scale_set_restart_minimum_set_gen.py | 1 + ..._rolling_upgrade_cancel_maximum_set_gen.py | 1 + ..._rolling_upgrade_cancel_minimum_set_gen.py | 1 + ...ling_upgrade_get_latest_maximum_set_gen.py | 1 + ...ling_upgrade_get_latest_minimum_set_gen.py | 1 + ...pgrade_start_os_upgrade_maximum_set_gen.py | 1 + ...pgrade_start_os_upgrade_minimum_set_gen.py | 1 + ...estration_service_state_maximum_set_gen.py | 3 + ...estration_service_state_minimum_set_gen.py | 3 + ...machine_scale_set_start_maximum_set_gen.py | 1 + ...machine_scale_set_start_minimum_set_gen.py | 1 + ...achine_scale_set_update_maximum_set_gen.py | 3 + ...ne_scale_set_vm_approve_rolling_upgrade.py | 1 + ...ttach_detach_data_disks_maximum_set_gen.py | 3 + ...ttach_detach_data_disks_minimum_set_gen.py | 3 + ...scale_set_vm_deallocate_maximum_set_gen.py | 1 + ...scale_set_vm_deallocate_minimum_set_gen.py | 1 + ...rtual_machine_scale_set_vm_delete_force.py | 1 + ...l_machine_scale_set_vm_extension_create.py | 3 + ...l_machine_scale_set_vm_extension_delete.py | 1 + ...tual_machine_scale_set_vm_extension_get.py | 1 + ...ual_machine_scale_set_vm_extension_list.py | 1 + ...l_machine_scale_set_vm_extension_update.py | 3 + ...iew_auto_placed_on_dedicated_host_group.py | 1 + ...machine_scale_set_vm_get_with_user_data.py | 1 + ...cale_set_vm_get_with_vm_size_properties.py | 1 + ...chine_scale_set_vm_list_maximum_set_gen.py | 1 + ...chine_scale_set_vm_list_minimum_set_gen.py | 1 + ..._vm_perform_maintenance_maximum_set_gen.py | 1 + ..._vm_perform_maintenance_minimum_set_gen.py | 1 + ..._scale_set_vm_power_off_maximum_set_gen.py | 1 + ..._scale_set_vm_power_off_minimum_set_gen.py | 1 + ...e_scale_set_vm_redeploy_maximum_set_gen.py | 1 + ...e_scale_set_vm_redeploy_minimum_set_gen.py | 1 + ...cale_set_vm_reimage_all_maximum_set_gen.py | 1 + ...cale_set_vm_reimage_all_minimum_set_gen.py | 1 + ...ne_scale_set_vm_reimage_maximum_set_gen.py | 1 + ...ne_scale_set_vm_reimage_minimum_set_gen.py | 1 + ...ne_scale_set_vm_restart_maximum_set_gen.py | 1 + ...ne_scale_set_vm_restart_minimum_set_gen.py | 1 + ...e_set_vm_retrieve_boot_diagnostics_data.py | 1 + ..._machine_scale_set_vm_simulate_eviction.py | 1 + ...hine_scale_set_vm_start_maximum_set_gen.py | 1 + ...hine_scale_set_vm_start_minimum_set_gen.py | 1 + ...ine_scale_set_vm_update_maximum_set_gen.py | 3 + ...ine_scale_set_vm_update_minimum_set_gen.py | 3 + sdk/compute/azure-mgmt-compute/setup.py | 6 +- .../tests/test_mgmt_compute_disks.py | 3 - .../tests/test_mgmt_compute_galleries.py | 1 - .../tests/test_mgmt_compute_vm.py | 3 - .../tests/test_mgmt_compute_vmss.py | 3 - 778 files changed, 96480 insertions(+), 221622 deletions(-) diff --git a/sdk/compute/azure-mgmt-compute/CHANGELOG.md b/sdk/compute/azure-mgmt-compute/CHANGELOG.md index 694b60d3f170..e7d0aaa6885c 100644 --- a/sdk/compute/azure-mgmt-compute/CHANGELOG.md +++ b/sdk/compute/azure-mgmt-compute/CHANGELOG.md @@ -1,11 +1,15 @@ # Release History -## 30.6.0 (2024-03-18) +## 30.6.0 (2024-03-15) ### Features Added - Model GalleryArtifactVersionFullSource has a new parameter virtual_machine_id +### Bugs Fixed + + - Fix api_version in next_link for paging operation + ## 30.5.0 (2024-01-22) ### Features Added diff --git a/sdk/compute/azure-mgmt-compute/_meta.json b/sdk/compute/azure-mgmt-compute/_meta.json index 9feb31dc97e3..e84b4623de55 100644 --- a/sdk/compute/azure-mgmt-compute/_meta.json +++ b/sdk/compute/azure-mgmt-compute/_meta.json @@ -1,12 +1,12 @@ { - "commit": "f1872412241b16b5e8f0e00408b8fd807bcb75a8", + "commit": "475747ff6322e9bf180b8911d871561b264379c3", "repository_url": "https://github.com/Azure/azure-rest-api-specs", "autorest": "3.9.7", "use": [ - "@autorest/python@6.7.1", - "@autorest/modelerfour@4.26.2" + "@autorest/python@6.13.7", + "@autorest/modelerfour@4.27.0" ], - "autorest_command": "autorest specification/compute/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.7.1 --use=@autorest/modelerfour@4.26.2 --version=3.9.7 --version-tolerant=False", + "autorest_command": "autorest specification/compute/resource-manager/readme.md --generate-sample=True --include-x-ms-examples-original-file=True --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.13.7 --use=@autorest/modelerfour@4.27.0 --version=3.9.7 --version-tolerant=False", "readme": "specification/compute/resource-manager/readme.md", "package-2023-03-01-only": "2023-04-17 22:42:05 -0400 b06a6f45e472dd07c2e0dab06b1e52dfe2684c88 Microsoft.Compute/ComputeRP/stable/2023-03-01/virtualMachineScaleSet.json", "package-2023-01-02-only": "2023-06-28 00:48:15 -0400 b1d41f8dc643415d4dc6ee82bdb2fef04ef45bc9 Microsoft.Compute/DiskRP/stable/2023-01-02/snapshot.json", diff --git a/sdk/compute/azure-mgmt-compute/assets.json b/sdk/compute/azure-mgmt-compute/assets.json index c43592e5ebb5..d5048c9c415c 100644 --- a/sdk/compute/azure-mgmt-compute/assets.json +++ b/sdk/compute/azure-mgmt-compute/assets.json @@ -2,5 +2,5 @@ "AssetsRepo": "Azure/azure-sdk-assets", "AssetsRepoPrefixPath": "python", "TagPrefix": "python/compute/azure-mgmt-compute", - "Tag": "python/compute/azure-mgmt-compute_90b04c9287" + "Tag": "python/compute/azure-mgmt-compute_21614a0446" } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_compute_management_client.py index 2f120bad0eb0..eec5ded6795b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_compute_management_client.py @@ -11,7 +11,9 @@ from typing import Any, Optional, TYPE_CHECKING +from azure.core.pipeline import policies from azure.mgmt.core import ARMPipelineClient +from azure.mgmt.core.policies import ARMAutoResourceProviderRegistrationPolicy from azure.profiles import KnownProfiles, ProfileDefinition from azure.profiles.multiapiclient import MultiApiClientMixin @@ -118,7 +120,25 @@ def __init__( if api_version: kwargs.setdefault('api_version', api_version) self._config = ComputeManagementClientConfiguration(credential, subscription_id, **kwargs) - self._client = ARMPipelineClient(base_url=base_url, config=self._config, **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(base_url=base_url, policies=_policies, **kwargs) super(ComputeManagementClient, self).__init__( api_version=api_version, profile=profile diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_configuration.py index 8f70bb7fa11e..c0f20db4ce78 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_configuration.py @@ -10,7 +10,6 @@ # -------------------------------------------------------------------------- from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -20,7 +19,7 @@ # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): +class ComputeManagementClientConfiguration: """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -42,12 +41,12 @@ def __init__( raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) kwargs.setdefault('sdk_moniker', 'azure-mgmt-compute/{}'.format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_serialization.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_serialization.py index a00658b1fc19..75e26c415d2c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_serialization.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/_serialization.py @@ -63,8 +63,8 @@ import isodate # type: ignore -from azure.core.exceptions import DeserializationError, SerializationError, raise_with_traceback -from azure.core.serialization import NULL as AzureCoreNull +from azure.core.exceptions import DeserializationError, SerializationError +from azure.core.serialization import NULL as CoreNull _BOM = codecs.BOM_UTF8.decode(encoding="utf-8") @@ -124,7 +124,7 @@ def deserialize_from_text(cls, data: Optional[Union[AnyStr, IO]], content_type: pass return ET.fromstring(data_as_str) # nosec - except ET.ParseError: + 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 @@ -143,7 +143,7 @@ def _json_attemp(data): # The function hack is because Py2.7 messes up with exception # context otherwise. _LOGGER.critical("Wasn't XML not JSON, failing") - raise_with_traceback(DeserializationError, "XML is invalid") + raise DeserializationError("XML is invalid") from err raise DeserializationError("Cannot deserialize content-type: {}".format(content_type)) @classmethod @@ -170,13 +170,6 @@ def deserialize_from_http_generics(cls, body_bytes: Optional[Union[AnyStr, IO]], return None -try: - basestring # type: ignore - unicode_str = unicode # type: ignore -except NameError: - basestring = str - unicode_str = str - _LOGGER = logging.getLogger(__name__) try: @@ -295,7 +288,7 @@ class Model(object): _validation: Dict[str, Dict[str, Any]] = {} def __init__(self, **kwargs: Any) -> None: - self.additional_properties: Dict[str, Any] = {} + 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__) @@ -340,7 +333,7 @@ def _create_xml_node(cls): return _create_xml_node(xml_map.get("name", cls.__name__), xml_map.get("prefix", None), xml_map.get("ns", None)) def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: - """Return the JSON that would be sent to azure from this model. + """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)`. @@ -351,7 +344,7 @@ def serialize(self, keep_readonly: bool = False, **kwargs: Any) -> JSON: :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) + return serializer._serialize(self, keep_readonly=keep_readonly, **kwargs) # type: ignore def as_dict( self, @@ -392,7 +385,7 @@ def my_key_transformer(key, attr_desc, value): :rtype: dict """ serializer = Serializer(self._infer_class_models()) - return serializer._serialize(self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs) + return serializer._serialize(self, key_transformer=key_transformer, keep_readonly=keep_readonly, **kwargs) # type: ignore @classmethod def _infer_class_models(cls): @@ -417,7 +410,7 @@ def deserialize(cls: Type[ModelType], data: Any, content_type: Optional[str] = N :raises: DeserializationError if something went wrong """ deserializer = Deserializer(cls._infer_class_models()) - return deserializer(cls.__name__, data, content_type=content_type) + return deserializer(cls.__name__, data, content_type=content_type) # type: ignore @classmethod def from_dict( @@ -447,7 +440,7 @@ def from_dict( if key_extractors is None else key_extractors ) - return deserializer(cls.__name__, data, content_type=content_type) + return deserializer(cls.__name__, data, content_type=content_type) # type: ignore @classmethod def _flatten_subtype(cls, key, objects): @@ -547,7 +540,7 @@ class Serializer(object): "multiple": lambda x, y: x % y != 0, } - def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]]=None): + def __init__(self, classes: Optional[Mapping[str, type]]=None): self.serialize_type = { "iso-8601": Serializer.serialize_iso, "rfc-1123": Serializer.serialize_rfc, @@ -563,7 +556,7 @@ def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]]=None): "[]": self.serialize_iter, "{}": self.serialize_dict, } - self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} + self.dependencies: Dict[str, type] = dict(classes) if classes else {} self.key_transformer = full_restapi_key_transformer self.client_side_validation = True @@ -651,7 +644,7 @@ def _serialize(self, target_obj, data_type=None, **kwargs): else: # That's a basic type # Integrate namespace if necessary local_node = _create_xml_node(xml_name, xml_prefix, xml_ns) - local_node.text = unicode_str(new_attr) + local_node.text = str(new_attr) serialized.append(local_node) # type: ignore else: # JSON for k in reversed(keys): # type: ignore @@ -670,7 +663,7 @@ def _serialize(self, target_obj, data_type=None, **kwargs): except (AttributeError, KeyError, TypeError) as err: msg = "Attribute {} in object {} cannot be serialized.\n{}".format(attr_name, class_name, str(target_obj)) - raise_with_traceback(SerializationError, msg, err) + raise SerializationError(msg) from err else: return serialized @@ -712,7 +705,7 @@ def body(self, data, data_type, **kwargs): ] data = deserializer._deserialize(data_type, data) except DeserializationError as err: - raise_with_traceback(SerializationError, "Unable to build a model: " + str(err), err) + raise SerializationError("Unable to build a model: " + str(err)) from err return self._serialize(data, data_type, **kwargs) @@ -732,6 +725,7 @@ def url(self, name, data, data_type, **kwargs): if kwargs.get("skip_quote") is True: output = str(output) + output = output.replace("{", quote("{")).replace("}", quote("}")) else: output = quote(str(output), safe="") except SerializationError: @@ -746,7 +740,7 @@ def query(self, name, data, data_type, **kwargs): :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 + :rtype: str, list :raises: TypeError if serialization fails. :raises: ValueError if data is None """ @@ -755,7 +749,7 @@ def query(self, name, data, data_type, **kwargs): if data_type.startswith("["): internal_data_type = data_type[1:-1] do_quote = not kwargs.get('skip_quote', False) - return str(self.serialize_iter(data, internal_data_type, do_quote=do_quote, **kwargs)) + 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) @@ -806,7 +800,7 @@ def serialize_data(self, data, data_type, **kwargs): raise ValueError("No value for given attribute") try: - if data is AzureCoreNull: + if data is CoreNull: return None if data_type in self.basic_types.values(): return self.serialize_basic(data, data_type, **kwargs) @@ -826,7 +820,7 @@ def serialize_data(self, data, data_type, **kwargs): except (ValueError, TypeError) as err: msg = "Unable to serialize value: {!r} as type: {!r}." - raise_with_traceback(SerializationError, msg.format(data, data_type), err) + raise SerializationError(msg.format(data, data_type)) from err else: return self._serialize(data, **kwargs) @@ -999,7 +993,7 @@ def serialize_object(self, attr, **kwargs): 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 unicode_str: + if obj_type is str: return self.serialize_unicode(attr) if obj_type is datetime.datetime: return self.serialize_iso(attr) @@ -1176,10 +1170,10 @@ def serialize_iso(attr, **kwargs): return date + microseconds + "Z" except (ValueError, OverflowError) as err: msg = "Unable to serialize datetime object." - raise_with_traceback(SerializationError, msg, err) + raise SerializationError(msg) from err except AttributeError as err: msg = "ISO-8601 object must be valid Datetime object." - raise_with_traceback(TypeError, msg, err) + raise TypeError(msg) from err @staticmethod def serialize_unix(attr, **kwargs): @@ -1215,7 +1209,6 @@ def rest_key_extractor(attr, attr_desc, 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 - # https://github.com/Azure/msrest-for-python/issues/197 return None key = ".".join(dict_keys[1:]) @@ -1236,7 +1229,6 @@ def rest_key_case_insensitive_extractor(attr, attr_desc, 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 - # https://github.com/Azure/msrest-for-python/issues/197 return None key = ".".join(dict_keys[1:]) @@ -1377,7 +1369,7 @@ class Deserializer(object): valid_date = re.compile(r"\d{4}[-]\d{2}[-]\d{2}T\d{2}:\d{2}:\d{2}" r"\.?\d*Z?[-+]?[\d{2}]?:?[\d{2}]?") - def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]]=None): + def __init__(self, classes: Optional[Mapping[str, type]]=None): self.deserialize_type = { "iso-8601": Deserializer.deserialize_iso, "rfc-1123": Deserializer.deserialize_rfc, @@ -1397,7 +1389,7 @@ def __init__(self, classes: Optional[Mapping[str, Type[ModelType]]]=None): "duration": (isodate.Duration, datetime.timedelta), "iso-8601": (datetime.datetime), } - self.dependencies: Dict[str, Type[ModelType]] = dict(classes) if classes else {} + 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 @@ -1450,7 +1442,7 @@ def _deserialize(self, target_obj, data): response, class_name = self._classify_target(target_obj, data) - if isinstance(response, basestring): + if isinstance(response, str): return self.deserialize_data(data, response) elif isinstance(response, type) and issubclass(response, Enum): return self.deserialize_enum(data, response) @@ -1487,7 +1479,7 @@ def _deserialize(self, target_obj, data): d_attrs[attr] = value except (AttributeError, TypeError, KeyError) as err: msg = "Unable to deserialize to object: " + class_name # type: ignore - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: additional_properties = self._build_additional_properties(attributes, data) return self._instantiate_model(response, d_attrs, additional_properties) @@ -1521,14 +1513,14 @@ def _classify_target(self, target, data): if target is None: return None, None - if isinstance(target, basestring): + if isinstance(target, str): try: target = self.dependencies[target] except KeyError: return target, target try: - target = target._classify(data, self.dependencies) + 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 @@ -1584,7 +1576,7 @@ def _unpack_content(raw_data, content_type=None): if hasattr(raw_data, "_content_consumed"): return RawDeserializer.deserialize_from_http_generics(raw_data.text, raw_data.headers) - if isinstance(raw_data, (basestring, bytes)) or hasattr(raw_data, "read"): + 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 @@ -1658,7 +1650,7 @@ def deserialize_data(self, data, data_type): except (ValueError, TypeError, AttributeError) as err: msg = "Unable to deserialize response data." msg += " Data: {}, {}".format(data, data_type) - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: return self._deserialize(obj_type, data) @@ -1706,7 +1698,7 @@ def deserialize_object(self, attr, **kwargs): if isinstance(attr, ET.Element): # Do no recurse on XML, just return the tree as-is return attr - if isinstance(attr, basestring): + if isinstance(attr, str): return self.deserialize_basic(attr, "str") obj_type = type(attr) if obj_type in self.basic_types: @@ -1763,7 +1755,7 @@ def deserialize_basic(self, attr, data_type): if data_type == "bool": if attr in [True, False, 1, 0]: return bool(attr) - elif isinstance(attr, basestring): + elif isinstance(attr, str): if attr.lower() in ["true", "1"]: return True elif attr.lower() in ["false", "0"]: @@ -1814,7 +1806,6 @@ def deserialize_enum(data, enum_obj): data = data.value if isinstance(data, int): # Workaround. We might consider remove it in the future. - # https://github.com/Azure/azure-rest-api-specs/issues/141 try: return list(enum_obj.__members__.values())[data] except IndexError: @@ -1868,10 +1859,10 @@ def deserialize_decimal(attr): if isinstance(attr, ET.Element): attr = attr.text try: - return decimal.Decimal(attr) # type: ignore + return decimal.Decimal(str(attr)) # type: ignore except decimal.DecimalException as err: msg = "Invalid decimal {}".format(attr) - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err @staticmethod def deserialize_long(attr): @@ -1899,7 +1890,7 @@ def deserialize_duration(attr): duration = isodate.parse_duration(attr) except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize duration object." - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: return duration @@ -1916,7 +1907,7 @@ def deserialize_date(attr): 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=None, defaultday=None) + return isodate.parse_date(attr, defaultmonth=0, defaultday=0) @staticmethod def deserialize_time(attr): @@ -1951,7 +1942,7 @@ def deserialize_rfc(attr): date_obj = date_obj.astimezone(tz=TZ_UTC) except ValueError as err: msg = "Cannot deserialize to rfc datetime object." - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: return date_obj @@ -1988,7 +1979,7 @@ def deserialize_iso(attr): raise OverflowError("Hit max or min date") except (ValueError, OverflowError, AttributeError) as err: msg = "Cannot deserialize datetime object." - raise_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: return date_obj @@ -2004,9 +1995,10 @@ def deserialize_unix(attr): 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_with_traceback(DeserializationError, msg, err) + raise DeserializationError(msg) from err else: return date_obj diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_compute_management_client.py index aa4273240504..7e8d01666ae0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_compute_management_client.py @@ -11,7 +11,9 @@ from typing import Any, Optional, TYPE_CHECKING +from azure.core.pipeline import policies from azure.mgmt.core import AsyncARMPipelineClient +from azure.mgmt.core.policies import AsyncARMAutoResourceProviderRegistrationPolicy from azure.profiles import KnownProfiles, ProfileDefinition from azure.profiles.multiapiclient import MultiApiClientMixin @@ -118,7 +120,25 @@ def __init__( if api_version: kwargs.setdefault('api_version', api_version) self._config = ComputeManagementClientConfiguration(credential, subscription_id, **kwargs) - self._client = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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(base_url=base_url, policies=_policies, **kwargs) super(ComputeManagementClient, self).__init__( api_version=api_version, profile=profile diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_configuration.py index 29849edf936a..6b53c8a4c3a9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/aio/_configuration.py @@ -10,7 +10,6 @@ # -------------------------------------------------------------------------- from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -20,7 +19,7 @@ # pylint: disable=unused-import,ungrouped-imports from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): +class ComputeManagementClientConfiguration: """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -42,12 +41,12 @@ def __init__( raise ValueError("Parameter 'credential' must not be None.") if subscription_id is None: raise ValueError("Parameter 'subscription_id' must not be None.") - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) self.credential = credential self.subscription_id = subscription_id self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default']) kwargs.setdefault('sdk_moniker', 'azure-mgmt-compute/{}'.format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_compute_management_client.py index c56e49fdce14..a1fc828b722f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -84,7 +86,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -116,7 +136,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2015-06-15" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -136,7 +156,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_configuration.py index 0bbf322d7749..0a4373142618 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2015-06-15") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_metadata.json index f7cd3610cf62..e1099244af4a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "availability_sets": "AvailabilitySetsOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_compute_management_client.py index bc56b164d73b..54dcb8f45ff0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -85,7 +87,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -117,7 +137,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2015-06-15" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -137,7 +159,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_configuration.py index cc3c7c918173..77639a258e7f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2015-06-15") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/operations/_operations.py index df0eb1447fce..9bfdae79ef2e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -90,10 +89,6 @@ build_virtual_machines_start_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -139,7 +134,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -150,7 +144,7 @@ async def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -162,11 +156,10 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -174,7 +167,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -183,12 +176,8 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -216,7 +205,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -224,16 +213,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -245,13 +233,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( @@ -263,7 +247,6 @@ async def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -282,21 +265,20 @@ async def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -310,13 +292,9 @@ async def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -326,7 +304,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -345,21 +322,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -371,13 +347,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: @@ -385,7 +357,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet] @@ -408,16 +379,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -428,14 +398,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -445,11 +415,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -461,10 +431,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -476,7 +442,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] @@ -499,17 +464,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -520,14 +484,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -537,11 +501,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -553,10 +517,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -592,7 +552,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -611,23 +570,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -639,13 +597,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -657,7 +611,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -676,21 +629,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -702,13 +654,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -736,7 +684,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -755,7 +702,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -764,16 +711,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -785,13 +731,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -819,7 +761,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -845,7 +787,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -854,16 +796,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -883,10 +824,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -913,14 +850,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -934,7 +863,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -950,18 +879,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -975,7 +896,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -988,20 +909,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1035,7 +945,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1045,24 +955,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1088,7 +996,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1097,16 +1005,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1118,13 +1025,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1153,14 +1056,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1174,7 +1069,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1190,18 +1085,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1215,7 +1102,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -1228,20 +1115,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1275,7 +1151,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1285,17 +1161,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -1314,22 +1188,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1343,13 +1216,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -1364,14 +1233,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -1402,7 +1263,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1412,17 +1273,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1444,7 +1303,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -1463,23 +1321,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1491,13 +1348,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -1536,7 +1389,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1555,7 +1407,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -1563,16 +1415,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1584,13 +1435,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -1622,7 +1469,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -1641,7 +1487,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -1651,16 +1497,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1672,13 +1517,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -1690,7 +1531,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -1709,21 +1549,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1735,13 +1574,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -1749,7 +1584,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -1768,20 +1602,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,13 +1626,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -1813,7 +1642,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -1832,22 +1660,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1859,13 +1686,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -1895,7 +1718,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1917,16 +1739,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1937,14 +1758,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -1954,11 +1775,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1970,8 +1791,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -1999,7 +1818,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] @@ -2022,16 +1840,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2042,14 +1859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -2059,11 +1876,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2075,8 +1892,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -2102,7 +1917,7 @@ async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -2128,7 +1943,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2136,16 +1951,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2159,13 +1973,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -2189,14 +1999,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2209,7 +2011,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2222,18 +2024,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2246,7 +2040,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -2257,19 +2051,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2302,7 +2086,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2315,20 +2099,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -2353,7 +2139,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2361,16 +2147,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2390,10 +2175,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -2415,14 +2196,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2435,7 +2208,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2447,18 +2220,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2468,7 +2233,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -2477,19 +2246,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2522,7 +2280,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2532,17 +2290,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2561,21 +2317,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2589,13 +2344,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -2607,14 +2358,6 @@ async def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2644,7 +2387,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2654,17 +2397,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2679,7 +2420,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -2698,22 +2438,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2725,13 +2464,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2750,21 +2485,20 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2778,13 +2512,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -2797,14 +2527,6 @@ async def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2834,7 +2556,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2847,17 +2569,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def generalize( @@ -2869,7 +2589,6 @@ async def generalize( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -2888,21 +2607,20 @@ async def generalize( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2914,13 +2632,9 @@ async def generalize( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -2929,7 +2643,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] @@ -2952,16 +2665,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2972,14 +2684,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2989,11 +2701,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3005,16 +2717,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] @@ -3037,15 +2744,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3056,14 +2762,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3073,11 +2779,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3089,8 +2795,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3102,7 +2806,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] @@ -3125,17 +2828,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3146,14 +2848,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3163,11 +2865,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3179,10 +2881,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -3200,21 +2898,20 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3228,13 +2925,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -3247,14 +2940,6 @@ async def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3284,7 +2969,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3297,17 +2982,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3326,21 +3009,20 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3354,13 +3036,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -3372,14 +3050,6 @@ async def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3409,7 +3079,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3422,17 +3092,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3451,21 +3119,20 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3479,13 +3146,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -3497,14 +3160,6 @@ async def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3534,7 +3189,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3547,17 +3202,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3576,21 +3229,20 @@ async def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3604,13 +3256,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -3622,14 +3270,6 @@ async def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3659,7 +3299,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3672,17 +3312,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -3706,7 +3344,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -3731,7 +3373,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -3739,16 +3381,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3768,10 +3409,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -3793,14 +3430,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -3813,7 +3442,7 @@ async def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3825,18 +3454,10 @@ async def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -3846,7 +3467,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -3854,20 +3479,9 @@ async def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -3900,7 +3514,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3910,17 +3524,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -3939,21 +3551,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3967,13 +3578,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -3985,14 +3592,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4022,7 +3621,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4032,17 +3631,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -4054,7 +3651,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4073,21 +3669,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4099,19 +3694,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -4140,7 +3731,7 @@ async def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4148,16 +3739,15 @@ async def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4171,13 +3761,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload async def begin_deallocate( @@ -4204,14 +3790,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4224,7 +3802,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4239,18 +3817,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4263,7 +3833,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -4275,20 +3845,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4321,7 +3880,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4334,23 +3893,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -4376,7 +3933,7 @@ async def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4384,16 +3941,15 @@ async def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4407,13 +3963,9 @@ async def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload async def begin_delete_instances( @@ -4438,14 +3990,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4458,7 +4002,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4471,18 +4015,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4495,7 +4031,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -4505,20 +4041,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4551,7 +4076,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4564,17 +4089,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_instance_view( @@ -4586,7 +4109,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -4605,21 +4127,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4631,13 +4152,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -4645,7 +4162,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4669,16 +4185,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4689,14 +4204,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -4706,11 +4221,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4722,17 +4237,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4756,15 +4266,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4775,14 +4284,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -4792,11 +4301,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4808,8 +4317,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -4821,7 +4328,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -4845,17 +4351,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4866,14 +4371,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -4883,11 +4388,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4899,15 +4404,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -4936,7 +4437,7 @@ async def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4944,16 +4445,15 @@ async def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4967,13 +4467,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload async def begin_power_off( @@ -5000,14 +4496,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5020,7 +4508,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5035,18 +4523,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5059,7 +4539,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -5071,20 +4551,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5117,7 +4586,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5130,23 +4599,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5175,7 +4642,7 @@ async def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5183,16 +4650,15 @@ async def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5206,13 +4672,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload async def begin_restart( @@ -5237,14 +4699,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5257,7 +4711,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5270,18 +4724,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5294,7 +4740,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -5304,20 +4750,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5350,7 +4785,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5363,23 +4798,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5408,7 +4841,7 @@ async def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5416,16 +4849,15 @@ async def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5439,13 +4871,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload async def begin_start( @@ -5470,14 +4898,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5490,7 +4910,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5503,18 +4923,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5527,7 +4939,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -5537,20 +4949,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5583,7 +4984,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5596,23 +4997,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5638,7 +5037,7 @@ async def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5646,16 +5045,15 @@ async def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5669,13 +5067,9 @@ async def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload async def begin_update_instances( @@ -5700,14 +5094,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5720,7 +5106,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5733,18 +5119,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5757,7 +5135,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -5767,20 +5145,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5813,7 +5180,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5826,17 +5193,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5855,21 +5220,20 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5883,13 +5247,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -5901,14 +5261,6 @@ async def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5938,7 +5290,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5951,17 +5303,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -6001,22 +5351,21 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6030,13 +5379,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -6050,14 +5395,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6088,7 +5425,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6101,17 +5438,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6130,22 +5465,21 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6159,13 +5493,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6181,14 +5511,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6219,7 +5541,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6232,17 +5554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6261,22 +5581,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6290,13 +5609,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -6310,14 +5625,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6348,7 +5655,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6358,17 +5665,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -6382,7 +5687,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -6401,22 +5705,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6428,13 +5731,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -6448,7 +5747,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6467,22 +5765,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6494,13 +5791,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6529,7 +5822,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -6553,7 +5845,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -6561,12 +5853,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6577,14 +5868,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -6594,11 +5885,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6610,10 +5901,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -6631,22 +5918,21 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6660,13 +5946,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -6682,14 +5964,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6720,7 +5994,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6733,17 +6007,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6762,22 +6034,21 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6791,13 +6062,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -6811,14 +6078,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6849,7 +6108,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6862,17 +6121,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6891,22 +6148,21 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6920,13 +6176,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -6940,14 +6192,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6978,7 +6222,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6991,14 +6235,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/models/_models_py3.py index 2cb606b4b014..58de56f77a61 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -202,7 +198,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -260,7 +256,7 @@ class AvailabilitySet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -335,7 +331,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet] @@ -450,7 +446,7 @@ def __init__(self, *, output: Optional[JSON] = None, **kwargs: Any) -> None: class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -609,7 +605,7 @@ def __init__(self, *, boot_diagnostics: Optional["_models.BootDiagnostics"] = No class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_key: Specifies the location of the disk encryption key, which is a Key Vault Secret. Required. @@ -690,7 +686,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
`:code:`
` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
`:code:`
` `List all available virtual machine sizes in an availability set @@ -722,7 +718,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
`:code:`
` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
`:code:`
` `List all available virtual machine sizes in an availability set @@ -891,7 +887,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -924,7 +920,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -1152,7 +1148,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
`:code:`
` Possible @@ -1270,7 +1266,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -1303,7 +1299,7 @@ class OSProfile(_serialization.Model): :code:`
`:code:`
` **Max-length (Windows):** 15 characters :code:`
`:code:`
` **Max-length (Linux):** 64 characters. :code:`
`:code:`
` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
`:code:`
` **Windows-only restriction:** Cannot end in "." :code:`
`:code:`
` **Disallowed values:** @@ -1314,10 +1310,10 @@ class OSProfile(_serialization.Model): :code:`
`:code:`
` **Max-length (Linux):** 64 characters :code:`
`:code:`
` **Max-length (Windows):** 20 characters :code:`
`:code:`
`:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -1330,16 +1326,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -1347,10 +1343,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2015_06_15.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2015_06_15.models.VaultSecretGroup] @@ -1384,7 +1380,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -1396,10 +1392,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -1412,16 +1408,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -1429,10 +1425,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2015_06_15.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -1503,7 +1499,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -1613,7 +1609,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -1631,7 +1627,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -1650,12 +1646,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2015_06_15.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2015_06_15.models.DataDisk] """ @@ -1682,12 +1678,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2015_06_15.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2015_06_15.models.DataDisk] """ super().__init__(**kwargs) @@ -1751,7 +1747,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -1933,7 +1929,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1968,10 +1964,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2015_06_15.models.SubResource @@ -1985,7 +1981,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -2060,10 +2056,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2015_06_15.models.SubResource @@ -2073,7 +2069,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -2135,7 +2131,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -2212,7 +2208,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2330,7 +2326,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -2374,7 +2370,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2597,7 +2593,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2653,7 +2649,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2843,7 +2839,7 @@ class VirtualMachineScaleSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3076,7 +3072,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -3103,7 +3099,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3267,10 +3263,10 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3353,7 +3349,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. Required. :vartype name: str @@ -3458,10 +3454,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3474,10 +3470,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: A base-64 encoded string of custom data. :vartype custom_data: str @@ -3525,10 +3521,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3541,10 +3537,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: A base-64 encoded string of custom data. :paramtype custom_data: str @@ -3675,7 +3671,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3720,10 +3716,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2015_06_15.models.SubResource @@ -3735,7 +3731,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -3816,10 +3812,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2015_06_15.models.SubResource @@ -3831,7 +3827,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -3852,7 +3848,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -3905,10 +3901,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/operations/_operations.py index 071ae1653ca8..84fa8ec0d846 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2015_06_15/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -168,7 +163,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -199,7 +194,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -232,7 +227,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -263,7 +258,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -309,7 +304,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -344,7 +339,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -379,7 +374,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -411,7 +406,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -485,7 +480,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -533,7 +528,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -564,7 +559,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -593,7 +588,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -713,7 +708,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -816,7 +811,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -847,7 +842,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -930,7 +925,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1085,7 +1080,7 @@ def build_virtual_machines_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1119,7 +1114,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1150,7 +1145,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1181,7 +1176,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1215,7 +1210,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1249,7 +1244,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1280,7 +1275,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1310,7 +1305,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1336,7 +1333,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1367,7 +1364,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1401,7 +1398,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1435,7 +1432,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1469,7 +1466,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1503,7 +1500,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1534,7 +1531,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1566,7 +1563,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1598,7 +1595,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1630,7 +1627,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1662,7 +1659,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1694,7 +1691,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1740,7 +1737,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1772,7 +1769,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1804,7 +1801,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1877,7 +1874,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -1888,7 +1884,7 @@ def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1900,11 +1896,10 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -1912,7 +1907,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -1921,12 +1916,8 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -1954,7 +1945,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -1962,16 +1953,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1983,13 +1973,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( @@ -2001,7 +1987,6 @@ def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -2020,21 +2005,20 @@ def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2048,13 +2032,9 @@ def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -2064,7 +2044,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2083,21 +2062,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2109,13 +2087,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: @@ -2123,7 +2097,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -2145,16 +2118,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2165,14 +2137,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -2182,11 +2154,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2198,10 +2170,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -2213,7 +2181,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -2235,17 +2202,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2256,14 +2222,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -2273,11 +2239,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2289,10 +2255,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -2328,7 +2290,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2347,23 +2308,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2375,13 +2335,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -2393,7 +2349,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2412,21 +2367,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2438,13 +2392,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -2472,7 +2422,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2491,7 +2440,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -2500,16 +2449,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2521,13 +2469,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -2555,7 +2499,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2581,7 +2525,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2590,16 +2534,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2619,10 +2562,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -2649,14 +2588,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2670,7 +2601,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2686,18 +2617,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2711,7 +2634,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -2724,20 +2647,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2771,7 +2683,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2781,24 +2693,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2824,7 +2734,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2833,16 +2743,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2854,13 +2763,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2889,14 +2794,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2910,7 +2807,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2926,18 +2823,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2951,7 +2840,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -2964,20 +2853,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3011,7 +2889,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3021,17 +2899,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -3050,22 +2926,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3079,13 +2954,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -3100,14 +2971,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3138,7 +3001,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3148,17 +3011,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3180,7 +3041,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -3199,23 +3059,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3227,13 +3086,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -3272,7 +3127,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3291,7 +3145,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3299,16 +3153,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3320,13 +3173,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -3358,7 +3207,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3377,7 +3225,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3387,16 +3235,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3408,13 +3255,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -3426,7 +3269,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3445,21 +3287,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3471,13 +3312,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -3485,7 +3322,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3504,20 +3340,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3529,13 +3364,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -3549,7 +3380,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3568,22 +3398,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3595,13 +3424,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -3631,7 +3456,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3653,16 +3477,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3673,14 +3496,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -3690,11 +3513,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3706,8 +3529,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -3735,7 +3556,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -3757,16 +3577,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3777,14 +3596,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3794,11 +3613,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3810,8 +3629,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -3837,7 +3654,7 @@ def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -3863,7 +3680,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3871,16 +3688,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3894,13 +3710,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -3924,14 +3736,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3944,7 +3748,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3957,18 +3761,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3981,7 +3777,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -3992,19 +3788,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4037,7 +3823,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4049,20 +3835,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -4087,7 +3875,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4095,16 +3883,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4124,10 +3911,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -4149,14 +3932,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] @@ -4168,7 +3943,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4180,18 +3955,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] @@ -4200,7 +3967,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -4209,19 +3980,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] @@ -4253,7 +4013,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4263,17 +4023,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4292,21 +4050,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4320,13 +4077,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -4338,14 +4091,6 @@ def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4375,7 +4120,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4385,17 +4130,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4410,7 +4153,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -4429,22 +4171,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4456,13 +4197,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4481,21 +4218,20 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4509,13 +4245,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -4528,14 +4260,6 @@ def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4565,7 +4289,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4577,17 +4301,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _models.OperationStatusResponse: @@ -4597,7 +4319,6 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -4616,21 +4337,20 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4642,13 +4362,9 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -4657,7 +4373,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4679,16 +4394,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4699,14 +4413,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4716,11 +4430,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4732,16 +4446,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4763,15 +4472,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4782,14 +4490,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4799,11 +4507,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4815,8 +4523,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4828,7 +4534,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2015_06_15.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -4850,17 +4555,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4871,14 +4575,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4888,11 +4592,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4904,10 +4608,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -4925,21 +4625,20 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4953,13 +4652,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -4972,14 +4667,6 @@ def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5009,7 +4696,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5021,17 +4708,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5050,21 +4735,20 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5078,13 +4762,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -5096,14 +4776,6 @@ def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5133,7 +4805,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5145,17 +4817,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5174,21 +4844,20 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5202,13 +4871,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -5220,14 +4885,6 @@ def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5257,7 +4914,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5269,17 +4926,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5298,21 +4953,20 @@ def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5326,13 +4980,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -5344,14 +4994,6 @@ def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5381,7 +5023,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5393,17 +5035,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -5427,7 +5067,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -5452,7 +5096,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -5460,16 +5104,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5489,10 +5132,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -5514,14 +5153,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5534,7 +5165,7 @@ def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5546,18 +5177,10 @@ def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5567,7 +5190,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -5575,20 +5202,9 @@ def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5621,7 +5237,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5631,17 +5247,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5660,21 +5274,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5688,13 +5301,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -5706,14 +5315,6 @@ def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5743,7 +5344,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5753,17 +5354,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -5773,7 +5372,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -5792,21 +5390,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5818,19 +5415,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5859,7 +5452,7 @@ def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5867,16 +5460,15 @@ def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5890,13 +5482,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload def begin_deallocate( @@ -5923,14 +5511,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5943,7 +5523,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5958,18 +5538,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5982,7 +5554,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -5994,20 +5566,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6040,7 +5601,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6052,23 +5613,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6094,7 +5653,7 @@ def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6102,16 +5661,15 @@ def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6125,13 +5683,9 @@ def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload def begin_delete_instances( @@ -6156,14 +5710,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6176,7 +5722,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6189,18 +5735,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6213,7 +5751,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -6223,20 +5761,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6269,7 +5796,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6281,17 +5808,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_instance_view( @@ -6303,7 +5828,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6322,21 +5846,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6348,13 +5871,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -6362,7 +5881,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6386,16 +5904,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6406,14 +5923,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6423,11 +5940,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6439,17 +5956,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6473,15 +5985,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6492,14 +6003,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -6509,11 +6020,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6525,8 +6036,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6538,7 +6047,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -6562,17 +6070,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6583,14 +6090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -6600,11 +6107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6616,15 +6123,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6653,7 +6156,7 @@ def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6661,16 +6164,15 @@ def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6684,13 +6186,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload def begin_power_off( @@ -6717,14 +6215,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6737,7 +6227,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6752,18 +6242,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6776,7 +6258,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -6788,20 +6270,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6834,7 +6305,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6846,23 +6317,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6891,7 +6360,7 @@ def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6899,16 +6368,15 @@ def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6922,13 +6390,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload def begin_restart( @@ -6953,14 +6417,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6973,7 +6429,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6986,18 +6442,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7010,7 +6458,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -7020,20 +6468,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7066,7 +6503,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7078,23 +6515,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7123,7 +6558,7 @@ def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7131,16 +6566,15 @@ def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7154,13 +6588,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload def begin_start( @@ -7185,14 +6615,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7205,7 +6627,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7218,18 +6640,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7242,7 +6656,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -7252,20 +6666,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7298,7 +6701,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7310,23 +6713,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7352,7 +6753,7 @@ def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7360,16 +6761,15 @@ def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7383,13 +6783,9 @@ def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload def begin_update_instances( @@ -7414,14 +6810,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7434,7 +6822,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7447,18 +6835,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7471,7 +6851,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -7481,20 +6861,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7527,7 +6896,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7539,17 +6908,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7568,21 +6935,20 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7596,13 +6962,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -7614,14 +6976,6 @@ def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7651,7 +7005,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7663,17 +7017,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -7713,22 +7065,21 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7742,13 +7093,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -7762,14 +7109,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7800,7 +7139,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7812,17 +7151,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7841,22 +7178,21 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7870,13 +7206,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -7892,14 +7224,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7930,7 +7254,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7942,17 +7266,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7971,22 +7293,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8000,13 +7321,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -8020,14 +7337,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8058,7 +7367,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8068,17 +7377,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -8092,7 +7399,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -8111,22 +7417,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8138,13 +7443,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -8158,7 +7459,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2015_06_15.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -8177,22 +7477,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8204,13 +7503,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8239,7 +7534,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -8263,7 +7557,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -8271,12 +7565,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8287,14 +7580,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -8304,11 +7597,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8320,10 +7613,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -8341,22 +7630,21 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8370,13 +7658,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -8392,14 +7676,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8430,7 +7706,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8442,17 +7718,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8471,22 +7745,21 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8500,13 +7773,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -8520,14 +7789,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8558,7 +7819,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8570,17 +7831,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8599,22 +7858,21 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2015-06-15")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8628,13 +7886,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -8648,14 +7902,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8686,7 +7932,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8698,14 +7944,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_compute_management_client.py index ec285ec92ab4..ecb3c9d113cf 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -84,7 +86,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -116,7 +136,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2016-03-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -136,7 +156,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_configuration.py index b6b34e890d83..f0d71d62f98c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2016-03-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_metadata.json index 6d780a88c12a..9f2dc081d323 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "availability_sets": "AvailabilitySetsOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_compute_management_client.py index 19e956f77758..a13ca90ebf06 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -85,7 +87,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -117,7 +137,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2016-03-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -137,7 +159,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_configuration.py index f35dffa3dde5..336e87bdaac1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2016-03-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/operations/_operations.py index 34ceecbc1c0c..8eddafbd8baf 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -92,10 +91,6 @@ build_virtual_machines_start_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -141,7 +136,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -152,7 +146,7 @@ async def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -164,11 +158,10 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -176,7 +169,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -185,12 +178,8 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -218,7 +207,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -226,16 +215,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -247,13 +235,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( @@ -265,7 +249,6 @@ async def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -284,21 +267,20 @@ async def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -312,13 +294,9 @@ async def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -328,7 +306,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -347,21 +324,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -373,13 +349,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -390,7 +362,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet] @@ -413,16 +384,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -433,14 +403,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -450,11 +420,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -466,17 +436,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet] @@ -499,16 +464,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -519,14 +483,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -536,11 +500,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -552,10 +516,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -567,7 +527,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] @@ -590,17 +549,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -611,14 +569,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -628,11 +586,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -644,10 +602,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -683,7 +637,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -702,23 +655,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -730,13 +682,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -748,7 +696,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -767,21 +714,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -793,13 +739,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -827,7 +769,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -846,7 +787,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -855,16 +796,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -876,13 +816,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -910,7 +846,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -936,7 +872,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -945,16 +881,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -974,10 +909,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1004,14 +935,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1025,7 +948,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1041,18 +964,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1066,7 +981,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1079,20 +994,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1126,7 +1030,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1136,24 +1040,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1179,7 +1081,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1188,16 +1090,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1209,13 +1110,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1244,14 +1141,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1265,7 +1154,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1281,18 +1170,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1306,7 +1187,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -1319,20 +1200,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1366,7 +1236,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1376,17 +1246,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -1405,22 +1273,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1434,13 +1301,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -1455,14 +1318,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -1493,7 +1348,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1503,17 +1358,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1535,7 +1388,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -1554,23 +1406,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1582,13 +1433,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -1623,7 +1470,6 @@ async def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -1642,22 +1488,21 @@ async def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,19 +1514,15 @@ async def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -1707,7 +1548,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -1715,16 +1556,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1738,13 +1578,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -1768,14 +1604,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1788,7 +1616,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1801,18 +1629,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1825,7 +1645,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -1836,19 +1656,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1881,7 +1691,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1894,20 +1704,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -1932,7 +1744,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -1940,16 +1752,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1969,10 +1780,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -1994,14 +1801,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2014,7 +1813,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2026,18 +1825,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2047,7 +1838,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -2056,19 +1851,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2101,7 +1885,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2111,17 +1895,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2140,21 +1922,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2168,13 +1949,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -2186,14 +1963,6 @@ async def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2223,7 +1992,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2233,17 +2002,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2258,7 +2025,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -2277,22 +2043,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2304,13 +2069,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2329,21 +2090,20 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2357,13 +2117,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -2376,14 +2132,6 @@ async def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2413,7 +2161,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2426,17 +2174,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def generalize( @@ -2448,7 +2194,6 @@ async def generalize( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -2467,21 +2212,20 @@ async def generalize( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2493,13 +2237,9 @@ async def generalize( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -2508,7 +2248,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -2531,16 +2270,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2551,14 +2289,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2568,11 +2306,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2584,16 +2322,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -2616,15 +2349,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2635,14 +2367,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2652,11 +2384,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2668,8 +2400,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2681,7 +2411,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] @@ -2704,17 +2433,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2725,14 +2453,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -2742,11 +2470,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2758,10 +2486,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -2779,21 +2503,20 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2807,13 +2530,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -2826,14 +2545,6 @@ async def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2863,7 +2574,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2876,17 +2587,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2905,21 +2614,20 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2933,13 +2641,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -2951,14 +2655,6 @@ async def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2988,7 +2684,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3001,17 +2697,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3030,21 +2724,20 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3058,13 +2751,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -3076,14 +2765,6 @@ async def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3113,7 +2794,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3126,17 +2807,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3155,21 +2834,20 @@ async def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3183,13 +2861,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -3201,14 +2875,6 @@ async def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3238,7 +2904,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3251,17 +2917,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -3300,7 +2964,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3319,7 +2982,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3327,16 +2990,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3348,13 +3010,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -3386,7 +3044,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3405,7 +3062,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3415,16 +3072,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3436,13 +3092,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -3454,7 +3106,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3473,21 +3124,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3499,13 +3149,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -3513,7 +3159,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3532,20 +3177,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3557,13 +3201,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -3577,7 +3217,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3596,22 +3235,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3623,13 +3261,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -3659,7 +3293,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3681,16 +3314,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3701,14 +3333,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -3718,11 +3350,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3734,8 +3366,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -3763,7 +3393,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] @@ -3786,16 +3415,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3806,14 +3434,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3823,11 +3451,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3839,8 +3467,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -3863,7 +3489,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -3888,7 +3518,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -3896,16 +3526,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3925,10 +3554,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -3950,14 +3575,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -3970,7 +3587,7 @@ async def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3982,18 +3599,10 @@ async def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4003,7 +3612,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -4011,20 +3624,9 @@ async def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4057,7 +3659,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4067,17 +3669,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -4096,21 +3696,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4124,13 +3723,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -4142,14 +3737,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4179,7 +3766,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4189,17 +3776,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -4211,7 +3796,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4230,21 +3814,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4256,19 +3839,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -4297,7 +3876,7 @@ async def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4305,16 +3884,15 @@ async def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4328,13 +3906,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload async def begin_deallocate( @@ -4361,14 +3935,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4381,7 +3947,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4396,18 +3962,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4420,7 +3978,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -4432,20 +3990,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4478,7 +4025,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4491,23 +4038,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -4533,7 +4078,7 @@ async def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4541,16 +4086,15 @@ async def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4564,13 +4108,9 @@ async def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload async def begin_delete_instances( @@ -4595,14 +4135,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4615,7 +4147,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4628,18 +4160,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4652,7 +4176,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -4662,20 +4186,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4708,7 +4221,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4721,17 +4234,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_instance_view( @@ -4743,7 +4254,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -4762,21 +4272,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4788,13 +4297,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -4802,7 +4307,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4826,16 +4330,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4846,14 +4349,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -4863,11 +4366,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4879,17 +4382,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4913,15 +4411,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4932,14 +4429,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -4949,11 +4446,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4965,8 +4462,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -4978,7 +4473,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -5002,17 +4496,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5023,14 +4516,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -5040,11 +4533,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5056,15 +4549,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5093,7 +4582,7 @@ async def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5101,16 +4590,15 @@ async def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5124,13 +4612,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload async def begin_power_off( @@ -5157,14 +4641,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5177,7 +4653,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5192,18 +4668,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5216,7 +4684,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -5228,20 +4696,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5274,7 +4731,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5287,23 +4744,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5332,7 +4787,7 @@ async def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5340,16 +4795,15 @@ async def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5363,13 +4817,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload async def begin_restart( @@ -5394,14 +4844,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5414,7 +4856,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5427,18 +4869,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5451,7 +4885,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -5461,20 +4895,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5507,7 +4930,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5520,23 +4943,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5565,7 +4986,7 @@ async def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5573,16 +4994,15 @@ async def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5596,13 +5016,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload async def begin_start( @@ -5627,14 +5043,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5647,7 +5055,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5660,18 +5068,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5684,7 +5084,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -5694,20 +5094,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5740,7 +5129,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5753,23 +5142,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5795,7 +5182,7 @@ async def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5803,16 +5190,15 @@ async def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5826,13 +5212,9 @@ async def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload async def begin_update_instances( @@ -5857,14 +5239,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5877,7 +5251,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5890,18 +5264,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5914,7 +5280,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -5924,20 +5290,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5970,7 +5325,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5983,17 +5338,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6012,21 +5365,20 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6040,13 +5392,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -6058,14 +5406,6 @@ async def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6095,7 +5435,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6108,17 +5448,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -6158,22 +5496,21 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6187,13 +5524,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -6207,14 +5540,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6245,7 +5570,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6258,17 +5583,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6287,22 +5610,21 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6316,13 +5638,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6338,14 +5656,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6376,7 +5686,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6389,17 +5699,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6418,22 +5726,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6447,13 +5754,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -6467,14 +5770,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6505,7 +5800,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6515,17 +5810,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -6539,7 +5832,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -6558,22 +5850,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6585,13 +5876,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -6605,7 +5892,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6624,22 +5910,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6651,13 +5936,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6686,7 +5967,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -6710,7 +5990,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -6718,12 +5998,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6734,14 +6013,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -6751,11 +6030,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6767,10 +6046,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -6788,22 +6063,21 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6817,13 +6091,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -6839,14 +6109,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6877,7 +6139,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6890,17 +6152,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6919,22 +6179,21 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6948,13 +6207,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -6968,14 +6223,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7006,7 +6253,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7019,17 +6266,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7048,22 +6293,21 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7077,13 +6321,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -7097,14 +6337,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7135,7 +6367,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7148,14 +6380,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/models/_models_py3.py index 809b1df1dab4..7d61969491d0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -202,7 +198,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -260,7 +256,7 @@ class AvailabilitySet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -335,7 +331,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet] @@ -450,7 +446,7 @@ def __init__(self, *, output: Optional[JSON] = None, **kwargs: Any) -> None: class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -684,7 +680,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -718,7 +714,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -889,7 +885,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -922,7 +918,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -995,7 +991,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.Usage] @@ -1154,7 +1150,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -1272,7 +1268,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -1305,7 +1301,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -1316,10 +1312,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -1332,16 +1328,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -1349,10 +1345,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2016_03_30.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2016_03_30.models.VaultSecretGroup] @@ -1386,7 +1382,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -1398,10 +1394,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -1414,16 +1410,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -1431,10 +1427,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2016_03_30.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -1505,7 +1501,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -1615,7 +1611,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -1633,7 +1629,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -1652,12 +1648,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2016_03_30.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2016_03_30.models.DataDisk] """ @@ -1684,12 +1680,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2016_03_30.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2016_03_30.models.DataDisk] """ super().__init__(**kwargs) @@ -1753,7 +1749,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -1935,7 +1931,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1972,10 +1968,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2016_03_30.models.SubResource @@ -1989,7 +1985,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -2073,10 +2069,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2016_03_30.models.SubResource @@ -2086,7 +2082,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -2150,7 +2146,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -2227,7 +2223,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2345,7 +2341,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -2389,7 +2385,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2671,7 +2667,7 @@ def __init__(self, *, type: Optional[Literal["SystemAssigned"]] = None, **kwargs class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2727,7 +2723,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2885,7 +2881,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -2923,7 +2919,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3200,7 +3196,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -3227,7 +3223,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3305,7 +3301,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet] @@ -3341,7 +3337,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetSku] @@ -3377,7 +3373,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet] @@ -3410,10 +3406,10 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3496,7 +3492,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. Required. :vartype name: str @@ -3601,10 +3597,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3617,10 +3613,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: A base-64 encoded string of custom data. :vartype custom_data: str @@ -3668,10 +3664,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3684,10 +3680,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: A base-64 encoded string of custom data. :paramtype custom_data: str @@ -3818,7 +3814,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3865,10 +3861,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2016_03_30.models.SubResource @@ -3880,7 +3876,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -3963,10 +3959,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2016_03_30.models.SubResource @@ -3976,7 +3972,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -3998,7 +3994,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -4051,10 +4047,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -4162,7 +4158,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVM] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/operations/_operations.py index 1f6d7b3dee3c..d2192c905744 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_03_30/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -140,7 +135,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -196,7 +191,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -227,7 +222,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -260,7 +255,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -291,7 +286,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -337,7 +332,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -372,7 +367,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -407,7 +402,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -439,7 +434,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -479,7 +474,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_get_extensions_request( +def build_virtual_machines_get_extensions_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -546,7 +541,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -649,7 +644,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -680,7 +675,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -763,7 +758,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -952,7 +947,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1000,7 +995,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1031,7 +1026,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1060,7 +1055,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1146,7 +1141,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1180,7 +1175,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1211,7 +1206,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1242,7 +1237,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1276,7 +1271,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1310,7 +1305,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1341,7 +1336,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1371,7 +1366,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1397,7 +1394,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1428,7 +1425,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1462,7 +1459,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1496,7 +1493,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1530,7 +1527,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1564,7 +1561,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1595,7 +1592,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1627,7 +1624,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1659,7 +1656,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1691,7 +1688,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1723,7 +1720,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1755,7 +1752,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1801,7 +1798,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1833,7 +1830,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1865,7 +1862,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1938,7 +1935,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -1949,7 +1945,7 @@ def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1961,11 +1957,10 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -1973,7 +1968,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -1982,12 +1977,8 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2015,7 +2006,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -2023,16 +2014,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +2034,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( @@ -2062,7 +2048,6 @@ def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -2081,21 +2066,20 @@ def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2109,13 +2093,9 @@ def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -2125,7 +2105,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2144,21 +2123,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2170,13 +2148,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -2187,7 +2161,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -2209,16 +2182,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2229,14 +2201,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -2246,11 +2218,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2262,17 +2234,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -2294,16 +2261,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2314,14 +2280,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -2331,11 +2297,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2347,10 +2313,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -2362,7 +2324,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -2384,17 +2345,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2405,14 +2365,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -2422,11 +2382,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2438,10 +2398,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -2477,7 +2433,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2496,23 +2451,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2524,13 +2478,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -2542,7 +2492,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2561,21 +2510,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2587,13 +2535,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -2621,7 +2565,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2640,7 +2583,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -2649,16 +2592,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2670,13 +2612,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -2704,7 +2642,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2730,7 +2668,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2739,16 +2677,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2768,10 +2705,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -2798,14 +2731,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2819,7 +2744,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2835,18 +2760,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2860,7 +2777,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -2873,20 +2790,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2920,7 +2826,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2930,24 +2836,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2973,7 +2877,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2982,16 +2886,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3003,13 +2906,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3038,14 +2937,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3059,7 +2950,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3075,18 +2966,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3100,7 +2983,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -3113,20 +2996,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3160,7 +3032,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3170,17 +3042,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -3199,22 +3069,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3228,13 +3097,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -3249,14 +3114,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3287,7 +3144,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3297,17 +3154,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3329,7 +3184,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -3348,23 +3202,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3376,13 +3229,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -3417,7 +3266,6 @@ def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3436,22 +3284,21 @@ def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3463,19 +3310,15 @@ def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -3501,7 +3344,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3509,16 +3352,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3532,13 +3374,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -3562,14 +3400,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3582,7 +3412,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3595,18 +3425,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3619,7 +3441,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -3630,19 +3452,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3675,7 +3487,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3687,20 +3499,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3725,7 +3539,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3733,16 +3547,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3762,10 +3575,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -3787,14 +3596,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -3806,7 +3607,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3818,18 +3619,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -3838,7 +3631,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -3847,19 +3644,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] @@ -3891,7 +3677,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3901,17 +3687,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3930,21 +3714,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3958,13 +3741,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -3976,14 +3755,6 @@ def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4013,7 +3784,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4023,17 +3794,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4048,7 +3817,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -4067,22 +3835,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4094,13 +3861,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4119,21 +3882,20 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4147,13 +3909,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -4166,14 +3924,6 @@ def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4203,7 +3953,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4215,17 +3965,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _models.OperationStatusResponse: @@ -4235,7 +3983,6 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -4254,21 +4001,20 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4280,13 +4026,9 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -4295,7 +4037,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4317,16 +4058,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4337,14 +4077,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4354,11 +4094,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4370,16 +4110,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4401,15 +4136,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4420,14 +4154,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4437,11 +4171,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4453,8 +4187,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4466,7 +4198,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -4488,17 +4219,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4509,14 +4239,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4526,11 +4256,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4542,10 +4272,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -4563,21 +4289,20 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4591,13 +4316,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -4610,14 +4331,6 @@ def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4647,7 +4360,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4659,17 +4372,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4688,21 +4399,20 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4716,13 +4426,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -4734,14 +4440,6 @@ def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4771,7 +4469,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4783,17 +4481,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4812,21 +4508,20 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4840,13 +4535,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -4858,14 +4549,6 @@ def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4895,7 +4578,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4907,17 +4590,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4936,21 +4617,20 @@ def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4964,13 +4644,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -4982,14 +4658,6 @@ def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5019,7 +4687,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5031,17 +4699,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -5080,7 +4746,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5099,7 +4764,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -5107,16 +4772,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5128,13 +4792,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5166,7 +4826,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5185,7 +4844,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -5195,16 +4854,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5216,13 +4874,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -5234,7 +4888,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5253,21 +4906,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5279,13 +4931,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -5293,7 +4941,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5312,20 +4959,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5337,13 +4983,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -5357,7 +4999,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5376,22 +5017,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5403,13 +5043,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5439,7 +5075,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5461,16 +5096,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5481,14 +5115,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5498,11 +5132,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5514,8 +5148,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -5543,7 +5175,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -5565,16 +5196,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5585,14 +5215,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5602,11 +5232,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5618,8 +5248,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -5642,7 +5270,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -5667,7 +5299,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -5675,16 +5307,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5704,10 +5335,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -5729,14 +5356,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5749,7 +5368,7 @@ def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5761,18 +5380,10 @@ def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5782,7 +5393,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -5790,20 +5405,9 @@ def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5836,7 +5440,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5846,17 +5450,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5875,21 +5477,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5903,13 +5504,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -5921,14 +5518,6 @@ def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5958,7 +5547,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5968,17 +5557,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -5988,7 +5575,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6007,21 +5593,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6033,19 +5618,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6074,7 +5655,7 @@ def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6082,16 +5663,15 @@ def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6105,13 +5685,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload def begin_deallocate( @@ -6138,14 +5714,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6158,7 +5726,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6173,18 +5741,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6197,7 +5757,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -6209,20 +5769,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6255,7 +5804,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6267,23 +5816,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6309,7 +5856,7 @@ def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6317,16 +5864,15 @@ def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6340,13 +5886,9 @@ def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload def begin_delete_instances( @@ -6371,14 +5913,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6391,7 +5925,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6404,18 +5938,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6428,7 +5954,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -6438,20 +5964,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6484,7 +5999,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6496,17 +6011,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_instance_view( @@ -6518,7 +6031,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6537,21 +6049,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6563,13 +6074,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -6577,7 +6084,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6601,16 +6107,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6621,14 +6126,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6638,11 +6143,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6654,17 +6159,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6688,15 +6188,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6707,14 +6206,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -6724,11 +6223,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6740,8 +6239,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6753,7 +6250,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -6777,17 +6273,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6798,14 +6293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -6815,11 +6310,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6831,15 +6326,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6868,7 +6359,7 @@ def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6876,16 +6367,15 @@ def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6899,13 +6389,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload def begin_power_off( @@ -6932,14 +6418,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6952,7 +6430,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6967,18 +6445,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6991,7 +6461,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -7003,20 +6473,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7049,7 +6508,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7061,23 +6520,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7106,7 +6563,7 @@ def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7114,16 +6571,15 @@ def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7137,13 +6593,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload def begin_restart( @@ -7168,14 +6620,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7188,7 +6632,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7201,18 +6645,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7225,7 +6661,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -7235,20 +6671,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7281,7 +6706,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7293,23 +6718,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7338,7 +6761,7 @@ def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7346,16 +6769,15 @@ def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7369,13 +6791,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload def begin_start( @@ -7400,14 +6818,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7420,7 +6830,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7433,18 +6843,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7457,7 +6859,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -7467,20 +6869,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7513,7 +6904,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7525,23 +6916,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7567,7 +6956,7 @@ def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7575,16 +6964,15 @@ def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7598,13 +6986,9 @@ def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload def begin_update_instances( @@ -7629,14 +7013,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7649,7 +7025,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7662,18 +7038,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7686,7 +7054,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -7696,20 +7064,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7742,7 +7099,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7754,17 +7111,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7783,21 +7138,20 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7811,13 +7165,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -7829,14 +7179,6 @@ def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7866,7 +7208,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7878,17 +7220,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -7928,22 +7268,21 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7957,13 +7296,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -7977,14 +7312,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8015,7 +7342,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8027,17 +7354,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8056,22 +7381,21 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8085,13 +7409,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -8107,14 +7427,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8145,7 +7457,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8157,17 +7469,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8186,22 +7496,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8215,13 +7524,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -8235,14 +7540,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8273,7 +7570,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8283,17 +7580,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -8307,7 +7602,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -8326,22 +7620,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8353,13 +7646,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -8373,7 +7662,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_03_30.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -8392,22 +7680,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8419,13 +7706,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8454,7 +7737,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -8478,7 +7760,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -8486,12 +7768,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8502,14 +7783,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -8519,11 +7800,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8535,10 +7816,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -8556,22 +7833,21 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8585,13 +7861,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -8607,14 +7879,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8645,7 +7909,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8657,17 +7921,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8686,22 +7948,21 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8715,13 +7976,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -8735,14 +7992,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8773,7 +8022,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8785,17 +8034,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8814,22 +8061,21 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2016-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8843,13 +8089,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -8863,14 +8105,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8901,7 +8135,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8913,14 +8147,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_compute_management_client.py index 3c5b16798308..079ede6df400 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -94,7 +96,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -137,7 +157,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2016-04-30-preview" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -157,7 +177,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_configuration.py index fb67119662ab..44542c23cb77 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2016-04-30-preview") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_metadata.json index 10a2b0689cd6..c8b95119c079 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "availability_sets": "AvailabilitySetsOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_compute_management_client.py index 6d142f23bfba..6e061dab653a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -94,7 +96,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -137,7 +157,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2016-04-30-preview" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -157,7 +179,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_configuration.py index aa05aa647f55..e0d000aef7ba 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2016-04-30-preview") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/operations/_operations.py index 431837eff848..78ec13713707 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -116,10 +115,6 @@ build_virtual_machines_start_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -165,7 +160,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -176,7 +170,7 @@ async def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -188,11 +182,10 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -200,7 +193,7 @@ async def create_or_update( @distributed_trace_async async def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -209,12 +202,8 @@ async def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -244,7 +233,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -252,16 +241,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -273,13 +261,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( @@ -291,7 +275,6 @@ async def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -312,21 +295,20 @@ async def delete( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -340,13 +322,9 @@ async def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -356,7 +334,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -377,21 +354,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar ) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -403,19 +379,14 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet] @@ -440,15 +411,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Availabi def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -459,14 +429,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -476,11 +446,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -492,17 +462,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet] @@ -527,16 +492,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -547,14 +511,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -564,11 +528,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -580,10 +544,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -595,7 +555,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -620,17 +579,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -641,14 +599,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -658,11 +616,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -674,10 +632,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -713,7 +667,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -734,23 +687,22 @@ async def get( ) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -762,13 +714,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -780,7 +728,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -801,21 +748,20 @@ async def list_types( ) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -827,13 +773,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -861,7 +803,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -882,7 +823,7 @@ async def list_versions( ) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -891,16 +832,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -912,13 +852,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -946,7 +882,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -974,7 +910,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -983,16 +919,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1012,10 +947,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1043,14 +974,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1064,7 +987,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1080,18 +1003,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1105,7 +1020,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1118,20 +1033,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1167,7 +1071,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1177,24 +1081,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1222,7 +1124,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1231,16 +1133,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1252,13 +1153,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1287,14 +1184,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1308,7 +1197,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1324,18 +1213,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1349,7 +1230,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -1362,20 +1243,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1411,7 +1281,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1421,17 +1291,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -1452,22 +1320,21 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1481,13 +1348,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -1502,14 +1365,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -1542,7 +1397,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1552,17 +1407,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1584,7 +1437,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -1605,23 +1457,22 @@ async def get( ) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1633,13 +1484,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -1674,7 +1521,6 @@ async def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -1695,22 +1541,21 @@ async def get_extensions( ) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1722,19 +1567,15 @@ async def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -1762,7 +1603,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -1770,16 +1611,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,13 +1633,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -1824,14 +1660,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1844,7 +1672,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1857,18 +1685,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1881,7 +1701,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -1892,20 +1712,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineCaptureParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1940,7 +1749,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1953,20 +1762,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -1993,7 +1804,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2001,16 +1812,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2030,10 +1840,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -2055,14 +1861,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2075,7 +1873,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2087,18 +1885,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2108,7 +1898,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -2117,19 +1911,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2164,7 +1947,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2174,17 +1957,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2205,21 +1986,20 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2233,13 +2013,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -2251,14 +2027,6 @@ async def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2290,7 +2058,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2300,17 +2068,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2325,7 +2091,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -2346,22 +2111,21 @@ async def get( ) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2373,13 +2137,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2400,21 +2160,20 @@ async def _convert_to_managed_disks_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2428,13 +2187,9 @@ async def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -2447,14 +2202,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2486,7 +2233,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2499,17 +2246,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2530,21 +2275,20 @@ async def _deallocate_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2558,13 +2302,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -2577,14 +2317,6 @@ async def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2616,7 +2348,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2629,17 +2361,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def generalize( @@ -2651,7 +2381,6 @@ async def generalize( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -2672,21 +2401,20 @@ async def generalize( ) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2698,13 +2426,9 @@ async def generalize( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -2713,7 +2437,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine] @@ -2738,16 +2461,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2758,14 +2480,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2775,11 +2497,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2791,16 +2513,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine] @@ -2825,15 +2542,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2844,14 +2560,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2861,11 +2577,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2877,8 +2593,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2890,7 +2604,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -2915,17 +2628,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2936,14 +2648,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -2953,11 +2665,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2969,10 +2681,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -2992,21 +2700,20 @@ async def _power_off_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3020,13 +2727,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -3039,14 +2742,6 @@ async def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3078,7 +2773,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3091,17 +2786,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3122,21 +2815,20 @@ async def _restart_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3150,13 +2842,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -3168,14 +2856,6 @@ async def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3207,7 +2887,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3220,17 +2900,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3251,21 +2929,20 @@ async def _start_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3279,13 +2956,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -3297,14 +2970,6 @@ async def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3336,7 +3001,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3349,17 +3014,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3380,21 +3043,20 @@ async def _redeploy_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3408,13 +3070,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -3426,14 +3084,6 @@ async def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3465,7 +3115,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3478,17 +3128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -3527,7 +3175,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3548,7 +3195,7 @@ async def get( ) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3556,16 +3203,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3577,13 +3223,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -3615,7 +3257,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3636,7 +3277,7 @@ async def list( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3646,16 +3287,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3667,13 +3307,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -3685,7 +3321,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3706,21 +3341,20 @@ async def list_offers( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3732,13 +3366,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -3746,7 +3376,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3767,20 +3396,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3792,13 +3420,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -3812,7 +3436,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -3833,22 +3456,21 @@ async def list_skus( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3860,13 +3482,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -3896,7 +3514,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Usage] @@ -3921,16 +3538,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3941,14 +3557,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -3958,11 +3574,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3974,8 +3590,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -4003,7 +3617,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -4028,16 +3641,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4048,14 +3660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4065,11 +3677,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4081,8 +3693,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -4105,7 +3715,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -4132,7 +3742,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -4140,16 +3750,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4169,10 +3778,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -4194,14 +3799,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -4213,7 +3810,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4225,18 +3822,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -4245,7 +3834,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -4254,19 +3843,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -4300,7 +3878,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4310,17 +3888,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -4341,21 +3917,20 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4369,13 +3944,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -4387,14 +3958,6 @@ async def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4426,7 +3989,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4436,17 +3999,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -4460,7 +4021,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -4481,22 +4041,21 @@ async def get( ) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4508,13 +4067,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -4522,7 +4077,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -4547,16 +4101,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4567,14 +4120,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -4584,11 +4137,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4600,16 +4153,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -4634,15 +4182,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4653,14 +4200,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -4670,11 +4217,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4686,8 +4233,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -4710,7 +4255,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -4737,7 +4286,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -4745,16 +4294,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4774,10 +4322,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -4799,14 +4343,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4819,7 +4355,7 @@ async def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4831,18 +4367,10 @@ async def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4852,7 +4380,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -4860,20 +4392,10 @@ async def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -4908,7 +4430,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4918,17 +4440,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -4949,21 +4469,20 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4977,13 +4496,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -4995,14 +4510,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5034,7 +4541,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5044,17 +4551,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -5066,7 +4571,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -5087,21 +4591,20 @@ async def get( ) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5113,19 +4616,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5156,7 +4655,7 @@ async def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5164,16 +4663,15 @@ async def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5187,13 +4685,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload async def begin_deallocate( @@ -5220,14 +4714,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5240,7 +4726,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5255,18 +4741,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5279,7 +4757,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -5291,20 +4769,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5339,7 +4806,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5352,23 +4819,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5396,7 +4861,7 @@ async def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5404,16 +4869,15 @@ async def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5427,13 +4891,9 @@ async def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload async def begin_delete_instances( @@ -5458,14 +4918,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5478,7 +4930,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5491,18 +4943,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5515,7 +4959,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -5525,21 +4969,10 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceRequiredIDs or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5574,7 +5007,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5587,17 +5020,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_instance_view( @@ -5609,7 +5040,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -5630,21 +5060,20 @@ async def get_instance_view( ) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5656,13 +5085,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -5670,7 +5095,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5696,16 +5120,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5716,14 +5139,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -5733,11 +5156,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5749,17 +5172,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5785,15 +5203,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5804,14 +5221,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -5821,11 +5238,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5837,8 +5254,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5850,7 +5265,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -5876,17 +5290,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5897,14 +5310,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -5914,11 +5327,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5930,15 +5343,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5969,7 +5378,7 @@ async def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5977,16 +5386,15 @@ async def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6000,13 +5408,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload async def begin_power_off( @@ -6033,14 +5437,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6053,7 +5449,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6068,18 +5464,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6092,7 +5480,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -6104,20 +5492,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6152,7 +5529,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6165,23 +5542,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6212,7 +5587,7 @@ async def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6220,16 +5595,15 @@ async def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6243,13 +5617,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload async def begin_restart( @@ -6274,14 +5644,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6294,7 +5656,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6307,18 +5669,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6331,7 +5685,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -6341,20 +5695,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6389,7 +5732,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6402,23 +5745,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6449,7 +5790,7 @@ async def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6457,16 +5798,15 @@ async def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6480,13 +5820,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload async def begin_start( @@ -6511,14 +5847,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6531,7 +5859,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6544,18 +5872,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6568,7 +5888,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -6578,20 +5898,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6626,7 +5935,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6639,23 +5948,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6683,7 +5990,7 @@ async def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6691,16 +5998,15 @@ async def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6714,13 +6020,9 @@ async def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload async def begin_update_instances( @@ -6745,14 +6047,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6765,7 +6059,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6778,18 +6072,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6802,7 +6088,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -6812,21 +6098,10 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceRequiredIDs or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6861,7 +6136,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6874,17 +6149,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6905,21 +6178,20 @@ async def _reimage_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6933,13 +6205,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -6951,14 +6219,6 @@ async def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6990,7 +6250,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7003,17 +6263,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7034,21 +6292,20 @@ async def _reimage_all_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7062,13 +6319,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -7081,14 +6334,6 @@ async def begin_reimage_all( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7120,7 +6365,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7133,17 +6378,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -7185,22 +6428,21 @@ async def _reimage_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7214,13 +6456,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -7234,14 +6472,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7274,7 +6504,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7287,17 +6517,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7318,22 +6546,21 @@ async def _reimage_all_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7347,13 +6574,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -7368,14 +6591,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7408,7 +6623,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7421,17 +6636,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7452,22 +6665,21 @@ async def _deallocate_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7481,13 +6693,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -7503,14 +6711,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7543,7 +6743,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7556,17 +6756,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7587,22 +6785,21 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7616,13 +6813,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -7636,14 +6829,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7676,7 +6861,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7686,17 +6871,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -7710,7 +6893,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7731,22 +6913,21 @@ async def get( ) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7758,13 +6939,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7778,7 +6955,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7799,22 +6975,21 @@ async def get_instance_view( ) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7826,13 +7001,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7857,7 +7028,6 @@ def list( :paramtype select: str :keyword expand: The expand expression to apply to the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7883,7 +7053,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7891,12 +7061,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7907,14 +7076,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7924,11 +7093,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7940,10 +7109,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -7963,22 +7128,21 @@ async def _power_off_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7992,13 +7156,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -8014,14 +7174,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8054,7 +7206,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8067,17 +7219,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8098,22 +7248,21 @@ async def _restart_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8127,13 +7276,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -8147,14 +7292,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8187,7 +7324,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8200,17 +7337,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8231,22 +7366,21 @@ async def _start_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8260,13 +7394,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -8280,14 +7410,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8320,7 +7442,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8333,17 +7455,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -8367,7 +7487,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -8394,7 +7514,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -8402,16 +7522,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8431,10 +7550,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -8458,14 +7573,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8476,7 +7583,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8490,18 +7597,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8509,7 +7608,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -8520,19 +7619,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8565,7 +7653,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8575,20 +7663,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -8615,7 +7701,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -8623,16 +7709,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8652,10 +7737,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -8679,14 +7760,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8697,7 +7770,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8711,18 +7784,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8730,7 +7795,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -8741,19 +7806,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -8786,7 +7840,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8796,17 +7850,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -8818,7 +7870,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -8839,21 +7890,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> ) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8865,13 +7915,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -8892,21 +7938,20 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8920,13 +7965,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -8940,14 +7981,6 @@ async def begin_delete( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8979,7 +8012,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8989,17 +8022,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -9007,7 +8038,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] @@ -9032,16 +8062,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9052,14 +8081,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -9069,11 +8098,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9085,15 +8114,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] @@ -9118,15 +8142,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9137,14 +8160,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -9154,11 +8177,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9170,13 +8193,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -9204,7 +8225,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -9212,16 +8233,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9235,13 +8255,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -9267,14 +8283,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -9287,7 +8295,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9302,18 +8310,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -9326,7 +8326,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -9338,19 +8338,9 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or + IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -9385,7 +8375,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9395,17 +8385,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -9426,21 +8414,20 @@ async def _revoke_access_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9454,13 +8441,9 @@ async def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -9474,14 +8457,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9513,7 +8488,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9526,17 +8501,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SnapshotsOperations: @@ -9560,7 +8533,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -9587,7 +8560,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -9595,16 +8568,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9624,10 +8596,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -9650,14 +8618,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9670,7 +8630,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9683,18 +8643,10 @@ async def begin_create_or_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9704,7 +8656,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -9714,19 +8666,8 @@ async def begin_create_or_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9761,7 +8702,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9771,20 +8712,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -9811,7 +8754,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -9819,16 +8762,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9848,10 +8790,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -9875,14 +8813,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9895,7 +8825,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9909,18 +8839,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9930,7 +8852,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -9940,19 +8866,8 @@ async def begin_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: @@ -9987,7 +8902,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9997,17 +8912,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -10018,7 +8931,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -10039,21 +8951,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) ) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10065,13 +8976,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -10092,21 +8999,20 @@ async def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10120,13 +9026,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -10139,14 +9041,6 @@ async def begin_delete( :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10178,7 +9072,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10188,17 +9082,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -10206,7 +9098,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] @@ -10231,16 +9122,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10251,14 +9141,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -10268,11 +9158,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10284,15 +9174,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] @@ -10317,15 +9202,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10336,14 +9220,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -10353,11 +9237,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10369,13 +9253,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -10403,7 +9285,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -10411,16 +9293,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10434,13 +9315,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -10465,14 +9342,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -10485,7 +9354,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10499,18 +9368,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -10523,7 +9384,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -10534,19 +9395,9 @@ async def begin_grant_access( group. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or + IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: @@ -10581,7 +9432,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10591,17 +9442,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -10622,21 +9471,20 @@ async def _revoke_access_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10650,13 +9498,9 @@ async def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -10669,14 +9513,6 @@ async def begin_revoke_access( :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10708,7 +9544,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10721,14 +9557,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_compute_management_client_enums.py index 1341f31ae077..644efe4f8448 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_compute_management_client_enums.py @@ -35,10 +35,9 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): EMPTY = "Empty" ATTACH = "Attach" FROM_IMAGE = "FromImage" - IMPORT = "Import" + IMPORT_ENUM = "Import" COPY = "Copy" RESTORE = "Restore" - IMPORT_ENUM = "Import" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_models_py3.py index 339daaad88e5..b421cd3e0948 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -225,7 +221,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -283,7 +279,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -372,7 +368,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet] @@ -487,10 +483,10 @@ def __init__(self, *, output: Optional[JSON] = None, **kwargs: Any) -> None: class CreationData(_serialization.Model): """Data used when creating a disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :vartype create_option: str or ~azure.mgmt.compute.v2016_04_30_preview.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -531,7 +527,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :paramtype create_option: str or ~azure.mgmt.compute.v2016_04_30_preview.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier @@ -559,7 +555,7 @@ def __init__( class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -727,7 +723,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -915,7 +911,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] @@ -1090,7 +1086,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None" and "Read". :vartype access: str or ~azure.mgmt.compute.v2016_04_30_preview.models.AccessLevel @@ -1125,7 +1121,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set `_ @@ -1157,7 +1153,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set `_ @@ -1190,7 +1186,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1257,7 +1253,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1337,7 +1333,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1375,7 +1371,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.Image] @@ -1409,7 +1405,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -1586,18 +1582,18 @@ def __init__( class ImageStorageProfile(_serialization.Model): """Describes a storage profile. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long Required. :vartype os_disk: ~azure.mgmt.compute.v2016_04_30_preview.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2016_04_30_preview.models.ImageDataDisk] """ @@ -1621,13 +1617,13 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long Required. :paramtype os_disk: ~azure.mgmt.compute.v2016_04_30_preview.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2016_04_30_preview.models.ImageDataDisk] """ super().__init__(**kwargs) @@ -1720,7 +1716,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2016_04_30_preview.models.SourceVault @@ -1753,7 +1749,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2016_04_30_preview.models.SourceVault @@ -1786,7 +1782,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -1819,7 +1815,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -1892,7 +1888,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.Usage] @@ -2067,7 +2063,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -2194,7 +2190,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -2229,7 +2225,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -2240,10 +2236,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2256,16 +2252,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2274,10 +2270,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2016_04_30_preview.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2016_04_30_preview.models.VaultSecretGroup] @@ -2311,7 +2307,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -2323,10 +2319,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2339,16 +2335,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2357,10 +2353,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2016_04_30_preview.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual @@ -2432,7 +2428,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -2514,7 +2510,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2625,7 +2621,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] @@ -2780,7 +2776,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -2798,7 +2794,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -2817,12 +2813,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2016_04_30_preview.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2016_04_30_preview.models.DataDisk] """ @@ -2849,12 +2845,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2016_04_30_preview.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2016_04_30_preview.models.DataDisk] """ super().__init__(**kwargs) @@ -2941,7 +2937,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -3125,7 +3121,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3163,10 +3159,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2016_04_30_preview.models.SubResource @@ -3181,7 +3177,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -3266,10 +3262,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2016_04_30_preview.models.SubResource @@ -3279,7 +3275,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -3343,7 +3339,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -3420,7 +3416,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3538,7 +3534,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -3582,7 +3578,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3864,7 +3860,7 @@ def __init__(self, *, type: Optional[Literal["SystemAssigned"]] = None, **kwargs class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3920,7 +3916,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4082,7 +4078,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine] @@ -4120,7 +4116,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4241,7 +4237,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -4512,7 +4508,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -4539,7 +4535,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4617,7 +4613,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet] @@ -4653,7 +4649,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetSku] @@ -4690,7 +4686,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet] @@ -4723,7 +4719,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Possible @@ -4750,10 +4746,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4836,7 +4832,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -4951,10 +4947,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -4967,10 +4963,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: A base-64 encoded string of custom data. :vartype custom_data: str @@ -5019,10 +5015,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5035,10 +5031,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: A base-64 encoded string of custom data. :paramtype custom_data: str @@ -5181,7 +5177,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5229,10 +5225,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2016_04_30_preview.models.SubResource @@ -5244,7 +5240,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -5328,10 +5324,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2016_04_30_preview.models.SubResource @@ -5341,7 +5337,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -5363,7 +5359,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5416,10 +5412,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -5540,7 +5536,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVM] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/operations/_operations.py index 8e265e81c5a0..8e8ace59b7be 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2016_04_30_preview/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -140,7 +135,9 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -192,7 +189,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -223,7 +220,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -256,7 +253,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -287,7 +284,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -333,7 +330,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -368,7 +365,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -403,7 +400,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -435,7 +432,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -475,7 +472,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_get_extensions_request( +def build_virtual_machines_get_extensions_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -542,7 +539,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -645,7 +642,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -676,7 +673,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -707,7 +704,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -790,7 +787,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -979,7 +976,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1027,7 +1024,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1058,7 +1055,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1087,7 +1084,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1271,7 +1268,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1325,7 +1322,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1359,7 +1356,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1390,7 +1387,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1421,7 +1418,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1455,7 +1452,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1489,7 +1486,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1520,7 +1517,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1550,7 +1547,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1576,7 +1575,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1607,7 +1606,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1641,7 +1640,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1675,7 +1674,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1709,7 +1708,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1743,7 +1742,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1774,7 +1773,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1805,7 +1804,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1837,7 +1836,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1869,7 +1868,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1901,7 +1900,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1933,7 +1932,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1965,7 +1964,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1997,7 +1996,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2043,7 +2042,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2075,7 +2074,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2107,7 +2106,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2269,7 +2268,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2518,7 +2517,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2678,7 +2677,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2689,7 +2687,7 @@ def create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2701,11 +2699,10 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2713,7 +2710,7 @@ def create_or_update( @distributed_trace def create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO], **kwargs: Any + self, resource_group_name: str, name: str, parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -2722,12 +2719,8 @@ def create_or_update( :param name: The name of the availability set. Required. :type name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2757,7 +2750,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -2765,16 +2758,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2786,13 +2778,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( @@ -2804,7 +2792,6 @@ def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -2825,21 +2812,20 @@ def delete( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2853,13 +2839,9 @@ def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -2869,7 +2851,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -2890,21 +2871,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An ) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2916,19 +2896,14 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet] @@ -2953,15 +2928,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.AvailabilityS def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2972,14 +2946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -2989,11 +2963,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3005,17 +2979,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.AvailabilitySet] @@ -3040,16 +3009,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3060,14 +3028,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3077,11 +3045,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3093,10 +3061,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -3108,7 +3072,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -3133,17 +3096,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3154,14 +3116,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3171,11 +3133,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3187,10 +3149,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -3226,7 +3184,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3247,23 +3204,22 @@ def get( ) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3275,13 +3231,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -3293,7 +3245,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3314,21 +3265,20 @@ def list_types( ) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3340,13 +3290,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -3374,7 +3320,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3395,7 +3340,7 @@ def list_versions( ) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3404,16 +3349,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3425,13 +3369,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3459,7 +3399,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3487,7 +3427,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3496,16 +3436,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3525,10 +3464,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -3556,14 +3491,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3577,7 +3504,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3593,18 +3520,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3618,7 +3537,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -3631,20 +3550,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3680,7 +3588,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3690,24 +3598,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3735,7 +3641,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3744,16 +3650,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3765,13 +3670,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3800,14 +3701,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3821,7 +3714,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3837,18 +3730,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3862,7 +3747,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -3875,20 +3760,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3924,7 +3798,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3934,17 +3808,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -3965,22 +3837,21 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3994,13 +3865,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -4015,14 +3882,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4055,7 +3914,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4065,17 +3924,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4097,7 +3954,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4118,23 +3974,22 @@ def get( ) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4146,13 +4001,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -4187,7 +4038,6 @@ def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4208,22 +4058,21 @@ def get_extensions( ) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4235,19 +4084,15 @@ def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -4275,7 +4120,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4283,16 +4128,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4306,13 +4150,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -4337,14 +4177,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4357,7 +4189,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4370,18 +4202,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4394,7 +4218,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -4405,20 +4229,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineCaptureParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4453,7 +4266,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4465,20 +4278,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -4505,7 +4320,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4513,16 +4328,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4542,10 +4356,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -4567,14 +4377,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4587,7 +4389,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4599,18 +4401,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4620,7 +4414,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -4629,19 +4427,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4676,7 +4463,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4686,17 +4473,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4717,21 +4502,20 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4745,13 +4529,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -4763,14 +4543,6 @@ def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4802,7 +4574,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4812,17 +4584,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4837,7 +4607,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -4858,22 +4627,21 @@ def get( ) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4885,13 +4653,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4912,21 +4676,20 @@ def _convert_to_managed_disks_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4703,9 @@ def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace def begin_convert_to_managed_disks( @@ -4959,14 +4718,6 @@ def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4998,7 +4749,7 @@ def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5010,17 +4761,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5041,21 +4790,20 @@ def _deallocate_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5069,13 +4817,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -5088,14 +4832,6 @@ def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5127,7 +4863,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5139,17 +4875,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _models.OperationStatusResponse: @@ -5159,7 +4893,6 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -5180,21 +4913,20 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ ) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5206,13 +4938,9 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -5221,7 +4949,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine] @@ -5246,16 +4973,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5266,14 +4992,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5283,11 +5009,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5299,16 +5025,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachine] @@ -5333,15 +5054,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5352,14 +5072,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5369,11 +5089,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5385,8 +5105,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5398,7 +5116,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -5423,17 +5140,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5444,14 +5160,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5461,11 +5177,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5477,10 +5193,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -5500,21 +5212,20 @@ def _power_off_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5528,13 +5239,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -5547,14 +5254,6 @@ def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5586,7 +5285,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5598,17 +5297,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5629,21 +5326,20 @@ def _restart_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5657,13 +5353,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -5675,14 +5367,6 @@ def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5714,7 +5398,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5726,17 +5410,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5757,21 +5439,20 @@ def _start_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5785,13 +5466,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -5803,14 +5480,6 @@ def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5842,7 +5511,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5854,17 +5523,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5885,21 +5552,20 @@ def _redeploy_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5913,13 +5579,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -5931,14 +5593,6 @@ def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5970,7 +5624,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5982,17 +5636,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -6031,7 +5683,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6052,7 +5703,7 @@ def get( ) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6060,16 +5711,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6081,13 +5731,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6119,7 +5765,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6140,7 +5785,7 @@ def list( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6150,16 +5795,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6171,13 +5815,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -6189,7 +5829,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6210,21 +5849,20 @@ def list_offers( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6236,13 +5874,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -6250,7 +5884,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6271,20 +5904,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6296,13 +5928,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -6316,7 +5944,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6337,22 +5964,21 @@ def list_skus( ) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6364,13 +5990,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -6400,7 +6022,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6424,16 +6045,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6444,14 +6064,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6461,11 +6081,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6477,8 +6097,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6506,7 +6124,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineSize] @@ -6531,16 +6148,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6551,14 +6167,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6568,11 +6184,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6584,8 +6200,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -6608,7 +6222,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -6635,7 +6249,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -6643,16 +6257,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6672,10 +6285,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -6697,14 +6306,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6715,7 +6316,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6727,18 +6328,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6746,7 +6339,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -6755,19 +6348,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6800,7 +6382,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6810,17 +6392,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -6841,21 +6421,20 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6869,13 +6448,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -6887,14 +6462,6 @@ def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6926,7 +6493,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6936,17 +6503,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -6960,7 +6525,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -6981,22 +6545,21 @@ def get( ) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7008,13 +6571,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -7022,7 +6581,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7046,16 +6604,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7066,14 +6623,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7083,11 +6640,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7099,16 +6656,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7132,15 +6684,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7151,14 +6702,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7168,11 +6719,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7184,8 +6735,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -7208,7 +6757,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { 401: ClientAuthenticationError, @@ -7235,7 +6788,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, name=name, subscription_id=self._config.subscription_id, @@ -7243,16 +6796,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7272,10 +6824,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -7297,14 +6845,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7317,7 +6857,7 @@ def begin_create_or_update( self, resource_group_name: str, name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7329,18 +6869,10 @@ def begin_create_or_update( :param name: The name of the VM scale set to create or update. Required. :type name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7350,7 +6882,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, name: str, parameters: Union[_models.VirtualMachineScaleSet, IO], **kwargs: Any + self, + resource_group_name: str, + name: str, + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -7358,20 +6894,10 @@ def begin_create_or_update( :type resource_group_name: str :param name: The name of the VM scale set to create or update. Required. :type name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7406,7 +6932,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7416,17 +6942,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7447,21 +6971,20 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7475,13 +6998,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -7493,14 +7012,6 @@ def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7532,7 +7043,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7542,17 +7053,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -7562,7 +7071,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7583,21 +7091,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - ) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7609,19 +7116,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7652,7 +7155,7 @@ def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7660,16 +7163,15 @@ def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7683,13 +7185,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload def begin_deallocate( @@ -7716,14 +7214,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7736,7 +7226,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7751,18 +7241,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7775,7 +7257,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7787,20 +7269,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7835,7 +7306,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7847,23 +7318,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7891,7 +7360,7 @@ def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7899,16 +7368,15 @@ def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7922,13 +7390,9 @@ def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload def begin_delete_instances( @@ -7953,14 +7417,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7973,7 +7429,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7986,18 +7442,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8010,7 +7458,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -8020,21 +7468,10 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceRequiredIDs or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8069,7 +7506,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8081,17 +7518,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_instance_view( @@ -8103,7 +7538,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -8124,21 +7558,20 @@ def get_instance_view( ) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8150,13 +7583,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -8164,7 +7593,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8190,16 +7618,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8210,14 +7637,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -8227,11 +7654,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8243,17 +7670,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8279,15 +7701,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8298,14 +7719,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -8315,11 +7736,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8331,8 +7752,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8344,7 +7763,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -8370,17 +7788,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8391,14 +7808,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -8408,11 +7825,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8424,15 +7841,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8463,7 +7876,7 @@ def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8471,16 +7884,15 @@ def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8494,13 +7906,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload def begin_power_off( @@ -8527,14 +7935,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8547,7 +7947,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8562,18 +7962,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8586,7 +7978,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -8598,20 +7990,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8646,7 +8027,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8658,23 +8039,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8705,7 +8084,7 @@ def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8713,16 +8092,15 @@ def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8736,13 +8114,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload def begin_restart( @@ -8767,14 +8141,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8787,7 +8153,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8800,18 +8166,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8824,7 +8182,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -8834,20 +8192,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8882,7 +8229,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8894,23 +8241,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8941,7 +8286,7 @@ def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8949,16 +8294,15 @@ def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8972,13 +8316,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload def begin_start( @@ -9003,14 +8343,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9023,7 +8355,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9036,18 +8368,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9060,7 +8384,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -9070,20 +8394,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9118,7 +8431,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9130,23 +8443,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9174,7 +8485,7 @@ def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9182,16 +8493,15 @@ def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9205,13 +8515,9 @@ def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload def begin_update_instances( @@ -9236,14 +8542,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9256,7 +8554,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9269,18 +8567,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9293,7 +8583,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9303,21 +8593,10 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceRequiredIDs or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9352,7 +8631,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9364,17 +8643,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9395,21 +8672,20 @@ def _reimage_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9423,13 +8699,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -9441,14 +8713,6 @@ def begin_reimage( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9480,7 +8744,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9492,17 +8756,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9523,21 +8785,20 @@ def _reimage_all_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9551,13 +8812,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @distributed_trace def begin_reimage_all( @@ -9570,14 +8827,6 @@ def begin_reimage_all( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9609,7 +8858,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9621,17 +8870,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetVMsOperations: @@ -9673,22 +8920,21 @@ def _reimage_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9702,13 +8948,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -9722,14 +8964,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9762,7 +8996,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9774,17 +9008,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9805,22 +9037,21 @@ def _reimage_all_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9834,13 +9065,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace def begin_reimage_all( @@ -9855,14 +9082,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9895,7 +9114,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9907,17 +9126,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9938,22 +9155,21 @@ def _deallocate_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9967,13 +9183,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -9989,14 +9201,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10029,7 +9233,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10041,17 +9245,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10072,22 +9274,21 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10101,13 +9302,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -10121,14 +9318,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10161,7 +9350,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10171,17 +9360,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -10195,7 +9382,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -10216,22 +9402,21 @@ def get( ) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10243,13 +9428,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -10263,7 +9444,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10284,22 +9464,21 @@ def get_instance_view( ) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10311,13 +9490,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10342,7 +9517,6 @@ def list( :paramtype select: str :keyword expand: The expand expression to apply to the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10368,7 +9542,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -10376,12 +9550,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10392,14 +9565,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -10409,11 +9582,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10425,10 +9598,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -10448,22 +9617,21 @@ def _power_off_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10477,13 +9645,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -10499,14 +9663,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10539,7 +9695,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10551,17 +9707,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10582,22 +9736,21 @@ def _restart_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10611,13 +9764,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -10631,14 +9780,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10671,7 +9812,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10683,17 +9824,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10714,22 +9853,21 @@ def _start_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10743,13 +9881,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -10763,14 +9897,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10803,7 +9929,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10815,17 +9941,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -10849,7 +9973,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -10876,7 +10000,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -10884,16 +10008,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10913,10 +10036,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -10940,14 +10059,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10958,7 +10069,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10972,18 +10083,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10991,7 +10094,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -11002,19 +10105,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11047,7 +10139,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11057,20 +10149,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -11097,7 +10187,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -11105,16 +10195,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11134,10 +10223,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -11161,14 +10246,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11179,7 +10256,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11193,18 +10270,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11212,7 +10281,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -11223,19 +10292,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2016_04_30_preview.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11268,7 +10326,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11278,17 +10336,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -11300,7 +10356,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -11321,21 +10376,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model ) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11347,13 +10401,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -11374,21 +10424,20 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11402,13 +10451,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -11422,14 +10467,6 @@ def begin_delete( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11461,7 +10498,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11471,17 +10508,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -11489,7 +10524,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11513,16 +10547,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11533,14 +10566,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -11550,11 +10583,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11566,15 +10599,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -11598,15 +10626,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11617,14 +10644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -11634,11 +10661,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11650,13 +10677,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -11684,7 +10709,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -11692,16 +10717,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11715,13 +10739,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -11747,14 +10767,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11765,7 +10777,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11780,18 +10792,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11802,7 +10806,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -11814,19 +10818,9 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or + IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11859,7 +10853,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11869,17 +10863,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -11900,21 +10892,20 @@ def _revoke_access_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11928,13 +10919,9 @@ def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace def begin_revoke_access( @@ -11948,14 +10935,6 @@ def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11987,7 +10966,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11999,17 +10978,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SnapshotsOperations: @@ -12033,7 +11010,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -12060,7 +11037,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -12068,16 +11045,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12097,10 +11073,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -12123,14 +11095,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12141,7 +11105,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12154,18 +11118,10 @@ def begin_create_or_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12173,7 +11129,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -12183,19 +11139,8 @@ def begin_create_or_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12228,7 +11173,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12238,20 +11183,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -12278,7 +11225,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -12286,16 +11233,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12315,10 +11261,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -12342,14 +11284,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12360,7 +11294,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12374,18 +11308,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12393,7 +11319,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -12403,19 +11333,8 @@ def begin_update( group. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2016_04_30_preview.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12448,7 +11367,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12458,17 +11377,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -12479,7 +11396,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -12500,21 +11416,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m ) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12526,13 +11441,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -12553,21 +11464,20 @@ def _delete_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12581,13 +11491,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -12600,14 +11506,6 @@ def begin_delete( :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12639,7 +11537,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12649,17 +11547,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -12667,7 +11563,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12691,16 +11586,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12711,14 +11605,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -12728,11 +11622,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12744,15 +11638,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -12776,15 +11665,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12795,14 +11683,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -12812,11 +11700,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12828,13 +11716,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -12862,7 +11748,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -12870,16 +11756,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12893,13 +11778,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -12924,14 +11805,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -12942,7 +11815,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12956,18 +11829,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -12978,7 +11843,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -12989,19 +11854,9 @@ def begin_grant_access( group. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2016_04_30_preview.models.GrantAccessData or + IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -13034,7 +11889,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13044,17 +11899,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -13075,21 +11928,20 @@ def _revoke_access_initial( ) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13103,13 +11955,9 @@ def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace def begin_revoke_access( @@ -13122,14 +11970,6 @@ def begin_revoke_access( :param snapshot_name: The name of the snapshot within the given subscription and resource group. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13161,7 +12001,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13173,14 +12013,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_compute_management_client.py index d1e363275adc..935f6daa5184 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -110,7 +112,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -159,7 +179,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-03-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -179,7 +199,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_configuration.py index 91dccd873d9a..f671d3751f1f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-03-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_metadata.json index 7eefd6432015..5a4ba1e7793c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "availability_sets": "AvailabilitySetsOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_compute_management_client.py index 42790477d603..658a9af574e9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -111,7 +113,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -160,7 +180,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-03-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -180,7 +202,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_configuration.py index f30a418e563c..c4fdc4007e7a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-03-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/operations/_operations.py index 66c14d74d9d9..4558db66d0fe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -131,10 +130,6 @@ build_virtual_machines_start_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -180,7 +175,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -191,7 +185,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -203,11 +197,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -218,7 +211,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -228,12 +221,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -261,7 +250,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -269,16 +258,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -290,13 +278,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( @@ -308,7 +292,6 @@ async def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -327,21 +310,20 @@ async def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -355,13 +337,9 @@ async def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -371,7 +349,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -390,21 +367,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -416,13 +392,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -433,7 +405,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet] @@ -456,16 +427,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -476,14 +446,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -493,11 +463,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -509,17 +479,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet] @@ -542,16 +507,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -562,14 +526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -579,11 +543,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -595,10 +559,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -610,7 +570,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] @@ -633,17 +592,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -654,14 +612,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -671,11 +629,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -687,10 +645,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -726,7 +680,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -745,23 +698,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -773,13 +725,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -791,7 +739,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -810,21 +757,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -836,13 +782,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -870,7 +812,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -889,7 +830,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -898,16 +839,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -919,13 +859,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -953,7 +889,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -979,7 +915,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -988,16 +924,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1017,10 +952,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1047,14 +978,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1068,7 +991,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1084,18 +1007,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1109,7 +1024,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1122,20 +1037,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1169,7 +1073,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1179,24 +1083,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1222,7 +1124,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1231,16 +1133,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1252,13 +1153,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1287,14 +1184,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1308,7 +1197,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1324,18 +1213,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1349,7 +1230,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -1362,20 +1243,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1409,7 +1279,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1419,17 +1289,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -1448,22 +1316,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1477,13 +1344,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -1498,14 +1361,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -1536,7 +1391,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1546,17 +1401,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1578,7 +1431,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,23 +1449,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1625,13 +1476,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -1666,7 +1513,6 @@ async def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -1685,22 +1531,21 @@ async def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1712,13 +1557,9 @@ async def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore @distributed_trace def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -1727,7 +1568,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -1750,16 +1590,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1770,14 +1609,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -1787,11 +1626,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1803,15 +1642,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -1837,7 +1672,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -1845,16 +1680,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1868,13 +1702,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -1898,14 +1728,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1918,7 +1740,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1931,18 +1753,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -1955,7 +1769,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -1966,19 +1780,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2011,7 +1815,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2024,20 +1828,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -2062,7 +1868,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2070,16 +1876,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2099,10 +1904,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -2124,14 +1925,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2144,7 +1937,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2156,18 +1949,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2177,7 +1962,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -2186,19 +1975,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2231,7 +2009,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2241,17 +2019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2270,21 +2046,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2298,13 +2073,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -2316,14 +2087,6 @@ async def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2353,7 +2116,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2363,17 +2126,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2388,7 +2149,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -2407,22 +2167,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2434,13 +2193,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -2452,7 +2207,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -2471,21 +2225,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2497,13 +2250,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2522,21 +2271,20 @@ async def _convert_to_managed_disks_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2550,13 +2298,9 @@ async def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -2569,14 +2313,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2606,7 +2342,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2619,17 +2355,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2648,21 +2382,20 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2676,13 +2409,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -2695,14 +2424,6 @@ async def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2732,7 +2453,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2745,17 +2466,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def generalize( @@ -2767,7 +2486,6 @@ async def generalize( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -2786,21 +2504,20 @@ async def generalize( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2812,13 +2529,9 @@ async def generalize( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -2827,7 +2540,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -2850,16 +2562,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2870,14 +2581,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2887,11 +2598,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2903,16 +2614,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -2935,15 +2641,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2954,14 +2659,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2971,11 +2676,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2987,8 +2692,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3000,7 +2703,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] @@ -3023,17 +2725,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3044,14 +2745,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3061,11 +2762,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3077,10 +2778,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -3098,21 +2795,20 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3126,13 +2822,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -3145,14 +2837,6 @@ async def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3182,7 +2866,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3195,17 +2879,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3224,21 +2906,20 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3252,13 +2933,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -3270,14 +2947,6 @@ async def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3307,7 +2976,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3320,17 +2989,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3349,21 +3016,20 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3377,13 +3043,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -3395,14 +3057,6 @@ async def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3432,7 +3086,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3445,17 +3099,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3474,21 +3126,20 @@ async def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3502,13 +3153,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -3520,14 +3167,6 @@ async def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3557,7 +3196,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3570,17 +3209,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _perform_maintenance_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3599,21 +3236,20 @@ async def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3627,13 +3263,9 @@ async def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -3645,14 +3277,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3682,7 +3306,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3695,20 +3319,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -3733,7 +3359,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3741,16 +3367,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3764,13 +3389,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -3793,14 +3414,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -3813,7 +3426,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3825,18 +3438,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -3846,7 +3451,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -3855,19 +3464,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -3900,7 +3498,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3913,17 +3511,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -3962,7 +3558,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3981,7 +3576,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -3989,16 +3584,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4010,13 +3604,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4048,7 +3638,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4067,7 +3656,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4077,16 +3666,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4098,13 +3686,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4116,7 +3700,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4135,21 +3718,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4161,13 +3743,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4175,7 +3753,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4194,20 +3771,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4219,13 +3795,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4239,7 +3811,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4258,22 +3829,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4285,13 +3855,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -4321,7 +3887,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4343,16 +3908,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4363,14 +3927,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -4380,11 +3944,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4396,8 +3960,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -4425,7 +3987,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] @@ -4448,16 +4009,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4468,14 +4028,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4485,11 +4045,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4501,8 +4061,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -4525,7 +4083,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -4550,7 +4108,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -4558,16 +4116,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4587,10 +4144,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -4612,14 +4165,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -4630,7 +4175,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4642,18 +4187,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -4661,7 +4198,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -4670,19 +4207,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -4713,7 +4239,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4723,17 +4249,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -4752,21 +4276,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4780,13 +4303,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -4798,14 +4317,6 @@ async def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4835,7 +4346,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4845,17 +4356,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -4869,7 +4378,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -4888,22 +4396,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4915,13 +4422,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -4929,7 +4432,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -4951,16 +4453,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4971,14 +4472,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -4988,11 +4489,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5004,16 +4505,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5035,15 +4531,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5054,14 +4549,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -5071,11 +4566,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5087,8 +4582,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class ResourceSkusOperations: """ @@ -5114,7 +4607,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.ResourceSku] @@ -5137,15 +4629,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ResourceSku"]: def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5156,14 +4647,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -5173,11 +4664,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5189,8 +4680,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -5216,7 +4705,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -5242,7 +4731,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5250,16 +4739,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5279,10 +4767,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -5304,14 +4788,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5324,7 +4800,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5336,18 +4812,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5360,7 +4828,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -5369,20 +4837,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5415,7 +4872,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5425,23 +4882,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -5467,7 +4922,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5475,16 +4930,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5496,13 +4950,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -5525,14 +4975,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5545,7 +4987,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5557,18 +4999,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5581,7 +5015,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -5590,20 +5024,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5636,7 +5060,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5646,17 +5070,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5675,21 +5097,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5703,13 +5124,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -5721,14 +5138,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5758,7 +5167,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5768,17 +5177,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -5790,7 +5197,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -5809,21 +5215,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5835,19 +5240,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -5876,7 +5277,7 @@ async def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5884,16 +5285,15 @@ async def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5907,13 +5307,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload async def begin_deallocate( @@ -5940,14 +5336,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5960,7 +5348,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5975,18 +5363,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5999,7 +5379,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -6011,20 +5391,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6057,7 +5426,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6070,23 +5439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6112,7 +5479,7 @@ async def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6120,16 +5487,15 @@ async def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6143,13 +5509,9 @@ async def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload async def begin_delete_instances( @@ -6174,14 +5536,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6194,7 +5548,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6207,18 +5561,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6231,7 +5577,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -6241,20 +5587,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6287,7 +5622,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6300,17 +5635,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_instance_view( @@ -6322,7 +5655,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6341,21 +5673,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6367,13 +5698,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -6381,7 +5708,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6405,16 +5731,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6425,14 +5750,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6442,11 +5767,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6458,17 +5783,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6492,15 +5812,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6511,14 +5830,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -6528,11 +5847,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6544,8 +5863,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6557,7 +5874,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -6581,17 +5897,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6602,14 +5917,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -6619,11 +5934,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6635,15 +5950,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6672,7 +5983,7 @@ async def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6680,16 +5991,15 @@ async def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6703,13 +6013,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload async def begin_power_off( @@ -6736,14 +6042,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6756,7 +6054,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6771,18 +6069,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6795,7 +6085,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -6807,20 +6097,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6853,7 +6132,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6866,23 +6145,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6911,7 +6188,7 @@ async def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6919,16 +6196,15 @@ async def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6942,13 +6218,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload async def begin_restart( @@ -6973,14 +6245,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6993,7 +6257,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7006,18 +6270,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7030,7 +6286,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -7040,20 +6296,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7086,7 +6331,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7099,23 +6344,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7144,7 +6387,7 @@ async def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7152,16 +6395,15 @@ async def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7175,13 +6417,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload async def begin_start( @@ -7206,14 +6444,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7226,7 +6456,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7239,18 +6469,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7263,7 +6485,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -7273,20 +6495,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7319,7 +6530,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7332,23 +6543,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7374,7 +6583,7 @@ async def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7382,16 +6591,15 @@ async def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7405,13 +6613,9 @@ async def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload async def begin_update_instances( @@ -7436,14 +6640,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7456,7 +6652,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7469,18 +6665,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7493,7 +6681,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -7503,20 +6691,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7549,7 +6726,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7562,23 +6739,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7607,7 +6782,7 @@ async def _reimage_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7615,16 +6790,15 @@ async def _reimage_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7638,13 +6812,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @overload async def begin_reimage( @@ -7669,14 +6839,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7689,7 +6851,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7702,18 +6864,10 @@ async def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7726,7 +6880,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -7736,20 +6890,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7782,7 +6925,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7795,23 +6938,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7840,7 +6981,7 @@ async def _reimage_all_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7848,16 +6989,15 @@ async def _reimage_all_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7871,13 +7011,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @overload async def begin_reimage_all( @@ -7903,14 +7039,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7923,7 +7051,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7937,18 +7065,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7961,7 +7081,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -7972,20 +7092,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8018,7 +7127,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8031,20 +7140,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -8069,7 +7176,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -8095,7 +7202,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -8104,16 +7211,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8133,10 +7239,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8164,14 +7266,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -8185,7 +7279,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8201,18 +7295,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -8226,7 +7312,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -8239,20 +7325,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -8286,7 +7361,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8296,17 +7371,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -8325,22 +7398,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8354,13 +7426,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -8375,14 +7443,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8413,7 +7473,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8423,17 +7483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -8455,7 +7513,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -8474,23 +7531,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8502,13 +7558,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8520,7 +7572,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -8544,17 +7595,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8565,14 +7615,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -8582,11 +7632,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8598,12 +7648,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -8640,21 +7686,20 @@ async def _cancel_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8668,13 +7713,9 @@ async def _cancel_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return deserialized # type: ignore @distributed_trace_async async def begin_cancel( @@ -8686,14 +7727,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8723,7 +7756,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8736,17 +7769,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_os_upgrade_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8765,21 +7796,20 @@ async def _start_os_upgrade_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8793,13 +7823,9 @@ async def _start_os_upgrade_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -8813,14 +7839,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8850,7 +7868,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8863,17 +7881,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_latest( @@ -8885,7 +7901,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -8904,21 +7919,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8930,13 +7944,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: @@ -8976,22 +7986,21 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9005,13 +8014,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -9025,14 +8030,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9063,7 +8060,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9076,17 +8073,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9105,22 +8100,21 @@ async def _reimage_all_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9134,13 +8128,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -9155,14 +8145,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9193,7 +8175,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9206,17 +8188,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9235,22 +8215,21 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9264,13 +8243,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -9286,14 +8261,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9324,7 +8291,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9337,17 +8304,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9366,22 +8331,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9395,13 +8359,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -9415,14 +8375,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9453,7 +8405,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9463,17 +8415,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -9487,7 +8437,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -9506,22 +8455,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9533,13 +8481,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -9553,7 +8497,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9572,22 +8515,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9599,13 +8541,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9634,7 +8572,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -9658,7 +8595,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -9666,12 +8603,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9682,14 +8618,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -9699,11 +8635,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9715,10 +8651,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -9736,22 +8668,21 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9765,13 +8696,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -9787,14 +8714,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9825,7 +8744,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9838,17 +8757,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9867,22 +8784,21 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9896,13 +8812,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -9916,14 +8828,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9954,7 +8858,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9967,17 +8871,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -9996,22 +8898,21 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10025,13 +8926,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -10045,14 +8942,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10083,7 +8972,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10096,17 +8985,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -10130,7 +9017,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -10155,7 +9042,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -10163,16 +9050,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10192,10 +9078,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -10219,14 +9101,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10237,7 +9111,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10251,18 +9125,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10270,7 +9136,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -10281,19 +9147,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2017_03_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2017_03_30.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10324,7 +9179,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10334,20 +9189,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -10372,7 +9225,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -10380,16 +9233,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10409,10 +9261,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -10436,14 +9284,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10454,7 +9294,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10468,18 +9308,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10487,7 +9319,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -10498,19 +9330,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2017_03_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2017_03_30.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10541,7 +9362,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10551,17 +9372,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -10573,7 +9392,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -10592,21 +9410,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10618,13 +9435,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -10643,21 +9456,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10671,13 +9483,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -10691,14 +9499,6 @@ async def begin_delete( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10728,7 +9528,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10738,17 +9538,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -10756,7 +9554,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10778,16 +9575,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10798,14 +9594,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -10815,11 +9611,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10831,15 +9627,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -10861,15 +9652,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10880,14 +9670,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -10897,11 +9687,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10913,13 +9703,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -10945,7 +9733,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -10953,16 +9741,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10976,13 +9763,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -11008,14 +9791,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -11027,7 +9802,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11042,18 +9817,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -11065,7 +9832,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -11077,19 +9844,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -11121,7 +9877,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11131,17 +9887,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -11160,21 +9914,20 @@ async def _revoke_access_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11188,13 +9941,9 @@ async def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -11208,14 +9957,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11245,7 +9986,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11258,17 +9999,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SnapshotsOperations: @@ -11292,7 +10031,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -11317,7 +10056,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -11325,16 +10064,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11354,10 +10092,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -11381,14 +10115,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11400,7 +10126,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11414,18 +10140,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11434,7 +10152,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -11445,19 +10163,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11489,7 +10196,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11499,20 +10206,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -11537,7 +10246,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -11545,16 +10254,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11574,10 +10282,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -11602,14 +10306,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11621,7 +10317,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11636,18 +10332,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11656,7 +10344,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -11667,19 +10359,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11711,7 +10392,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11721,17 +10402,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -11743,7 +10422,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -11762,21 +10440,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11788,13 +10465,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -11813,21 +10486,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11841,13 +10513,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -11861,14 +10529,6 @@ async def begin_delete( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11898,7 +10558,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11908,17 +10568,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -11926,7 +10584,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -11949,16 +10606,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11969,14 +10625,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -11986,11 +10642,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12002,15 +10658,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -12033,15 +10684,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12052,14 +10702,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -12069,11 +10719,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12085,13 +10735,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -12117,7 +10765,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -12125,16 +10773,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12148,13 +10795,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -12180,14 +10823,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -12199,7 +10834,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12214,18 +10849,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -12237,7 +10864,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -12249,19 +10876,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] @@ -12293,7 +10909,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12303,17 +10919,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -12332,21 +10946,20 @@ async def _revoke_access_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12360,13 +10973,9 @@ async def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -12380,14 +10989,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12417,7 +11018,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12430,17 +11031,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -12469,7 +11068,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -12493,16 +11091,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12513,14 +11110,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -12530,11 +11127,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12546,10 +11143,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -12558,7 +11151,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -12577,21 +11169,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12603,10 +11194,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_compute_management_client_enums.py index 0ec781f2b09c..7990dddd8f72 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_compute_management_client_enums.py @@ -35,9 +35,8 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): EMPTY = "Empty" ATTACH = "Attach" FROM_IMAGE = "FromImage" - IMPORT = "Import" - COPY = "Copy" IMPORT_ENUM = "Import" + COPY = "Copy" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_models_py3.py index d312626cfa09..56d38e22e5ea 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -225,7 +221,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -283,7 +279,7 @@ class AvailabilitySet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -365,7 +361,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet] @@ -480,10 +476,10 @@ def __init__(self, *, output: Optional[JSON] = None, **kwargs: Any) -> None: class CreationData(_serialization.Model): """Data used when creating a disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Import". + values are: "Empty", "Attach", "FromImage", "Import", and "Copy". :vartype create_option: str or ~azure.mgmt.compute.v2017_03_30.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -523,7 +519,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", and "Copy". :paramtype create_option: str or ~azure.mgmt.compute.v2017_03_30.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in @@ -549,7 +545,7 @@ def __init__( class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -715,7 +711,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -912,7 +908,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.Disk] @@ -1107,7 +1103,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None" and "Read". :vartype access: str or ~azure.mgmt.compute.v2017_03_30.models.AccessLevel @@ -1142,7 +1138,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1183,7 +1179,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1224,7 +1220,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1291,7 +1287,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1383,7 +1379,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1421,7 +1417,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.Image] @@ -1455,7 +1451,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -1643,18 +1639,18 @@ def __init__( class ImageStorageProfile(_serialization.Model): """Describes a storage profile. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long Required. :vartype os_disk: ~azure.mgmt.compute.v2017_03_30.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.ImageDataDisk] """ @@ -1678,13 +1674,13 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long Required. :paramtype os_disk: ~azure.mgmt.compute.v2017_03_30.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.ImageDataDisk] """ super().__init__(**kwargs) @@ -1777,7 +1773,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2017_03_30.models.SourceVault @@ -1810,7 +1806,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2017_03_30.models.SourceVault @@ -1843,7 +1839,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -1876,7 +1872,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -1949,7 +1945,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.Usage] @@ -2197,7 +2193,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -2320,7 +2316,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -2353,7 +2349,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -2364,10 +2360,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2380,16 +2376,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2397,10 +2393,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2017_03_30.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2017_03_30.models.VaultSecretGroup] @@ -2434,7 +2430,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -2446,10 +2442,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2462,16 +2458,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2479,10 +2475,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2017_03_30.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -2553,7 +2549,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -2812,7 +2808,7 @@ def __init__(self, **kwargs: Any) -> None: class ResourceSkusResult(_serialization.Model): """The Compute List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the subscription. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.ResourceSku] @@ -3003,7 +2999,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3066,7 +3062,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3129,7 +3125,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3203,7 +3199,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -3250,7 +3246,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3283,7 +3279,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.RunCommandDocumentBase] @@ -3319,7 +3315,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3455,7 +3451,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3561,7 +3557,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.Snapshot] @@ -3701,7 +3697,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -3719,7 +3715,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -3738,12 +3734,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2017_03_30.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.DataDisk] """ @@ -3770,12 +3766,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2017_03_30.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.DataDisk] """ super().__init__(**kwargs) @@ -3885,7 +3881,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -4067,7 +4063,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4106,10 +4102,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2017_03_30.models.SubResource @@ -4123,7 +4119,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -4211,10 +4207,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2017_03_30.models.SubResource @@ -4224,7 +4220,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -4289,7 +4285,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -4366,7 +4362,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4484,7 +4480,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -4528,7 +4524,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4833,7 +4829,7 @@ def __init__(self, *, type: Optional[Literal["SystemAssigned"]] = None, **kwargs class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4889,7 +4885,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5056,7 +5052,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -5094,7 +5090,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5226,7 +5222,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -5404,10 +5400,10 @@ def __init__( self.provisioning_state = None -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension] @@ -5540,7 +5536,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5567,7 +5563,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5688,7 +5684,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet] @@ -5724,7 +5720,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetSku] @@ -5760,7 +5756,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet] @@ -5793,7 +5789,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -5821,10 +5817,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5904,7 +5900,7 @@ def __init__( self.ip_configurations = ip_configurations -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -5929,7 +5925,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2017_03_30.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -5954,7 +5950,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2017_03_30.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -5968,7 +5964,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -6079,10 +6075,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6095,16 +6091,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6112,10 +6108,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2017_03_30.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6158,10 +6154,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6174,16 +6170,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6191,10 +6187,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2017_03_30.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6210,10 +6206,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -6260,10 +6256,12 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -6374,12 +6372,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetDataDisk] """ @@ -6407,12 +6405,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetDataDisk] """ @@ -6507,7 +6505,7 @@ def __init__( self.single_placement_group = single_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -6613,7 +6611,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -6691,7 +6689,7 @@ def __init__( self.ip_configurations = ip_configurations -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar network_interface_configurations: The list of network configurations. @@ -6822,7 +6820,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -6866,7 +6864,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -6980,7 +6978,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7027,10 +7025,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2017_03_30.models.SubResource @@ -7042,7 +7040,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -7125,10 +7123,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2017_03_30.models.SubResource @@ -7138,7 +7136,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7160,7 +7158,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7213,10 +7211,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -7343,7 +7341,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVM] @@ -7402,7 +7400,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -7451,7 +7449,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/operations/_operations.py index 374b7d59fb8a..8e8415fbe7e0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_03_30/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -140,7 +135,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -196,7 +191,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -227,7 +222,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -260,7 +255,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -291,7 +286,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -337,7 +332,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -372,7 +367,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -407,7 +402,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -439,7 +434,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -479,7 +474,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_get_extensions_request( +def build_virtual_machines_get_extensions_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -512,7 +507,9 @@ def build_virtual_machines_get_extensions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -574,7 +571,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -677,7 +674,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -708,7 +705,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -739,7 +736,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -770,7 +767,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -853,7 +850,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1008,7 +1005,7 @@ def build_virtual_machines_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1039,7 +1036,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1107,7 +1104,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1155,7 +1152,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1186,7 +1183,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1215,7 +1212,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1399,7 +1396,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1477,7 +1474,7 @@ def build_resource_skus_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1511,7 +1508,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1545,7 +1542,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1576,7 +1573,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1607,7 +1604,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1641,7 +1638,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1675,7 +1672,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1706,7 +1703,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1736,7 +1733,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1762,7 +1761,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1793,7 +1792,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1827,7 +1826,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1861,7 +1860,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1895,7 +1894,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1929,7 +1928,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1963,7 +1962,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1997,7 +1996,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2032,7 +2031,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2064,7 +2063,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2104,7 +2103,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2135,7 +2134,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2166,7 +2165,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2197,7 +2196,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2228,7 +2227,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2260,7 +2259,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2292,7 +2291,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2324,7 +2323,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2356,7 +2355,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2388,7 +2387,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2420,7 +2419,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2466,7 +2465,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2498,7 +2497,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2530,7 +2529,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2692,7 +2691,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2941,7 +2940,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3060,7 +3059,9 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3087,7 +3088,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3159,7 +3160,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3170,7 +3170,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3182,11 +3182,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3197,7 +3196,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -3207,12 +3206,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3240,7 +3235,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3248,16 +3243,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3269,13 +3263,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( @@ -3287,7 +3277,6 @@ def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -3306,21 +3295,20 @@ def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3334,13 +3322,9 @@ def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -3350,7 +3334,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3369,21 +3352,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3395,13 +3377,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -3412,7 +3390,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3434,16 +3411,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3454,14 +3430,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3471,11 +3447,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3487,17 +3463,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3519,16 +3490,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3539,14 +3509,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3556,11 +3526,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3572,10 +3542,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -3587,7 +3553,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -3609,17 +3574,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3630,14 +3594,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3647,11 +3611,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3663,10 +3627,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -3702,7 +3662,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3721,23 +3680,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3749,13 +3707,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -3767,7 +3721,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3786,21 +3739,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3812,13 +3764,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -3846,7 +3794,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3865,7 +3812,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3874,16 +3821,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3895,13 +3841,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3929,7 +3871,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3955,7 +3897,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3964,16 +3906,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3993,10 +3934,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -4023,14 +3960,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4044,7 +3973,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4060,18 +3989,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4085,7 +4006,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4098,20 +4019,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4145,7 +4055,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4155,24 +4065,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4198,7 +4106,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4207,16 +4115,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4228,13 +4135,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4263,14 +4166,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4284,7 +4179,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4300,18 +4195,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4325,7 +4212,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4338,20 +4225,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4385,7 +4261,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4395,17 +4271,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4424,22 +4298,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4453,13 +4326,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -4474,14 +4343,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4512,7 +4373,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4522,17 +4383,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4554,7 +4413,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4573,23 +4431,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4601,13 +4458,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -4642,7 +4495,6 @@ def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4661,22 +4513,21 @@ def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4688,13 +4539,9 @@ def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore @distributed_trace def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -4703,7 +4550,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4725,16 +4571,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4745,14 +4590,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4762,11 +4607,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4778,15 +4623,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -4812,7 +4653,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4820,16 +4661,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4843,13 +4683,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -4873,14 +4709,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4893,7 +4721,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4906,18 +4734,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4930,7 +4750,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -4941,19 +4761,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4986,7 +4796,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4998,20 +4808,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5036,7 +4848,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5044,16 +4856,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5073,10 +4884,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -5098,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -5117,7 +4916,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5129,18 +4928,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -5149,7 +4940,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -5158,19 +4953,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] @@ -5202,7 +4986,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5212,17 +4996,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5241,21 +5023,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5269,13 +5050,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -5287,14 +5064,6 @@ def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5324,7 +5093,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5334,17 +5103,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5359,7 +5126,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -5378,22 +5144,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5405,13 +5170,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -5423,7 +5184,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,21 +5202,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5468,13 +5227,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5493,21 +5248,20 @@ def _convert_to_managed_disks_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5521,13 +5275,9 @@ def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace def begin_convert_to_managed_disks( @@ -5540,14 +5290,6 @@ def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5577,7 +5319,7 @@ def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5589,17 +5331,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5618,21 +5358,20 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5646,13 +5385,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -5665,14 +5400,6 @@ def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5702,7 +5429,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5714,17 +5441,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _models.OperationStatusResponse: @@ -5734,7 +5459,6 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -5753,21 +5477,20 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5779,13 +5502,9 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -5794,7 +5513,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -5816,16 +5534,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5836,14 +5553,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5853,11 +5570,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5869,16 +5586,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -5900,15 +5612,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5919,14 +5630,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5936,11 +5647,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5952,8 +5663,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5965,7 +5674,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -5987,17 +5695,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6008,14 +5715,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6025,11 +5732,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6041,10 +5748,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -6062,21 +5765,20 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6090,13 +5792,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -6109,14 +5807,6 @@ def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6146,7 +5836,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6158,17 +5848,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6187,21 +5875,20 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6215,13 +5902,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -6233,14 +5916,6 @@ def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6270,7 +5945,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6282,17 +5957,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6311,21 +5984,20 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6339,13 +6011,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -6357,14 +6025,6 @@ def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6394,7 +6054,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6406,17 +6066,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6435,21 +6093,20 @@ def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6463,13 +6120,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -6481,14 +6134,6 @@ def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6518,7 +6163,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6530,17 +6175,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _perform_maintenance_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6559,21 +6202,20 @@ def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6587,13 +6229,9 @@ def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace def begin_perform_maintenance( @@ -6605,14 +6243,6 @@ def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6642,7 +6272,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6654,20 +6284,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -6692,7 +6324,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6700,16 +6332,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6723,13 +6354,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -6752,14 +6379,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.RunCommandResult] @@ -6771,7 +6390,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6783,18 +6402,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.RunCommandResult] @@ -6803,7 +6414,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -6812,19 +6427,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.RunCommandResult] @@ -6856,7 +6460,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6868,17 +6472,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -6917,7 +6519,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6936,7 +6537,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6944,16 +6545,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6965,13 +6565,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7003,7 +6599,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7022,7 +6617,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -7032,16 +6627,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7053,13 +6647,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -7071,7 +6661,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7090,21 +6679,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7116,13 +6704,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -7130,7 +6714,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7149,20 +6732,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7174,13 +6756,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -7194,7 +6772,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7213,22 +6790,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7240,13 +6816,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -7276,7 +6848,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7298,16 +6869,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7318,14 +6888,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -7335,11 +6905,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7351,8 +6921,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -7380,7 +6948,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7402,16 +6969,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7422,14 +6988,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7439,11 +7005,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7455,8 +7021,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -7479,7 +7043,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7504,7 +7068,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7512,16 +7076,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7541,10 +7104,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -7566,14 +7125,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7584,7 +7135,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7596,18 +7147,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7615,7 +7158,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -7624,19 +7167,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7667,7 +7199,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7677,17 +7209,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -7706,21 +7236,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7734,13 +7263,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -7752,14 +7277,6 @@ def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7789,7 +7306,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7799,17 +7316,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -7823,7 +7338,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -7842,22 +7356,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7869,13 +7382,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -7883,7 +7392,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7905,16 +7413,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7925,14 +7432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7942,11 +7449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7958,16 +7465,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7989,15 +7491,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8008,14 +7509,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8025,11 +7526,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8041,8 +7542,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class ResourceSkusOperations: """ @@ -8068,7 +7567,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: @@ -8090,15 +7588,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.ResourceSku"]: def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8109,14 +7606,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -8126,11 +7623,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8142,8 +7639,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -8169,7 +7664,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8195,7 +7690,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8203,16 +7698,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8232,10 +7726,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -8257,14 +7747,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8277,7 +7759,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8289,18 +7771,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8313,7 +7787,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8322,20 +7796,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8368,7 +7831,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8378,23 +7841,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8420,7 +7881,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8428,16 +7889,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8449,13 +7909,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8478,14 +7934,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8498,7 +7946,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8510,18 +7958,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8534,7 +7974,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -8543,20 +7983,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8589,7 +8019,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8599,17 +8029,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8628,21 +8056,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8656,13 +8083,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -8674,14 +8097,6 @@ def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8711,7 +8126,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8721,17 +8136,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -8741,7 +8154,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -8760,21 +8172,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8786,19 +8197,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8827,7 +8234,7 @@ def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8835,16 +8242,15 @@ def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8858,13 +8264,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload def begin_deallocate( @@ -8891,14 +8293,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8911,7 +8305,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8926,18 +8320,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8950,7 +8336,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -8962,20 +8348,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9008,7 +8383,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9020,23 +8395,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9062,7 +8435,7 @@ def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9070,16 +8443,15 @@ def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9093,13 +8465,9 @@ def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload def begin_delete_instances( @@ -9124,14 +8492,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9144,7 +8504,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9157,18 +8517,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9181,7 +8533,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -9191,20 +8543,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9237,7 +8578,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9249,17 +8590,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_instance_view( @@ -9271,7 +8610,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9290,21 +8628,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9316,13 +8653,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -9330,7 +8663,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9354,16 +8686,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9374,14 +8705,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -9391,11 +8722,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9407,17 +8738,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9441,15 +8767,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9460,14 +8785,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -9477,11 +8802,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9493,8 +8818,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9506,7 +8829,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -9530,17 +8852,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9551,14 +8872,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -9568,11 +8889,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9584,15 +8905,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9621,7 +8938,7 @@ def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9629,16 +8946,15 @@ def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9652,13 +8968,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload def begin_power_off( @@ -9685,14 +8997,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9705,7 +9009,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9720,18 +9024,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9744,7 +9040,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -9756,20 +9052,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9802,7 +9087,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9814,23 +9099,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9859,7 +9142,7 @@ def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9867,16 +9150,15 @@ def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9890,13 +9172,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload def begin_restart( @@ -9921,14 +9199,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9941,7 +9211,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9954,18 +9224,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9978,7 +9240,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -9988,20 +9250,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10034,7 +9285,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10046,23 +9297,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10091,7 +9340,7 @@ def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10099,16 +9348,15 @@ def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10122,13 +9370,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload def begin_start( @@ -10153,14 +9397,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10173,7 +9409,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10186,18 +9422,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10210,7 +9438,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -10220,20 +9448,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10266,7 +9483,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10278,23 +9495,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10320,7 +9535,7 @@ def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10328,16 +9543,15 @@ def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10351,13 +9565,9 @@ def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload def begin_update_instances( @@ -10382,14 +9592,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10402,7 +9604,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10415,18 +9617,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10439,7 +9633,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -10449,20 +9643,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10495,7 +9678,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10507,23 +9690,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10552,7 +9733,7 @@ def _reimage_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10560,16 +9741,15 @@ def _reimage_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10583,13 +9763,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @overload def begin_reimage( @@ -10614,14 +9790,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10634,7 +9802,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10647,18 +9815,10 @@ def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10671,7 +9831,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -10681,20 +9841,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10727,7 +9876,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10739,23 +9888,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10784,7 +9931,7 @@ def _reimage_all_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10792,16 +9939,15 @@ def _reimage_all_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10815,13 +9961,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @overload def begin_reimage_all( @@ -10847,14 +9989,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10867,7 +10001,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10881,18 +10015,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10905,7 +10031,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -10916,20 +10042,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10962,7 +10077,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10974,20 +10089,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11012,7 +10125,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -11038,7 +10151,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -11047,16 +10160,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11076,10 +10188,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11107,14 +10215,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11128,7 +10228,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11144,18 +10244,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11169,7 +10261,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -11182,20 +10274,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11229,7 +10310,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11239,17 +10320,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -11268,22 +10347,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11297,13 +10375,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -11318,14 +10392,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11356,7 +10422,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11366,17 +10432,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -11398,7 +10462,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11417,23 +10480,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11445,13 +10507,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11463,7 +10521,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11487,17 +10544,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11508,14 +10564,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -11525,11 +10581,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11541,12 +10597,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11583,21 +10635,20 @@ def _cancel_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11611,13 +10662,9 @@ def _cancel_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return deserialized # type: ignore @distributed_trace def begin_cancel( @@ -11629,14 +10676,6 @@ def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11666,7 +10705,7 @@ def begin_cancel( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11678,17 +10717,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_os_upgrade_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11707,21 +10744,20 @@ def _start_os_upgrade_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11735,13 +10771,9 @@ def _start_os_upgrade_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return deserialized # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11755,14 +10787,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11792,7 +10816,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11804,17 +10828,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_latest( @@ -11826,7 +10848,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11845,21 +10866,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11871,13 +10891,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: @@ -11917,22 +10933,21 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11946,13 +10961,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -11966,14 +10977,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12004,7 +11007,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12016,17 +11019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12045,22 +11046,21 @@ def _reimage_all_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12074,13 +11074,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace def begin_reimage_all( @@ -12095,14 +11091,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12133,7 +11121,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12145,17 +11133,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12174,22 +11160,21 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12203,13 +11188,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -12225,14 +11206,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12263,7 +11236,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12275,17 +11248,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12304,22 +11275,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12333,13 +11303,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -12353,14 +11319,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12391,7 +11349,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12401,17 +11359,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -12425,7 +11381,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -12444,22 +11399,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12471,13 +11425,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -12491,7 +11441,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -12510,22 +11459,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12537,13 +11485,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12572,7 +11516,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12596,7 +11539,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -12604,12 +11547,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12620,14 +11562,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -12637,11 +11579,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12653,10 +11595,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -12674,22 +11612,21 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12703,13 +11640,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -12725,14 +11658,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12763,7 +11688,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12775,17 +11700,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12804,22 +11727,21 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12833,13 +11755,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -12853,14 +11771,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12891,7 +11801,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12903,17 +11813,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12932,22 +11840,21 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12961,13 +11868,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -12981,14 +11884,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13019,7 +11914,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13031,17 +11926,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -13065,7 +11958,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -13090,7 +11983,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -13098,16 +11991,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13127,10 +12019,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -13154,14 +12042,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13172,7 +12052,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13186,18 +12066,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13205,7 +12077,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -13216,19 +12088,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2017_03_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2017_03_30.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13259,7 +12120,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13269,20 +12130,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -13307,7 +12166,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -13315,16 +12174,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13344,10 +12202,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -13371,14 +12225,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13389,7 +12235,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13403,18 +12249,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13422,7 +12260,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -13433,19 +12271,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2017_03_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2017_03_30.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13476,7 +12303,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13486,17 +12313,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -13508,7 +12333,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -13527,21 +12351,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13553,13 +12376,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -13578,21 +12397,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13606,13 +12424,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -13626,14 +12440,6 @@ def begin_delete( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13663,7 +12469,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13673,17 +12479,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -13691,7 +12495,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13713,16 +12516,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13733,14 +12535,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -13750,11 +12552,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13766,15 +12568,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13796,15 +12593,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13815,14 +12611,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -13832,11 +12628,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13848,13 +12644,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -13880,7 +12674,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -13888,16 +12682,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13911,13 +12704,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -13943,14 +12732,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -13961,7 +12742,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13976,18 +12757,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -13998,7 +12771,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -14010,19 +12783,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -14053,7 +12815,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14063,17 +12825,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -14092,21 +12852,20 @@ def _revoke_access_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14120,13 +12879,9 @@ def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace def begin_revoke_access( @@ -14140,14 +12895,6 @@ def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14177,7 +12924,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14189,17 +12936,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SnapshotsOperations: @@ -14223,7 +12968,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -14248,7 +12993,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -14256,16 +13001,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14285,10 +13029,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -14312,14 +13052,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14330,7 +13062,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14344,18 +13076,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14363,7 +13087,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -14374,19 +13098,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14417,7 +13130,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14427,20 +13140,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -14465,7 +13180,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -14473,16 +13188,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14502,10 +13216,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -14530,14 +13240,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14548,7 +13250,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14563,18 +13265,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14582,7 +13276,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -14593,19 +13291,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2017_03_30.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14636,7 +13323,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14646,17 +13333,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -14668,7 +13353,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -14687,21 +13371,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14713,13 +13396,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -14738,21 +13417,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14766,13 +13444,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -14786,14 +13460,6 @@ def begin_delete( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14823,7 +13489,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14833,17 +13499,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -14851,7 +13515,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14873,16 +13536,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14893,14 +13555,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -14910,11 +13572,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14926,15 +13588,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_03_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -14956,15 +13613,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14975,14 +13631,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -14992,11 +13648,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15008,13 +13664,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -15040,7 +13694,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -15048,16 +13702,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15071,13 +13724,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -15103,14 +13752,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -15121,7 +13762,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15136,18 +13777,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -15158,7 +13791,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -15170,19 +13803,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2017_03_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_03_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -15213,7 +13835,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15223,17 +13845,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -15252,21 +13872,20 @@ def _revoke_access_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15280,13 +13899,9 @@ def _revoke_access_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return deserialized # type: ignore @distributed_trace def begin_revoke_access( @@ -15300,14 +13915,6 @@ def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -15337,7 +13944,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15349,17 +13956,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -15388,7 +13993,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -15412,16 +14016,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15432,14 +14035,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -15449,11 +14052,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15465,10 +14068,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -15477,7 +14076,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_03_30.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -15496,21 +14094,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-03-30")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15522,10 +14119,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_compute_management_client.py index bf75b1b1cc8b..3df50358aaaa 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -49,7 +51,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -59,7 +79,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-09-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -79,7 +99,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_configuration.py index 9d654bd10166..260ae128b160 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-09-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_metadata.json index 5704819394d0..c81a5821f654 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": false, + "has_public_lro_operations": false, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "resource_skus": "ResourceSkusOperations" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_compute_management_client.py index 2aac2c58a958..310d07c84775 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -49,7 +51,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -59,7 +79,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-09-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -79,7 +101,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_configuration.py index f5ca04d3e724..35dca4c89496 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-09-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/operations/_operations.py index 1a5135aff291..4dd7b9161fb2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -57,7 +57,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_09_01.models.ResourceSku] @@ -80,15 +79,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ResourceSku"]: def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -99,14 +97,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -116,11 +114,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -131,5 +129,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/models/_models_py3.py index 57df16b38760..fb91b105180c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/models/_models_py3.py @@ -303,7 +303,7 @@ def __init__(self, **kwargs: Any) -> None: class ResourceSkusResult(_serialization.Model): """The List Resource Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the subscription. Required. :vartype value: list[~azure.mgmt.compute.v2017_09_01.models.ResourceSku] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/operations/_operations.py index 86e19019b4c4..8a4f416f64ff 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_09_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -84,7 +84,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ResourceSku"]: """Gets the list of Microsoft.Compute SKUs available for your Subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_09_01.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: @@ -106,15 +105,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.ResourceSku"]: def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -125,14 +123,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -142,11 +140,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -157,5 +155,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_compute_management_client.py index 491dfb0053ae..a5b8b8546bb3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -107,7 +109,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -153,7 +173,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -173,7 +193,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_configuration.py index 73b259e4ddde..2a4659eb01f3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_metadata.json index dba50ebbc548..05af5c4b12f7 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_compute_management_client.py index 20ea38b98223..ea79be143654 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -108,7 +110,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -154,7 +174,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2017-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -174,7 +196,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_configuration.py index 006891cc34bc..8fd504c7b379 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2017-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/operations/_operations.py index 6714e9cce5b6..4d17d6ac5668 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -127,10 +126,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -159,7 +154,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -183,14 +177,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -201,14 +194,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -218,11 +211,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -234,8 +227,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -278,7 +269,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -289,7 +279,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -301,11 +291,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -316,7 +305,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -326,12 +315,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -359,7 +344,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -367,16 +352,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -388,13 +372,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -417,7 +397,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -428,7 +407,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -440,11 +419,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -455,7 +433,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -465,12 +443,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -498,7 +472,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -506,16 +480,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -527,13 +500,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( @@ -545,7 +514,6 @@ async def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -564,21 +532,20 @@ async def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -592,13 +559,9 @@ async def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -608,7 +571,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -627,21 +589,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -653,13 +614,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -670,7 +627,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet] @@ -693,16 +649,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -713,14 +668,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -730,11 +685,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -746,17 +701,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet] @@ -779,16 +729,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -799,14 +748,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -816,11 +765,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -832,10 +781,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -847,7 +792,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] @@ -870,17 +814,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -891,14 +834,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -908,11 +851,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -924,10 +867,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -963,7 +902,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -982,23 +920,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1010,13 +947,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -1028,7 +961,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1047,21 +979,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1073,13 +1004,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -1107,7 +1034,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1126,7 +1052,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -1135,16 +1061,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1156,13 +1081,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -1190,7 +1111,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1216,7 +1137,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1225,16 +1146,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1254,10 +1174,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1284,14 +1200,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1305,7 +1213,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1321,18 +1229,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1346,7 +1246,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1359,20 +1259,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1406,7 +1295,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1416,24 +1305,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1459,7 +1346,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1468,16 +1355,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1489,13 +1375,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1524,14 +1406,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1545,7 +1419,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1561,18 +1435,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1586,7 +1452,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -1599,20 +1465,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1646,7 +1501,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1656,17 +1511,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -1685,22 +1538,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1714,13 +1566,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -1735,14 +1583,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -1773,7 +1613,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1783,17 +1623,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1815,7 +1653,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -1834,23 +1671,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1862,13 +1698,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -1903,7 +1735,6 @@ async def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -1922,22 +1753,21 @@ async def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1949,13 +1779,9 @@ async def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore @distributed_trace def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -1964,7 +1790,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -1987,16 +1812,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2007,14 +1831,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -2024,11 +1848,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2040,15 +1864,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -2074,7 +1894,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2082,16 +1902,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2105,13 +1924,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -2135,14 +1950,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2155,7 +1962,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2168,18 +1975,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2192,7 +1991,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -2203,19 +2002,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -2248,7 +2037,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,20 +2050,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -2299,7 +2090,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2307,16 +2098,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2336,10 +2126,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -2361,14 +2147,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2381,7 +2159,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2393,18 +2171,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2414,7 +2184,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -2423,19 +2197,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2468,7 +2231,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2478,20 +2241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -2516,7 +2281,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -2524,16 +2289,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2553,10 +2317,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -2578,14 +2338,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2598,7 +2350,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2610,18 +2362,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2631,7 +2375,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -2640,19 +2388,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -2685,7 +2422,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2695,17 +2432,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2724,21 +2459,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2752,13 +2486,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -2770,14 +2500,6 @@ async def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -2807,7 +2529,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2817,17 +2539,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2842,7 +2562,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -2861,22 +2580,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2888,13 +2606,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -2906,7 +2620,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -2925,21 +2638,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2951,13 +2663,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -2976,21 +2684,20 @@ async def _convert_to_managed_disks_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3004,13 +2711,9 @@ async def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -3023,14 +2726,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3060,7 +2755,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3073,17 +2768,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3102,21 +2795,20 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3130,13 +2822,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -3149,14 +2837,6 @@ async def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3186,7 +2866,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3199,17 +2879,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def generalize( @@ -3221,7 +2899,6 @@ async def generalize( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -3240,21 +2917,20 @@ async def generalize( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3266,13 +2942,9 @@ async def generalize( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -3281,7 +2953,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -3304,16 +2975,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3324,14 +2994,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3341,11 +3011,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3357,16 +3027,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -3389,15 +3054,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3408,14 +3072,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3425,11 +3089,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3441,8 +3105,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3454,7 +3116,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] @@ -3477,17 +3138,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3498,14 +3158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3515,11 +3175,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3531,10 +3191,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -3552,21 +3208,20 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3580,13 +3235,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -3599,14 +3250,6 @@ async def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3636,7 +3279,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3649,17 +3292,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3678,21 +3319,20 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3706,13 +3346,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -3724,14 +3360,6 @@ async def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3761,7 +3389,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3774,17 +3402,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3803,21 +3429,20 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3831,13 +3456,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -3849,14 +3470,6 @@ async def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -3886,7 +3499,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3899,17 +3512,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3928,21 +3539,20 @@ async def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3956,13 +3566,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -3974,14 +3580,6 @@ async def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4011,7 +3609,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4024,17 +3622,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _perform_maintenance_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4053,21 +3649,20 @@ async def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4081,13 +3676,9 @@ async def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -4099,14 +3690,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4136,7 +3719,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4149,20 +3732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -4187,7 +3772,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4195,16 +3780,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4218,13 +3802,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -4247,14 +3827,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -4267,7 +3839,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4279,18 +3851,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -4300,7 +3864,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -4309,19 +3877,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -4354,7 +3911,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4367,17 +3924,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -4416,7 +3971,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4435,7 +3989,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4443,16 +3997,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4464,13 +4017,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4502,7 +4051,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4521,7 +4069,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4531,16 +4079,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4552,13 +4099,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4570,7 +4113,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,21 +4131,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4615,13 +4156,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4629,7 +4166,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4648,20 +4184,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4673,13 +4208,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4693,7 +4224,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4712,22 +4242,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4739,13 +4268,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -4775,7 +4300,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4797,16 +4321,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4817,14 +4340,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -4834,11 +4357,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4850,8 +4373,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -4879,7 +4400,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] @@ -4902,16 +4422,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4922,14 +4441,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4939,11 +4458,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4955,8 +4474,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -4979,7 +4496,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5004,7 +4521,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5012,16 +4529,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5041,10 +4557,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -5066,14 +4578,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5084,7 +4588,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5096,18 +4600,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5115,7 +4611,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -5124,19 +4620,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5167,7 +4652,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5177,20 +4662,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5215,7 +4702,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5223,16 +4710,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5252,10 +4738,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -5277,14 +4759,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5295,7 +4769,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5307,18 +4781,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5326,7 +4792,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -5335,19 +4805,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5378,7 +4837,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5388,17 +4847,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -5417,21 +4874,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5445,13 +4901,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -5463,14 +4915,6 @@ async def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5500,7 +4944,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5510,17 +4954,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -5534,7 +4976,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -5553,22 +4994,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5580,13 +5020,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -5594,7 +5030,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5616,16 +5051,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5636,14 +5070,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -5653,11 +5087,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5669,16 +5103,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5700,15 +5129,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5719,14 +5147,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -5736,11 +5164,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5752,8 +5180,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -5779,7 +5205,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -5805,7 +5231,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -5813,16 +5239,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5842,10 +5267,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -5867,14 +5288,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5887,7 +5300,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5899,18 +5312,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5923,7 +5328,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -5932,20 +5337,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -5978,7 +5372,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5988,23 +5382,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6030,7 +5422,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6038,16 +5430,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6059,13 +5450,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6088,14 +5475,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6108,7 +5487,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6120,18 +5499,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6144,7 +5515,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6153,20 +5524,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6199,7 +5560,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6209,17 +5570,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6238,21 +5597,20 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6266,13 +5624,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -6284,14 +5638,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6321,7 +5667,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6331,17 +5677,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -6353,7 +5697,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6372,21 +5715,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6398,19 +5740,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6439,7 +5777,7 @@ async def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6447,16 +5785,15 @@ async def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6470,13 +5807,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload async def begin_deallocate( @@ -6503,14 +5836,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6523,7 +5848,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6538,18 +5863,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6562,7 +5879,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -6574,20 +5891,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6620,7 +5926,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6633,23 +5939,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -6675,7 +5979,7 @@ async def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6683,16 +5987,15 @@ async def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6706,13 +6009,9 @@ async def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload async def begin_delete_instances( @@ -6737,14 +6036,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6757,7 +6048,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6770,18 +6061,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6794,7 +6077,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -6804,20 +6087,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6850,7 +6122,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6863,17 +6135,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_instance_view( @@ -6885,7 +6155,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6904,21 +6173,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6930,13 +6198,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -6944,7 +6208,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6968,16 +6231,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6988,14 +6250,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7005,11 +6267,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7021,17 +6283,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7055,15 +6312,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7074,14 +6330,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7091,11 +6347,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7107,8 +6363,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7120,7 +6374,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7144,17 +6397,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7165,14 +6417,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7182,11 +6434,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7198,10 +6450,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7212,7 +6460,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7236,17 +6483,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7257,14 +6503,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7274,11 +6520,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7290,15 +6536,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7327,7 +6569,7 @@ async def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7335,16 +6577,15 @@ async def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7358,13 +6599,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload async def begin_power_off( @@ -7391,14 +6628,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7411,7 +6640,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7426,18 +6655,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7450,7 +6671,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -7462,20 +6683,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7508,7 +6718,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7521,23 +6731,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7566,7 +6774,7 @@ async def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7574,16 +6782,15 @@ async def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7597,13 +6804,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload async def begin_restart( @@ -7628,14 +6831,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7648,7 +6843,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7661,18 +6856,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7685,7 +6872,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -7695,20 +6882,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7741,7 +6917,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7754,23 +6930,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -7799,7 +6973,7 @@ async def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7807,16 +6981,15 @@ async def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7830,13 +7003,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload async def begin_start( @@ -7861,14 +7030,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7881,7 +7042,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7894,18 +7055,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7918,7 +7071,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -7928,20 +7081,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -7974,7 +7116,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7987,23 +7129,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8032,7 +7172,7 @@ async def _redeploy_initial( else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8040,16 +7180,15 @@ async def _redeploy_initial( content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8063,13 +7202,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return deserialized # type: ignore @overload async def begin_redeploy( @@ -8095,14 +7230,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8115,7 +7242,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8129,18 +7256,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8153,7 +7272,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8164,20 +7283,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8210,7 +7318,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8223,23 +7331,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _perform_maintenance_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8268,7 +7374,7 @@ async def _perform_maintenance_initial( else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8276,16 +7382,15 @@ async def _perform_maintenance_initial( content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8299,13 +7404,9 @@ async def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return deserialized # type: ignore @overload async def begin_perform_maintenance( @@ -8333,14 +7434,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8353,7 +7446,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8369,18 +7462,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8393,7 +7478,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8406,20 +7491,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8452,7 +7526,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8465,23 +7539,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8507,7 +7579,7 @@ async def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8515,16 +7587,15 @@ async def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8538,13 +7609,9 @@ async def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload async def begin_update_instances( @@ -8569,14 +7636,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8589,7 +7648,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8602,18 +7661,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8626,7 +7677,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -8636,20 +7687,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8682,7 +7722,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8695,23 +7735,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8740,7 +7778,7 @@ async def _reimage_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8748,16 +7786,15 @@ async def _reimage_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8771,13 +7808,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @overload async def begin_reimage( @@ -8802,14 +7835,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8822,7 +7847,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8835,18 +7860,10 @@ async def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8859,7 +7876,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -8869,20 +7886,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8915,7 +7921,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8928,23 +7934,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -8973,7 +7977,7 @@ async def _reimage_all_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8981,16 +7985,15 @@ async def _reimage_all_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9004,13 +8007,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @overload async def begin_reimage_all( @@ -9036,14 +8035,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9056,7 +8047,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9070,18 +8061,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9094,7 +8077,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.OperationStatusResponse]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9105,20 +8088,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9151,7 +8123,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9164,20 +8136,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -9190,7 +8160,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9209,22 +8178,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9236,16 +8204,12 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -9270,7 +8234,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -9296,7 +8260,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -9305,16 +8269,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9334,10 +8297,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9365,14 +8324,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9386,7 +8337,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9402,18 +8353,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9427,7 +8370,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -9440,20 +8383,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9487,7 +8419,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9497,17 +8429,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -9526,22 +8456,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9555,13 +8484,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -9576,14 +8501,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9614,7 +8531,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9624,17 +8541,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -9656,7 +8571,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9675,23 +8589,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9703,13 +8616,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9721,7 +8630,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9745,17 +8653,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9766,14 +8673,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -9783,11 +8690,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9799,12 +8706,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -9841,21 +8744,20 @@ async def _cancel_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9869,13 +8771,9 @@ async def _cancel_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return deserialized # type: ignore @distributed_trace_async async def begin_cancel( @@ -9887,14 +8785,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9924,7 +8814,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9937,17 +8827,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_os_upgrade_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9966,21 +8854,20 @@ async def _start_os_upgrade_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9994,13 +8881,9 @@ async def _start_os_upgrade_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -10014,14 +8897,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10051,7 +8926,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10064,17 +8939,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_latest( @@ -10086,7 +8959,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -10105,21 +8977,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10131,13 +9002,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -10177,22 +9044,21 @@ async def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10206,13 +9072,9 @@ async def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage( @@ -10226,14 +9088,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10264,7 +9118,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10277,17 +9131,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10306,22 +9158,21 @@ async def _reimage_all_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10335,13 +9186,9 @@ async def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -10356,14 +9203,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10394,7 +9233,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10407,17 +9246,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10436,22 +9273,21 @@ async def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10465,13 +9301,9 @@ async def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10487,14 +9319,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10525,7 +9349,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10538,24 +9362,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -10581,7 +9403,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -10590,16 +9412,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10619,10 +9440,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -10649,14 +9466,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10670,7 +9479,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10686,18 +9495,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10711,7 +9512,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -10724,19 +9525,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10770,7 +9560,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10780,17 +9570,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10809,22 +9597,21 @@ async def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10838,13 +9625,9 @@ async def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def begin_delete( @@ -10858,14 +9641,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10896,7 +9671,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10906,17 +9681,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -10930,7 +9703,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -10949,22 +9721,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10976,13 +9747,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -10996,7 +9763,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11015,22 +9781,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11042,13 +9807,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11077,7 +9838,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11101,7 +9861,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -11109,12 +9869,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11125,14 +9884,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -11142,11 +9901,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11158,10 +9917,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -11179,22 +9934,21 @@ async def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11208,13 +9962,9 @@ async def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace_async async def begin_power_off( @@ -11230,14 +9980,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11268,7 +10010,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11281,17 +10023,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11310,22 +10050,21 @@ async def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11339,13 +10078,9 @@ async def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace_async async def begin_restart( @@ -11359,14 +10094,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11397,7 +10124,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11410,17 +10137,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11439,22 +10164,21 @@ async def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11468,13 +10192,9 @@ async def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace_async async def begin_start( @@ -11488,14 +10208,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11526,7 +10238,7 @@ async def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11539,17 +10251,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _redeploy_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11568,22 +10278,21 @@ async def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11597,13 +10306,9 @@ async def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return deserialized # type: ignore @distributed_trace_async async def begin_redeploy( @@ -11618,14 +10323,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11656,7 +10353,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11669,17 +10366,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _perform_maintenance_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11698,22 +10393,21 @@ async def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11727,13 +10421,9 @@ async def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11747,14 +10437,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11785,7 +10467,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11798,17 +10480,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -11832,7 +10512,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -11857,23 +10537,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11887,13 +10566,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -11915,14 +10590,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -11932,7 +10599,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -11941,18 +10608,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -11962,7 +10621,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -11970,19 +10629,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12014,7 +10663,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12027,20 +10676,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12065,23 +10712,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12095,13 +10741,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -12122,14 +10764,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12139,7 +10773,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12147,18 +10781,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12168,7 +10794,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12176,19 +10802,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12220,7 +10835,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12233,17 +10848,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -12272,7 +10885,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -12296,16 +10908,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12316,14 +10927,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -12333,11 +10944,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12349,10 +10960,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -12361,7 +10968,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -12380,21 +10986,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12406,10 +11011,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/models/_models_py3.py index d709b1bbf59f..64deaa788a08 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -224,7 +220,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -282,7 +278,7 @@ class AvailabilitySet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -364,7 +360,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet] @@ -634,7 +630,7 @@ def __init__(self, **kwargs: Any) -> None: class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -893,7 +889,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -950,7 +946,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1007,7 +1003,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1074,7 +1070,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1166,7 +1162,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.Image] @@ -1200,7 +1196,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -1391,12 +1387,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2017_12_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -1421,12 +1417,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2017_12_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -1570,7 +1566,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -1603,7 +1599,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -1676,7 +1672,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.Usage] @@ -1710,7 +1706,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -2060,7 +2056,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -2192,7 +2188,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -2225,7 +2221,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -2236,10 +2232,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2252,16 +2248,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2269,10 +2265,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2017_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2017_12_01.models.VaultSecretGroup] @@ -2306,7 +2302,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -2318,10 +2314,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2334,16 +2330,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2351,10 +2347,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2017_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -2425,7 +2421,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -2496,7 +2492,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -2769,7 +2765,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2832,7 +2828,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -2895,7 +2891,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -2969,7 +2965,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -3016,7 +3012,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3049,7 +3045,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.RunCommandDocumentBase] @@ -3085,7 +3081,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3245,7 +3241,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -3263,7 +3259,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -3282,12 +3278,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2017_12_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.DataDisk] """ @@ -3314,12 +3310,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2017_12_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.DataDisk] """ super().__init__(**kwargs) @@ -3354,7 +3350,7 @@ def __init__(self, **kwargs: Any) -> None: class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3371,57 +3367,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -3457,7 +3402,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -3607,7 +3552,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -3789,7 +3734,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3828,10 +3773,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -3845,7 +3790,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -3933,10 +3878,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -3946,7 +3891,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -4011,7 +3956,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -4088,7 +4033,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4206,7 +4151,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -4250,7 +4195,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4531,7 +4476,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2017_12_01.models.ResourceIdentityType :ivar identity_ids: The list of user identities associated with the Virtual Machine. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :vartype identity_ids: list[str] """ @@ -4562,7 +4507,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2017_12_01.models.ResourceIdentityType :keyword identity_ids: The list of user identities associated with the Virtual Machine. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :paramtype identity_ids: list[str] """ super().__init__(**kwargs) @@ -4575,7 +4520,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4631,7 +4576,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4819,7 +4764,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -4857,7 +4802,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5005,7 +4950,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -5192,10 +5137,10 @@ def __init__( self.provisioning_state = None -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension] @@ -5271,7 +5216,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2017_12_01.models.ResourceIdentityType :ivar identity_ids: The list of user identities associated with the virtual machine scale set. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :vartype identity_ids: list[str] """ @@ -5303,7 +5248,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2017_12_01.models.ResourceIdentityType :keyword identity_ids: The list of user identities associated with the virtual machine scale set. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :paramtype identity_ids: list[str] """ super().__init__(**kwargs) @@ -5350,7 +5295,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5377,7 +5322,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5495,10 +5440,10 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -5540,7 +5485,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet] @@ -5576,7 +5521,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetSku] @@ -5612,7 +5557,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet] @@ -5645,7 +5590,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -5673,10 +5618,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5763,7 +5708,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -5788,7 +5733,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2017_12_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -5813,7 +5758,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2017_12_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -5827,7 +5772,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -5947,10 +5892,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5963,16 +5908,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5980,10 +5925,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2017_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6026,10 +5971,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6042,16 +5987,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6059,10 +6004,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2017_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6078,10 +6023,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -6128,10 +6073,12 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -6242,12 +6189,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -6275,12 +6222,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -6375,7 +6322,7 @@ def __init__( self.single_placement_group = single_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -6481,7 +6428,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -6566,7 +6513,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar network_interface_configurations: The list of network configurations. @@ -6706,7 +6653,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -6750,7 +6697,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -6864,7 +6811,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6911,10 +6858,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -6926,7 +6873,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -7009,10 +6956,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -7022,7 +6969,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7044,7 +6991,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7097,10 +7044,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -7236,7 +7183,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM] @@ -7295,7 +7242,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -7357,7 +7304,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -7525,10 +7472,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -7542,7 +7489,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -7617,10 +7564,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2017_12_01.models.SubResource @@ -7630,7 +7577,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/operations/_operations.py index bdbe91908280..ace48d2df02a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2017_12_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -193,7 +188,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -249,7 +244,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -280,7 +275,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -313,7 +308,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -344,7 +339,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -390,7 +385,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -425,7 +420,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +455,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -492,7 +487,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -532,7 +527,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_get_extensions_request( +def build_virtual_machines_get_extensions_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -565,7 +560,9 @@ def build_virtual_machines_get_extensions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -627,7 +624,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -764,7 +761,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -795,7 +792,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -826,7 +823,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -857,7 +854,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -940,7 +937,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1095,7 +1092,7 @@ def build_virtual_machines_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1126,7 +1123,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1194,7 +1191,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1242,7 +1239,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1270,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1302,7 +1299,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1520,7 +1517,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1574,7 +1571,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1608,7 +1605,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1642,7 +1639,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1673,7 +1670,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1704,7 +1701,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1738,7 +1735,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1772,7 +1769,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1803,7 +1800,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1833,7 +1830,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1859,7 +1858,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1890,7 +1889,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1921,7 +1920,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1955,7 +1954,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1989,7 +1988,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2023,7 +2022,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2057,7 +2056,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2091,7 +2090,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2125,7 +2124,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2159,7 +2158,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2193,7 +2192,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2230,7 +2229,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2265,7 +2264,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2297,7 +2296,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2337,7 +2336,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2368,7 +2367,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2399,7 +2398,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2430,7 +2429,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2461,7 +2460,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2493,7 +2492,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2525,7 +2524,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2557,7 +2556,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2592,7 +2591,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2624,7 +2623,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2656,7 +2655,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2688,7 +2687,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2734,7 +2733,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2766,7 +2765,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2798,7 +2797,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2830,7 +2829,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2862,7 +2861,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2894,7 +2893,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2927,7 +2926,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2960,7 +2959,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2987,7 +2988,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3042,7 +3043,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -3066,14 +3066,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3084,14 +3083,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -3101,11 +3100,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3117,8 +3116,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -3161,7 +3158,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3172,7 +3168,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3184,11 +3180,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3199,7 +3194,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -3209,12 +3204,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3242,7 +3233,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3250,16 +3241,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3271,13 +3261,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -3300,7 +3286,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3311,7 +3296,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3323,11 +3308,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3338,7 +3322,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -3348,12 +3332,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3381,7 +3361,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3389,16 +3369,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3410,13 +3389,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( @@ -3428,7 +3403,6 @@ def delete( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or None or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.OperationStatusResponse or None :raises ~azure.core.exceptions.HttpResponseError: @@ -3447,21 +3421,20 @@ def delete( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3475,13 +3448,9 @@ def delete( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -3491,7 +3460,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3510,21 +3478,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3536,13 +3503,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -3553,7 +3516,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3575,16 +3537,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3595,14 +3556,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3612,11 +3573,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3628,17 +3589,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3660,16 +3616,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3680,14 +3635,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3697,11 +3652,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3713,10 +3668,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -3728,7 +3679,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -3750,17 +3700,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3771,14 +3720,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3788,11 +3737,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3804,10 +3753,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class VirtualMachineExtensionImagesOperations: """ @@ -3843,7 +3788,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3862,23 +3806,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3890,13 +3833,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -3908,7 +3847,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3927,21 +3865,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3953,13 +3890,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -3987,7 +3920,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4006,7 +3938,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -4015,16 +3947,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4036,13 +3967,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -4070,7 +3997,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4096,7 +4023,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4105,16 +4032,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4134,10 +4060,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -4164,14 +4086,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4185,7 +4099,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4201,18 +4115,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4226,7 +4132,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4239,20 +4145,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4286,7 +4181,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4296,24 +4191,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4339,7 +4232,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4348,16 +4241,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4369,13 +4261,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4404,14 +4292,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4425,7 +4305,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4441,18 +4321,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4466,7 +4338,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4479,20 +4351,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4526,7 +4387,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4536,17 +4397,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4565,22 +4424,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4594,13 +4452,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -4615,14 +4469,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -4653,7 +4499,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4663,17 +4509,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4695,7 +4539,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4714,23 +4557,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4742,13 +4584,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -4783,7 +4621,6 @@ def get_extensions( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4802,22 +4639,21 @@ def get_extensions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machines_get_extensions_request( + _request = build_virtual_machines_get_extensions_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_extensions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4829,13 +4665,9 @@ def get_extensions( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_extensions.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore @distributed_trace def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -4844,7 +4676,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -4866,16 +4697,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4886,14 +4716,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4903,11 +4733,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4919,15 +4749,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -4953,7 +4779,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4961,16 +4787,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4984,13 +4809,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -5014,14 +4835,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5034,7 +4847,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5047,18 +4860,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5071,7 +4876,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5082,19 +4887,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5127,7 +4922,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5139,20 +4934,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5177,7 +4974,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5185,16 +4982,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5214,10 +5010,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -5239,14 +5031,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5258,7 +5042,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5270,18 +5054,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5290,7 +5066,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -5299,19 +5079,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5343,7 +5112,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5353,20 +5122,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5391,7 +5162,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5399,16 +5170,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5428,10 +5198,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -5453,14 +5219,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5472,7 +5230,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5484,18 +5242,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5504,7 +5254,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -5513,19 +5267,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] @@ -5557,7 +5300,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5567,17 +5310,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5596,21 +5337,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5624,13 +5364,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -5642,14 +5378,6 @@ def begin_delete( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5679,7 +5407,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5689,17 +5417,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5714,7 +5440,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -5733,22 +5458,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5760,13 +5484,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -5778,7 +5498,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -5797,21 +5516,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5823,13 +5541,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5848,21 +5562,20 @@ def _convert_to_managed_disks_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5876,13 +5589,9 @@ def _convert_to_managed_disks_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return deserialized # type: ignore @distributed_trace def begin_convert_to_managed_disks( @@ -5895,14 +5604,6 @@ def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -5932,7 +5633,7 @@ def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5944,17 +5645,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5973,21 +5672,20 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6001,13 +5699,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -6020,14 +5714,6 @@ def begin_deallocate( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6057,7 +5743,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6069,17 +5755,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _models.OperationStatusResponse: @@ -6089,7 +5773,6 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OperationStatusResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.OperationStatusResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -6108,21 +5791,20 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.OperationStatusResponse] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6134,13 +5816,9 @@ def generalize(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -6149,7 +5827,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -6171,16 +5848,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6191,14 +5867,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6208,11 +5884,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6224,16 +5900,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -6255,15 +5926,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6274,14 +5944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6291,11 +5961,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6307,8 +5977,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6320,7 +5988,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6342,17 +6009,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6363,14 +6029,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6380,11 +6046,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6396,10 +6062,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -6417,21 +6079,20 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6445,13 +6106,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -6464,14 +6121,6 @@ def begin_power_off( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6501,7 +6150,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6513,17 +6162,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6542,21 +6189,20 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6570,13 +6216,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -6588,14 +6230,6 @@ def begin_restart( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6625,7 +6259,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6637,17 +6271,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6666,21 +6298,20 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6694,13 +6325,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -6712,14 +6339,6 @@ def begin_start( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6749,7 +6368,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6761,17 +6380,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6790,21 +6407,20 @@ def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6818,13 +6434,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -6836,14 +6448,6 @@ def begin_redeploy( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6873,7 +6477,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6885,17 +6489,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _perform_maintenance_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6914,21 +6516,20 @@ def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6942,13 +6543,9 @@ def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace def begin_perform_maintenance( @@ -6960,14 +6557,6 @@ def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -6997,7 +6586,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7009,20 +6598,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -7047,7 +6638,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7055,16 +6646,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7078,13 +6668,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -7107,14 +6693,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.RunCommandResult] @@ -7126,7 +6704,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7138,18 +6716,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.RunCommandResult] @@ -7158,7 +6728,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -7167,19 +6741,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.RunCommandResult] @@ -7211,7 +6774,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7223,17 +6786,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -7272,7 +6833,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -7291,7 +6851,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -7299,16 +6859,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7320,13 +6879,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7358,7 +6913,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7377,7 +6931,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -7387,16 +6941,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7408,13 +6961,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -7426,7 +6975,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7445,21 +6993,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7471,13 +7018,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -7485,7 +7028,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7504,20 +7046,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7529,13 +7070,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -7549,7 +7086,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -7568,22 +7104,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7595,13 +7130,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -7631,7 +7162,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7653,16 +7183,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7673,14 +7202,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -7690,11 +7219,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7706,8 +7235,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -7735,7 +7262,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7757,16 +7283,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7777,14 +7302,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7794,11 +7319,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7810,8 +7335,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -7834,7 +7357,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7859,7 +7382,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7867,16 +7390,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7896,10 +7418,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -7921,14 +7439,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7939,7 +7449,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7951,18 +7461,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7970,7 +7472,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -7979,19 +7481,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8022,7 +7513,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8032,20 +7523,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8070,7 +7563,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8078,16 +7571,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8107,10 +7599,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -8132,14 +7620,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8150,7 +7630,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8162,18 +7642,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8181,7 +7653,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -8190,19 +7666,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8233,7 +7698,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8243,17 +7708,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, image_name: str, **kwargs: Any @@ -8272,21 +7735,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8300,13 +7762,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -8318,14 +7776,6 @@ def begin_delete( :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -8355,7 +7805,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8365,17 +7815,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -8389,7 +7837,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -8408,22 +7855,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8435,13 +7881,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -8449,7 +7891,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8471,16 +7912,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8491,14 +7931,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8508,11 +7948,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8524,16 +7964,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2017_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8555,15 +7990,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8574,14 +8008,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8591,11 +8025,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8607,8 +8041,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -8634,7 +8066,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8660,7 +8092,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8668,16 +8100,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8697,10 +8128,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -8722,14 +8149,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8742,7 +8161,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8754,18 +8173,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8778,7 +8189,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8787,20 +8198,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8833,7 +8233,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8843,23 +8243,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8885,7 +8283,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8893,16 +8291,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8914,13 +8311,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8943,14 +8336,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8963,7 +8348,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8975,18 +8360,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8999,7 +8376,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9008,20 +8385,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9054,7 +8421,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9064,17 +8431,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9093,21 +8458,20 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9121,13 +8485,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -9139,14 +8499,6 @@ def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9176,7 +8528,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9186,17 +8538,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -9206,7 +8556,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9225,21 +8574,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9251,19 +8599,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9292,7 +8636,7 @@ def _deallocate_initial( else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9300,16 +8644,15 @@ def _deallocate_initial( content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9323,13 +8666,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return deserialized # type: ignore @overload def begin_deallocate( @@ -9356,14 +8695,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9376,7 +8707,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9391,18 +8722,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9415,7 +8738,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9427,20 +8750,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9473,7 +8785,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9485,23 +8797,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -9527,7 +8837,7 @@ def _delete_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9535,16 +8845,15 @@ def _delete_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9558,13 +8867,9 @@ def _delete_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return deserialized # type: ignore @overload def begin_delete_instances( @@ -9589,14 +8894,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9609,7 +8906,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9622,18 +8919,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9646,7 +8935,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Deletes virtual machines in a VM scale set. @@ -9656,20 +8945,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -9702,7 +8980,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9714,17 +8992,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_instance_view( @@ -9736,7 +9012,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9755,21 +9030,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9781,13 +9055,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -9795,7 +9065,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9819,16 +9088,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9839,14 +9107,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -9856,11 +9124,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9872,17 +9140,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9906,15 +9169,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9925,14 +9187,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -9942,11 +9204,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9958,8 +9220,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9971,7 +9231,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -9995,17 +9254,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10016,14 +9274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10033,11 +9291,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10049,10 +9307,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10063,7 +9317,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10087,17 +9340,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10108,14 +9360,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10125,11 +9377,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10141,15 +9393,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10178,7 +9426,7 @@ def _power_off_initial( else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10186,16 +9434,15 @@ def _power_off_initial( content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10209,13 +9456,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return deserialized # type: ignore @overload def begin_power_off( @@ -10242,14 +9485,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10262,7 +9497,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10277,18 +9512,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10301,7 +9528,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -10313,20 +9540,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10359,7 +9575,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10371,23 +9587,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10416,7 +9630,7 @@ def _restart_initial( else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10424,16 +9638,15 @@ def _restart_initial( content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10447,13 +9660,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return deserialized # type: ignore @overload def begin_restart( @@ -10478,14 +9687,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10498,7 +9699,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10511,18 +9712,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10535,7 +9728,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Restarts one or more virtual machines in a VM scale set. @@ -10545,20 +9738,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10591,7 +9773,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10603,23 +9785,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10648,7 +9828,7 @@ def _start_initial( else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10656,16 +9836,15 @@ def _start_initial( content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10679,13 +9858,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return deserialized # type: ignore @overload def begin_start( @@ -10710,14 +9885,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10730,7 +9897,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10743,18 +9910,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10767,7 +9926,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Starts one or more virtual machines in a VM scale set. @@ -10777,20 +9936,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10823,7 +9971,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10835,23 +9983,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -10880,7 +10026,7 @@ def _redeploy_initial( else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10888,16 +10034,15 @@ def _redeploy_initial( content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10911,13 +10056,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return deserialized # type: ignore @overload def begin_redeploy( @@ -10943,14 +10084,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -10963,7 +10096,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10977,18 +10110,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11001,7 +10126,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11012,20 +10137,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11058,7 +10172,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11070,23 +10184,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _perform_maintenance_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -11115,7 +10227,7 @@ def _perform_maintenance_initial( else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11123,16 +10235,15 @@ def _perform_maintenance_initial( content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11146,13 +10257,9 @@ def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return deserialized # type: ignore @overload def begin_perform_maintenance( @@ -11180,14 +10287,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11200,7 +10299,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11216,18 +10315,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11240,7 +10331,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11253,20 +10344,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11299,7 +10379,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11311,23 +10391,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_instances_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -11353,7 +10431,7 @@ def _update_instances_initial( else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11361,16 +10439,15 @@ def _update_instances_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11384,13 +10461,9 @@ def _update_instances_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return deserialized # type: ignore @overload def begin_update_instances( @@ -11415,14 +10488,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11435,7 +10500,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11448,18 +10513,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11472,7 +10529,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11482,20 +10539,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11528,7 +10574,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11540,23 +10586,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -11585,7 +10629,7 @@ def _reimage_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11593,16 +10637,15 @@ def _reimage_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11616,13 +10659,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return deserialized # type: ignore @overload def begin_reimage( @@ -11647,14 +10686,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11667,7 +10698,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11680,18 +10711,10 @@ def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11704,7 +10727,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -11714,20 +10737,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11760,7 +10772,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11772,23 +10784,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: error_map = { @@ -11817,7 +10827,7 @@ def _reimage_all_initial( else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11825,16 +10835,15 @@ def _reimage_all_initial( content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11848,13 +10857,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return deserialized # type: ignore @overload def begin_reimage_all( @@ -11880,14 +10885,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11900,7 +10897,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11914,18 +10911,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11938,7 +10927,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.OperationStatusResponse]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -11949,20 +10938,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -11995,7 +10973,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12007,20 +10985,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12033,7 +11009,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12052,22 +11027,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12079,16 +11053,12 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12113,7 +11083,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12139,7 +11109,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12148,16 +11118,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12177,10 +11146,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -12208,14 +11173,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12229,7 +11186,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12245,18 +11202,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12270,7 +11219,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -12283,20 +11232,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12330,7 +11268,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12340,17 +11278,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -12369,22 +11305,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12398,13 +11333,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -12419,14 +11350,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12457,7 +11380,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12467,17 +11390,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -12499,7 +11420,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -12518,23 +11438,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12546,13 +11465,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12564,7 +11479,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12588,17 +11502,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12609,14 +11522,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -12626,11 +11539,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12642,12 +11555,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12684,21 +11593,20 @@ def _cancel_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12712,13 +11620,9 @@ def _cancel_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return deserialized # type: ignore @distributed_trace def begin_cancel( @@ -12730,14 +11634,6 @@ def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12767,7 +11663,7 @@ def begin_cancel( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12779,17 +11675,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_os_upgrade_initial( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12808,21 +11702,20 @@ def _start_os_upgrade_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12836,13 +11729,9 @@ def _start_os_upgrade_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return deserialized # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -12856,14 +11745,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -12893,7 +11774,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12905,17 +11786,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_latest( @@ -12927,7 +11806,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -12946,21 +11824,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12972,13 +11849,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -13018,22 +11891,21 @@ def _reimage_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13047,13 +11919,9 @@ def _reimage_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return deserialized # type: ignore @distributed_trace def begin_reimage( @@ -13067,14 +11935,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13105,7 +11965,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13117,17 +11977,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _reimage_all_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13146,22 +12004,21 @@ def _reimage_all_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13175,13 +12032,9 @@ def _reimage_all_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return deserialized # type: ignore @distributed_trace def begin_reimage_all( @@ -13196,14 +12049,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13234,7 +12079,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13246,17 +12091,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _deallocate_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13275,22 +12118,21 @@ def _deallocate_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13304,13 +12146,9 @@ def _deallocate_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return deserialized # type: ignore @distributed_trace def begin_deallocate( @@ -13326,14 +12164,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13364,7 +12194,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13376,24 +12206,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -13419,7 +12247,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13428,16 +12256,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13457,10 +12284,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -13487,14 +12310,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13508,7 +12323,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13524,18 +12339,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13549,7 +12356,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -13562,19 +12369,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13608,7 +12404,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13618,17 +12414,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13647,22 +12441,21 @@ def _delete_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13676,13 +12469,9 @@ def _delete_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def begin_delete( @@ -13696,14 +12485,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -13734,7 +12515,7 @@ def begin_delete( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13744,17 +12525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -13768,7 +12547,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -13787,22 +12565,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13814,13 +12591,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13834,7 +12607,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13853,22 +12625,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13880,13 +12651,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13915,7 +12682,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13939,7 +12705,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13947,12 +12713,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13963,14 +12728,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13980,11 +12745,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13996,10 +12761,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> Optional[_models.OperationStatusResponse]: @@ -14017,22 +12778,21 @@ def _power_off_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14046,13 +12806,9 @@ def _power_off_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return deserialized # type: ignore @distributed_trace def begin_power_off( @@ -14068,14 +12824,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14106,7 +12854,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14118,17 +12866,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _restart_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14147,22 +12893,21 @@ def _restart_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14176,13 +12921,9 @@ def _restart_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return deserialized # type: ignore @distributed_trace def begin_restart( @@ -14196,14 +12937,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14234,7 +12967,7 @@ def begin_restart( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14246,17 +12979,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _start_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14275,22 +13006,21 @@ def _start_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14304,13 +13034,9 @@ def _start_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return deserialized # type: ignore @distributed_trace def begin_start( @@ -14324,14 +13050,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14362,7 +13080,7 @@ def begin_start( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14374,17 +13092,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _redeploy_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14403,22 +13119,21 @@ def _redeploy_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14432,13 +13147,9 @@ def _redeploy_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return deserialized # type: ignore @distributed_trace def begin_redeploy( @@ -14453,14 +13164,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14491,7 +13194,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14503,17 +13206,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _perform_maintenance_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14532,22 +13233,21 @@ def _perform_maintenance_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[Optional[_models.OperationStatusResponse]] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14561,13 +13261,9 @@ def _perform_maintenance_initial( deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return deserialized # type: ignore @distributed_trace def begin_perform_maintenance( @@ -14581,14 +13277,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either OperationStatusResponse or the result of cls(response) :rtype: @@ -14619,7 +13307,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): deserialized = self._deserialize("OperationStatusResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14631,17 +13319,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.OperationStatusResponse].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[_models.OperationStatusResponse]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -14665,7 +13351,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -14690,23 +13376,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14720,13 +13405,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -14748,14 +13429,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14765,7 +13438,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -14774,18 +13447,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14795,7 +13460,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -14803,19 +13468,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14847,7 +13502,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14859,20 +13514,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -14897,23 +13550,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14927,13 +13579,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -14954,14 +13602,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14971,7 +13611,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -14979,18 +13619,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15000,7 +13632,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15008,19 +13640,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2017_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15052,7 +13673,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15064,17 +13685,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -15103,7 +13722,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -15127,16 +13745,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15147,14 +13764,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -15164,11 +13781,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15180,10 +13797,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -15192,7 +13805,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2017_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -15211,21 +13823,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2017-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15237,10 +13848,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_compute_management_client.py index 392779a3e0d8..4f63bcae112b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -117,7 +119,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -170,7 +190,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -190,7 +210,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_configuration.py index efaccd087f70..bc115f94803a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_metadata.json index 28fa4065ec63..c048ce1d16d0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_compute_management_client.py index 6a5e9a67f0b9..ef21c825434a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -118,7 +120,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -171,7 +191,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -191,7 +213,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_configuration.py index f5c9eacad833..f8dd90dd2230 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/operations/_operations.py index 62d1e9c5c468..b265ec05d623 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -150,10 +149,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -182,7 +177,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -206,14 +200,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -224,14 +217,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -241,11 +234,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -257,8 +250,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -301,7 +292,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -312,7 +302,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -324,11 +314,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -339,7 +328,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -349,12 +338,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -382,7 +367,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -390,16 +375,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -411,13 +395,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -440,7 +420,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -451,7 +430,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -463,11 +442,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -478,7 +456,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -488,12 +466,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -521,7 +495,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -529,16 +503,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -550,13 +523,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -568,7 +537,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -587,21 +555,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -611,11 +578,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -625,7 +588,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -644,21 +606,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -670,13 +631,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -687,7 +644,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet] @@ -710,16 +666,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -730,14 +685,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -747,11 +702,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -763,17 +718,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet] @@ -796,16 +746,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -816,14 +765,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -833,11 +782,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -849,10 +798,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -864,7 +809,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] @@ -887,17 +831,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -908,14 +851,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -925,11 +868,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -941,10 +884,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -988,7 +927,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -999,7 +937,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1012,11 +950,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1027,7 +964,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1037,12 +974,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1070,7 +1003,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1078,16 +1011,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1107,10 +1039,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1133,7 +1061,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1144,7 +1071,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1157,11 +1084,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1172,7 +1098,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1182,12 +1108,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1215,7 +1138,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1223,16 +1146,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1244,13 +1166,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1262,7 +1180,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1281,21 +1198,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1305,11 +1221,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1321,7 +1233,6 @@ async def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1340,21 +1251,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1366,19 +1276,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1402,15 +1307,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1421,14 +1325,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1438,11 +1342,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1454,10 +1358,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1466,7 +1366,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1490,16 +1389,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1510,14 +1408,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1527,11 +1425,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1543,10 +1441,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -1582,7 +1476,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1601,23 +1494,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1629,13 +1521,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -1647,7 +1535,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1666,21 +1553,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1692,13 +1578,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -1726,7 +1608,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1745,7 +1626,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -1754,16 +1635,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1775,13 +1655,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -1809,7 +1685,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1835,7 +1711,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1844,16 +1720,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1873,10 +1748,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1903,14 +1774,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1924,7 +1787,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1940,18 +1803,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1965,7 +1820,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1978,20 +1833,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2025,7 +1869,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2035,24 +1879,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2078,7 +1920,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2087,16 +1929,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2108,13 +1949,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2143,14 +1980,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2164,7 +1993,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2180,18 +2009,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2205,7 +2026,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -2218,20 +2039,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2265,7 +2075,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2275,17 +2085,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -2304,22 +2112,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2329,11 +2136,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2348,14 +2151,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2383,7 +2178,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2392,17 +2187,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2424,7 +2215,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -2443,23 +2233,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,13 +2260,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2491,7 +2276,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2510,22 +2294,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2537,13 +2320,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -2582,7 +2361,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2601,7 +2379,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2609,16 +2387,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2630,13 +2407,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2668,7 +2441,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2687,7 +2459,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2697,16 +2469,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2718,13 +2489,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -2736,7 +2503,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2755,21 +2521,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2781,13 +2546,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -2795,7 +2556,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2814,20 +2574,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2839,13 +2598,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -2859,7 +2614,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2878,22 +2632,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2905,13 +2658,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -2941,7 +2690,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2963,16 +2711,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2983,14 +2730,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -3000,11 +2747,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3016,8 +2763,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -3046,7 +2791,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -3069,16 +2813,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3089,14 +2832,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3106,11 +2849,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3122,15 +2865,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -3156,7 +2895,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3164,16 +2903,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3187,13 +2925,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -3217,14 +2951,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3237,7 +2963,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3250,18 +2976,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3274,7 +2992,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -3285,19 +3003,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3330,7 +3038,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3342,20 +3050,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3380,7 +3090,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3388,16 +3098,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3417,10 +3126,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -3442,14 +3147,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3462,7 +3159,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3474,18 +3171,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3495,7 +3184,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -3504,19 +3197,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3549,7 +3231,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3559,20 +3241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3597,7 +3281,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3605,16 +3289,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3634,10 +3317,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -3659,14 +3338,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3679,7 +3350,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3691,18 +3362,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3712,7 +3375,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -3721,19 +3388,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3766,7 +3422,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3776,17 +3432,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3805,21 +3459,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3829,11 +3482,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -3843,14 +3492,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3877,7 +3518,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3886,17 +3527,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -3911,7 +3548,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -3930,22 +3566,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3957,13 +3592,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -3975,7 +3606,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -3994,21 +3624,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4020,13 +3649,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4045,21 +3670,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4069,11 +3693,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -4086,14 +3706,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4120,7 +3732,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4129,17 +3741,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4158,21 +3766,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4182,11 +3789,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4197,14 +3800,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4231,7 +3826,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4240,17 +3835,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -4262,7 +3853,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4281,21 +3871,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4305,11 +3894,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -4318,7 +3903,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -4341,16 +3925,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4361,14 +3944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4378,11 +3961,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4394,16 +3977,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -4426,15 +4004,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4445,14 +4022,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4462,11 +4039,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4478,8 +4055,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4491,7 +4066,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] @@ -4514,17 +4088,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4535,14 +4108,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4552,11 +4125,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4568,10 +4141,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -4589,21 +4158,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4613,11 +4181,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4628,14 +4192,6 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4662,7 +4218,7 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4671,17 +4227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4700,21 +4252,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4724,11 +4275,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4738,14 +4285,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4772,7 +4311,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4781,17 +4320,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4810,21 +4345,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4834,11 +4368,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4848,14 +4378,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4882,7 +4404,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4891,17 +4413,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4920,21 +4438,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4944,11 +4461,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4958,14 +4471,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4992,7 +4497,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5001,17 +4506,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5030,21 +4531,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5054,11 +4554,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -5070,14 +4566,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5104,7 +4592,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5113,20 +4601,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -5151,7 +4639,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5159,16 +4647,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5182,13 +4669,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -5211,14 +4694,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5231,7 +4706,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5243,18 +4718,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5264,7 +4731,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -5273,19 +4744,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5318,7 +4778,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5330,17 +4790,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -5369,7 +4827,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] @@ -5392,16 +4849,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5412,14 +4868,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5429,11 +4885,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5445,8 +4901,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -5469,7 +4923,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5494,7 +4948,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5502,16 +4956,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5531,10 +4984,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -5556,14 +5005,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5574,7 +5015,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5586,18 +5027,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5605,7 +5038,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -5614,19 +5047,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5657,7 +5079,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5667,20 +5089,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5705,7 +5129,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5713,16 +5137,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5742,10 +5165,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -5767,14 +5186,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5785,7 +5196,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5797,18 +5208,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5816,7 +5219,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -5825,19 +5232,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5868,7 +5264,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5878,17 +5274,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -5907,21 +5301,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5931,11 +5324,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5945,14 +5334,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5979,7 +5360,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5988,17 +5369,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6012,7 +5389,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -6031,22 +5407,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6058,13 +5433,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -6072,7 +5443,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6094,16 +5464,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6114,14 +5483,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6131,11 +5500,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6147,16 +5516,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6178,15 +5542,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6197,14 +5560,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6214,11 +5577,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6230,8 +5593,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6257,7 +5618,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6283,7 +5644,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6291,16 +5652,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6320,10 +5680,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -6345,14 +5701,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6365,7 +5713,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6377,18 +5725,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6401,7 +5741,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6410,20 +5750,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6456,7 +5785,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6466,23 +5795,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6508,7 +5835,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6516,16 +5843,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6537,13 +5863,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6566,14 +5888,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6586,7 +5900,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6598,18 +5912,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6622,7 +5928,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6631,20 +5937,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6677,7 +5973,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6687,17 +5983,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6716,21 +6010,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6740,11 +6033,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6756,14 +6045,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6790,7 +6071,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6799,17 +6080,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6821,7 +6098,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6840,21 +6116,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6866,19 +6141,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -6907,7 +6178,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6915,16 +6186,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6934,11 +6204,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -6965,14 +6231,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6983,7 +6241,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6998,18 +6256,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7020,7 +6270,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7032,20 +6282,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7075,7 +6314,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7084,23 +6323,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -7126,7 +6361,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7134,16 +6369,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7153,11 +6387,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -7182,14 +6412,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7200,7 +6422,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7213,18 +6435,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7235,7 +6449,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -7245,20 +6459,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7288,7 +6491,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7297,17 +6500,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -7319,7 +6518,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7338,21 +6536,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7364,13 +6561,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -7378,7 +6571,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7402,16 +6594,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7422,14 +6613,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7439,11 +6630,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7455,17 +6646,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7489,15 +6675,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7508,14 +6693,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7525,11 +6710,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7541,8 +6726,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7554,7 +6737,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7578,17 +6760,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7599,14 +6780,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7616,11 +6797,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7632,10 +6813,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7646,7 +6823,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7670,17 +6846,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7691,14 +6866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7708,11 +6883,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7724,15 +6899,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7761,7 +6932,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7769,16 +6940,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7788,11 +6958,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -7819,14 +6985,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7837,7 +6995,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7852,18 +7010,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7874,7 +7024,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -7886,20 +7036,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7929,7 +7068,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7938,23 +7077,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7983,7 +7118,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7991,16 +7126,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8010,11 +7144,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -8039,14 +7169,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8057,7 +7179,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8070,18 +7192,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8092,7 +7206,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8102,20 +7216,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8145,7 +7248,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8154,23 +7257,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8199,7 +7298,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8207,16 +7306,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8226,11 +7324,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -8255,14 +7349,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8273,7 +7359,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8286,18 +7372,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8308,7 +7386,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8318,20 +7396,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8361,7 +7428,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8370,23 +7437,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8415,7 +7478,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8423,16 +7486,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8442,11 +7504,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -8472,14 +7530,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8490,7 +7540,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8504,18 +7554,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8526,7 +7568,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8537,20 +7579,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8580,7 +7611,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8589,23 +7620,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8634,7 +7661,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8642,16 +7669,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8661,11 +7687,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -8693,14 +7715,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8711,7 +7725,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8727,18 +7741,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8749,7 +7755,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8762,20 +7768,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8805,7 +7800,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8814,23 +7809,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8856,7 +7847,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8864,16 +7855,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8883,11 +7873,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -8912,14 +7898,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8930,7 +7908,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8943,18 +7921,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8965,7 +7935,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -8975,20 +7945,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9018,7 +7977,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9027,23 +7986,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9072,7 +8027,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9080,16 +8035,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9099,11 +8053,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -9128,14 +8078,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9146,7 +8088,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9159,18 +8101,10 @@ async def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9181,7 +8115,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -9191,20 +8125,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9234,7 +8157,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9243,23 +8166,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9288,7 +8207,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9296,16 +8215,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9315,11 +8233,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -9345,14 +8259,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9363,7 +8269,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9377,18 +8283,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9399,7 +8297,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9410,20 +8308,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9453,7 +8340,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9462,20 +8349,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -9488,7 +8371,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9507,22 +8389,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9534,16 +8415,12 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -9568,7 +8445,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -9594,7 +8471,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -9603,16 +8480,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9632,10 +8508,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9663,14 +8535,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9684,7 +8548,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9700,18 +8564,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9725,7 +8581,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -9738,20 +8594,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9785,7 +8630,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9795,17 +8640,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -9824,22 +8667,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9849,11 +8691,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9868,14 +8706,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9903,7 +8733,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9912,17 +8742,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9944,7 +8770,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9963,23 +8788,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9991,13 +8815,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10009,7 +8829,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10033,17 +8852,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10054,14 +8872,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10071,11 +8889,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10087,12 +8905,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10129,21 +8943,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10153,11 +8966,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -10169,14 +8978,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10203,7 +9004,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10212,17 +9013,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10241,21 +9038,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10265,11 +9061,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -10283,14 +9075,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10317,7 +9101,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10326,17 +9110,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -10348,7 +9128,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -10367,21 +9146,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10393,13 +9171,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -10439,22 +9213,21 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10464,11 +9237,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage( @@ -10482,14 +9251,6 @@ async def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10517,7 +9278,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10526,17 +9287,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10555,22 +9312,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10580,11 +9336,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -10599,14 +9351,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10634,7 +9378,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10643,17 +9387,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10672,22 +9412,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10697,11 +9436,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10717,14 +9452,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10752,7 +9479,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10761,24 +9488,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -10804,7 +9527,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -10813,16 +9536,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10842,10 +9564,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -10872,14 +9590,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10893,7 +9603,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10909,18 +9619,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10934,7 +9636,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -10947,19 +9649,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -10993,7 +9684,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11003,17 +9694,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11032,22 +9721,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11057,11 +9745,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -11075,14 +9759,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11110,7 +9786,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11119,17 +9795,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11143,7 +9815,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -11162,22 +9833,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11189,13 +9859,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -11209,7 +9875,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11228,22 +9893,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11255,13 +9919,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11290,7 +9950,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11314,7 +9973,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -11322,12 +9981,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11338,14 +9996,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -11355,11 +10013,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11371,10 +10029,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -11392,22 +10046,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11417,11 +10070,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11437,14 +10086,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11472,7 +10113,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11481,17 +10122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11510,22 +10147,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11535,11 +10171,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -11553,14 +10185,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11588,7 +10212,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11597,17 +10221,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11626,22 +10246,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11651,11 +10270,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -11669,14 +10284,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11704,7 +10311,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11713,17 +10320,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11742,22 +10345,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11767,11 +10369,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -11786,14 +10384,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11821,7 +10411,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11830,17 +10420,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11859,22 +10445,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11884,11 +10469,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11902,14 +10483,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11937,7 +10510,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11946,24 +10519,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -11989,7 +10558,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11998,16 +10567,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12021,13 +10589,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12053,14 +10617,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12074,7 +10630,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12088,18 +10644,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12113,7 +10661,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -12125,19 +10673,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12171,7 +10708,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12183,17 +10720,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -12217,7 +10752,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12242,23 +10777,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12272,13 +10806,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -12300,14 +10830,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12317,7 +10839,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12326,18 +10848,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12347,7 +10861,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12355,19 +10869,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12399,7 +10903,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12412,20 +10916,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12450,23 +10952,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12480,13 +10981,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -12507,14 +11004,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12524,7 +11013,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12532,18 +11021,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12553,7 +11034,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12561,19 +11042,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12605,7 +11075,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12618,17 +11088,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -12657,7 +11125,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -12681,16 +11148,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12701,14 +11167,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -12718,11 +11184,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12734,10 +11200,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -12746,7 +11208,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -12765,21 +11226,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12791,13 +11251,9 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class DisksOperations: @@ -12821,7 +11277,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -12846,7 +11302,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -12854,16 +11310,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12883,10 +11338,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -12910,14 +11361,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -12928,7 +11371,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12942,18 +11385,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -12961,7 +11396,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -12972,19 +11407,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_04_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_04_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13015,7 +11439,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13025,20 +11449,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -13063,7 +11485,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -13071,16 +11493,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13100,10 +11521,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -13127,14 +11544,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13145,7 +11554,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13159,18 +11568,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13178,7 +11579,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -13189,19 +11590,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_04_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_04_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13232,7 +11622,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13242,17 +11632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -13264,7 +11652,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -13283,21 +11670,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13309,13 +11695,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -13334,21 +11716,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13358,11 +11739,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -13374,14 +11751,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13408,7 +11777,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13417,17 +11786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -13435,7 +11800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13457,16 +11821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13477,14 +11840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -13494,11 +11857,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13510,15 +11873,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -13540,15 +11898,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13559,14 +11916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -13576,11 +11933,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13592,13 +11949,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -13624,7 +11979,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -13632,16 +11987,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13655,13 +12009,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -13687,14 +12037,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -13706,7 +12048,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13721,18 +12063,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -13744,7 +12078,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -13756,19 +12090,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -13800,7 +12123,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13812,17 +12135,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -13841,21 +12162,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13865,11 +12185,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -13883,14 +12199,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13917,7 +12225,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -13928,17 +12236,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -13962,7 +12266,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -13987,7 +12291,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -13995,16 +12299,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14024,10 +12327,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -14051,14 +12350,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14070,7 +12361,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14084,18 +12375,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14104,7 +12387,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -14115,19 +12398,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14159,7 +12431,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14169,20 +12441,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -14207,7 +12481,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -14215,16 +12489,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14244,10 +12517,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -14272,14 +12541,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14291,7 +12552,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14306,18 +12567,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14326,7 +12579,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -14337,19 +12594,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14381,7 +12627,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14391,17 +12637,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -14413,7 +12657,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -14432,21 +12675,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14458,13 +12700,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -14483,21 +12721,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14507,11 +12744,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -14523,14 +12756,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14557,7 +12782,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14566,17 +12791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -14584,7 +12805,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14607,16 +12827,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14627,14 +12846,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -14644,11 +12863,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14660,15 +12879,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -14691,15 +12905,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14710,14 +12923,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -14727,11 +12940,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14743,13 +12956,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -14775,7 +12986,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -14783,16 +12994,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14806,13 +13016,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -14838,14 +13044,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -14857,7 +13055,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14872,18 +13070,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -14895,7 +13085,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -14907,19 +13097,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] @@ -14951,7 +13130,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14963,17 +13142,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -14992,21 +13169,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15016,11 +13192,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -15034,14 +13206,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15068,7 +13232,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -15079,14 +13243,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_compute_management_client_enums.py index 14cb472fea04..13c750cfc880 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_compute_management_client_enums.py @@ -35,10 +35,9 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): EMPTY = "Empty" ATTACH = "Attach" FROM_IMAGE = "FromImage" - IMPORT = "Import" + IMPORT_ENUM = "Import" COPY = "Copy" RESTORE = "Restore" - IMPORT_ENUM = "Import" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_models_py3.py index 0bf932b3fa9e..042b976e3cbb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -247,7 +243,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -305,7 +301,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -398,7 +394,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet] @@ -665,10 +661,10 @@ def __init__(self, **kwargs: Any) -> None: class CreationData(_serialization.Model): """Data used when creating a disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :vartype create_option: str or ~azure.mgmt.compute.v2018_04_01.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -708,7 +704,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :paramtype create_option: str or ~azure.mgmt.compute.v2018_04_01.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in @@ -734,7 +730,7 @@ def __init__( class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -909,7 +905,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1106,7 +1102,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.Disk] @@ -1277,7 +1273,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None" and "Read". :vartype access: str or ~azure.mgmt.compute.v2018_04_01.models.AccessLevel @@ -1312,7 +1308,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1369,7 +1365,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1426,7 +1422,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1493,7 +1489,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1586,7 +1582,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1624,7 +1620,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.Image] @@ -1658,7 +1654,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -1850,12 +1846,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_04_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -1880,12 +1876,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_04_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -2030,7 +2026,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_04_01.models.SourceVault @@ -2063,7 +2059,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_04_01.models.SourceVault @@ -2096,7 +2092,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -2129,7 +2125,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -2202,7 +2198,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.Usage] @@ -2236,7 +2232,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -2524,7 +2520,7 @@ class OSDisk(_serialization.Model): virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -2656,7 +2652,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -2689,7 +2685,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -2700,10 +2696,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2716,16 +2712,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2733,10 +2729,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_04_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2018_04_01.models.VaultSecretGroup] @@ -2770,7 +2766,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -2782,10 +2778,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2798,16 +2794,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2815,10 +2811,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_04_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -2891,7 +2887,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2972,7 +2968,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup] @@ -3010,22 +3006,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -3096,7 +3081,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3369,7 +3354,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3432,7 +3417,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3495,7 +3480,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3569,7 +3554,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -3616,7 +3601,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3649,7 +3634,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.RunCommandDocumentBase] @@ -3685,7 +3670,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3794,7 +3779,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3900,7 +3885,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.Snapshot] @@ -4075,7 +4060,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -4093,7 +4078,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -4112,12 +4097,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_04_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.DataDisk] """ @@ -4144,12 +4129,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_04_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.DataDisk] """ super().__init__(**kwargs) @@ -4184,7 +4169,7 @@ def __init__(self, **kwargs: Any) -> None: class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4201,57 +4186,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -4287,7 +4221,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -4437,7 +4371,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -4619,7 +4553,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4658,10 +4592,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -4679,7 +4613,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -4769,10 +4703,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -4786,7 +4720,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -4852,7 +4786,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -4942,7 +4876,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5060,7 +4994,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -5104,7 +5038,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5385,7 +5319,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_04_01.models.ResourceIdentityType :ivar identity_ids: The list of user identities associated with the Virtual Machine. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :vartype identity_ids: list[str] """ @@ -5416,7 +5350,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2018_04_01.models.ResourceIdentityType :keyword identity_ids: The list of user identities associated with the Virtual Machine. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :paramtype identity_ids: list[str] """ super().__init__(**kwargs) @@ -5429,7 +5363,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5485,7 +5419,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5673,7 +5607,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -5711,7 +5645,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5870,7 +5804,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -6057,10 +5991,10 @@ def __init__( self.provisioning_state = None -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension] @@ -6136,7 +6070,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_04_01.models.ResourceIdentityType :ivar identity_ids: The list of user identities associated with the virtual machine scale set. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :vartype identity_ids: list[str] """ @@ -6168,7 +6102,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2018_04_01.models.ResourceIdentityType :keyword identity_ids: The list of user identities associated with the virtual machine scale set. The user identity references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/identities/{identityName}'. # pylint: disable=line-too-long :paramtype identity_ids: list[str] """ super().__init__(**kwargs) @@ -6215,7 +6149,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -6242,7 +6176,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6386,10 +6320,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -6431,7 +6365,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet] @@ -6467,7 +6401,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetSku] @@ -6503,7 +6437,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet] @@ -6536,7 +6470,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -6566,10 +6500,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6656,7 +6590,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -6681,7 +6615,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2018_04_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -6706,7 +6640,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2018_04_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -6720,7 +6654,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -6851,10 +6785,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6867,16 +6801,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6884,10 +6818,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_04_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6930,10 +6864,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6946,16 +6880,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6963,10 +6897,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_04_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6982,10 +6916,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -7039,10 +6973,12 @@ def __init__( self.ip_tags = ip_tags -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -7153,12 +7089,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetDataDisk] """ @@ -7186,12 +7122,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetDataDisk] """ @@ -7297,7 +7233,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -7403,7 +7339,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -7488,7 +7424,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar network_interface_configurations: The list of network configurations. @@ -7639,7 +7575,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -7683,7 +7619,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -7797,7 +7733,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7846,10 +7782,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -7861,7 +7797,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -7946,10 +7882,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -7959,7 +7895,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7982,7 +7918,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8035,10 +7971,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -8174,7 +8110,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM] @@ -8233,7 +8169,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -8295,7 +8231,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -8463,10 +8399,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -8484,7 +8420,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -8561,10 +8497,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_04_01.models.SubResource @@ -8578,7 +8514,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/operations/_operations.py index d365624ad9d3..f87b5eebdd1a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_04_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +401,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -432,7 +429,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -462,7 +459,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -495,7 +492,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -526,7 +523,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -572,7 +569,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -607,7 +604,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -642,7 +639,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -668,7 +665,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -708,7 +705,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -775,7 +772,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -823,7 +820,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -854,7 +851,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -883,7 +880,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +939,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1004,7 +1003,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1134,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1165,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1191,7 +1190,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1216,7 +1215,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1293,7 +1292,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1424,7 +1423,7 @@ def build_virtual_machines_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1449,7 +1448,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1636,7 +1635,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1690,7 +1689,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1724,7 +1723,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1758,7 +1757,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1783,7 +1782,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1814,7 +1813,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1845,7 +1844,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1876,7 +1875,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1907,7 +1906,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1937,7 +1936,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1963,7 +1964,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1994,7 +1995,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2025,7 +2026,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2056,7 +2057,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2087,7 +2088,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2118,7 +2119,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2149,7 +2150,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2180,7 +2181,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2211,7 +2212,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2242,7 +2243,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2273,7 +2274,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2310,7 +2311,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2345,7 +2346,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2371,7 +2372,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2411,7 +2412,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2442,7 +2443,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2467,7 +2468,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2492,7 +2493,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2523,7 +2524,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2549,7 +2550,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2575,7 +2576,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2601,7 +2602,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2636,7 +2637,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2662,7 +2663,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2694,7 +2695,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2726,7 +2727,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2772,7 +2773,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2798,7 +2799,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2824,7 +2825,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2850,7 +2851,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2876,7 +2877,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2902,7 +2903,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2937,7 +2938,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2970,7 +2971,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3003,7 +3004,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3030,7 +3033,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3185,7 +3188,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3422,7 +3425,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3559,7 +3562,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -3583,14 +3585,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3601,14 +3602,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -3618,11 +3619,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3634,8 +3635,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -3678,7 +3677,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3689,7 +3687,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3701,11 +3699,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3716,7 +3713,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -3726,12 +3723,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3759,7 +3752,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3767,16 +3760,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3788,13 +3780,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -3817,7 +3805,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3828,7 +3815,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3840,11 +3827,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3855,7 +3841,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -3865,12 +3851,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3898,7 +3880,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3906,16 +3888,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3927,13 +3908,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -3945,7 +3922,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3964,21 +3940,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3988,11 +3963,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -4002,7 +3973,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4021,21 +3991,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4047,13 +4016,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -4064,7 +4029,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4086,16 +4050,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4106,14 +4069,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4123,11 +4086,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4139,17 +4102,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4171,16 +4129,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4191,14 +4148,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4208,11 +4165,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4224,10 +4181,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -4239,7 +4192,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -4261,17 +4213,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4282,14 +4233,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4299,11 +4250,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4315,10 +4266,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -4362,7 +4309,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4373,7 +4319,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4386,11 +4332,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4401,7 +4346,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -4411,12 +4356,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4444,7 +4385,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4452,16 +4393,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4481,10 +4421,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -4507,7 +4443,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4518,7 +4453,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4531,11 +4466,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4546,7 +4480,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -4556,12 +4490,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,7 +4520,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4597,16 +4528,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4618,13 +4548,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -4636,7 +4562,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4655,21 +4580,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4679,11 +4603,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -4695,7 +4615,6 @@ def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4714,21 +4633,20 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4740,19 +4658,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -4776,15 +4689,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4795,14 +4707,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -4812,11 +4724,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4828,10 +4740,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -4840,7 +4748,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -4864,16 +4771,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4884,14 +4790,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -4901,11 +4807,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4917,10 +4823,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -4956,7 +4858,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4975,23 +4876,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4903,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -5021,7 +4917,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5040,21 +4935,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5066,13 +4960,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -5100,7 +4990,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5119,7 +5008,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -5128,16 +5017,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5149,13 +5037,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -5183,7 +5067,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -5209,7 +5093,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -5218,16 +5102,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5247,10 +5130,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -5277,14 +5156,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5298,7 +5169,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5314,18 +5185,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5339,7 +5202,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -5352,20 +5215,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5399,7 +5251,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5409,24 +5261,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -5452,7 +5302,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -5461,16 +5311,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5482,13 +5331,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5517,14 +5362,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5538,7 +5375,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5554,18 +5391,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5579,7 +5408,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -5592,20 +5421,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5639,7 +5457,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5649,17 +5467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -5678,22 +5494,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5703,11 +5518,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5722,14 +5533,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5757,7 +5560,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5766,17 +5569,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -5798,7 +5597,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5817,23 +5615,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5845,13 +5642,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5865,7 +5658,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -5884,22 +5676,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5911,13 +5702,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -5956,7 +5743,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5975,7 +5761,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -5983,16 +5769,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6004,13 +5789,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6042,7 +5823,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6061,7 +5841,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6071,16 +5851,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6092,13 +5871,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -6110,7 +5885,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6129,21 +5903,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6155,13 +5928,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -6169,7 +5938,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,20 +5956,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6213,13 +5980,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -6233,7 +5996,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6252,22 +6014,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6279,13 +6040,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -6315,7 +6072,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6337,16 +6093,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6357,14 +6112,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6374,11 +6129,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6390,8 +6145,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -6420,7 +6173,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -6442,16 +6194,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6462,14 +6213,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6479,11 +6230,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6495,15 +6246,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -6529,7 +6276,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6537,16 +6284,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6560,13 +6306,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -6590,14 +6332,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6610,7 +6344,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6623,18 +6357,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6647,7 +6373,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -6658,19 +6384,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6703,7 +6419,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6715,20 +6431,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6753,7 +6471,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6761,16 +6479,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6790,10 +6507,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -6815,14 +6528,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -6834,7 +6539,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6846,18 +6551,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -6866,7 +6563,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -6875,19 +6576,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -6919,7 +6609,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6929,20 +6619,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6967,7 +6659,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6975,16 +6667,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7004,10 +6695,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -7029,14 +6716,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -7048,7 +6727,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7060,18 +6739,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -7080,7 +6751,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -7089,19 +6764,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] @@ -7133,7 +6797,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7143,17 +6807,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7172,21 +6834,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7196,11 +6857,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7210,14 +6867,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7244,7 +6893,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7253,17 +6902,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7278,7 +6923,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -7297,22 +6941,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7324,13 +6967,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -7342,7 +6981,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7361,21 +6999,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7387,13 +7024,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7412,21 +7045,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7436,11 +7068,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7451,14 +7079,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7485,7 +7105,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7494,17 +7114,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7523,21 +7139,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7547,11 +7162,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7562,14 +7173,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7596,7 +7199,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7605,17 +7208,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -7627,7 +7226,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7646,21 +7244,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7670,11 +7267,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -7683,7 +7276,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -7705,16 +7297,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7725,14 +7316,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7742,11 +7333,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7758,16 +7349,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -7789,15 +7375,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7808,14 +7393,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7825,11 +7410,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7841,8 +7426,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7854,7 +7437,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7876,17 +7458,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7897,14 +7478,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7914,11 +7495,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7930,10 +7511,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -7951,21 +7528,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7975,11 +7551,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7990,14 +7562,6 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8024,7 +7588,7 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8033,17 +7597,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8062,21 +7622,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8086,11 +7645,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8100,14 +7655,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8134,7 +7681,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8143,17 +7690,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8172,21 +7715,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8196,11 +7738,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8210,14 +7748,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8244,7 +7774,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8253,17 +7783,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8282,21 +7808,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8306,11 +7831,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8320,14 +7841,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8354,7 +7867,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8363,17 +7876,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8392,21 +7901,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8416,11 +7924,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8430,14 +7934,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8464,7 +7960,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8473,20 +7969,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -8511,7 +8007,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8519,16 +8015,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8542,13 +8037,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -8571,14 +8062,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -8590,7 +8073,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8602,18 +8085,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -8622,7 +8097,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -8631,19 +8110,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -8675,7 +8143,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8687,17 +8155,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -8726,7 +8192,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -8748,16 +8213,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8768,14 +8232,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -8785,11 +8249,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8801,8 +8265,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -8825,7 +8287,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8850,7 +8312,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8858,16 +8320,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8887,10 +8348,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -8912,14 +8369,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8930,7 +8379,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8942,18 +8391,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8961,7 +8402,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -8970,19 +8411,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9013,7 +8443,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9023,20 +8453,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -9061,7 +8493,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -9069,16 +8501,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9098,10 +8529,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -9123,14 +8550,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9141,7 +8560,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9153,18 +8572,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9172,7 +8583,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -9181,19 +8596,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9224,7 +8628,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9234,17 +8638,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -9263,21 +8665,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9287,11 +8688,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9301,14 +8698,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9335,7 +8724,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9344,17 +8733,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9368,7 +8753,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -9387,22 +8771,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9414,13 +8797,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -9428,7 +8807,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9450,16 +8828,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9470,14 +8847,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -9487,11 +8864,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9503,16 +8880,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9534,15 +8906,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9553,14 +8924,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -9570,11 +8941,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9586,8 +8957,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -9613,7 +8982,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9639,7 +9008,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9647,16 +9016,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9676,10 +9044,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -9701,14 +9065,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9721,7 +9077,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9733,18 +9089,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9757,7 +9105,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -9766,20 +9114,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9812,7 +9149,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9822,23 +9159,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9864,7 +9199,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9872,16 +9207,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9893,13 +9227,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9922,14 +9252,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9942,7 +9264,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9954,18 +9276,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9978,7 +9292,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9987,20 +9301,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10033,7 +9337,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10043,17 +9347,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10072,21 +9374,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10096,11 +9397,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10110,14 +9407,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10144,7 +9433,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10153,17 +9442,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -10173,7 +9458,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -10192,21 +9476,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10218,19 +9501,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10259,7 +9538,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10267,16 +9546,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10286,11 +9564,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -10317,14 +9591,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10335,7 +9601,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10350,18 +9616,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10372,7 +9630,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -10384,20 +9642,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10427,7 +9674,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10436,23 +9683,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10478,7 +9721,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10486,16 +9729,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10505,11 +9747,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -10534,14 +9772,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10552,7 +9782,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10565,18 +9795,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10587,7 +9809,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -10597,20 +9819,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10640,7 +9851,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10649,17 +9860,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -10671,7 +9878,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10690,21 +9896,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10716,13 +9921,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -10730,7 +9931,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10754,16 +9954,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10774,14 +9973,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10791,11 +9990,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10807,17 +10006,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10841,15 +10035,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10860,14 +10053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10877,11 +10070,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10893,8 +10086,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10906,7 +10097,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10930,17 +10120,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10951,14 +10140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10968,11 +10157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10984,10 +10173,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10998,7 +10183,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -11022,17 +10206,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11043,14 +10226,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -11060,11 +10243,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11076,15 +10259,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11113,7 +10292,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11121,16 +10300,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11140,11 +10318,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -11171,14 +10345,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11189,7 +10355,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11204,18 +10370,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11226,7 +10384,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -11238,20 +10396,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11281,7 +10428,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11290,23 +10437,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11335,7 +10478,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11343,16 +10486,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11362,11 +10504,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -11391,14 +10529,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11409,7 +10539,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11422,18 +10552,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11444,7 +10566,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -11454,20 +10576,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11497,7 +10608,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11506,23 +10617,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11551,7 +10658,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11559,16 +10666,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11578,11 +10684,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -11607,14 +10709,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11625,7 +10719,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11638,18 +10732,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11660,7 +10746,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -11670,20 +10756,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11713,7 +10788,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11722,23 +10797,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11767,7 +10838,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11775,16 +10846,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11794,11 +10864,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -11824,14 +10890,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11842,7 +10900,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11856,18 +10914,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11878,7 +10928,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11889,20 +10939,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11932,7 +10971,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11941,23 +10980,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11986,7 +11021,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11994,16 +11029,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12013,11 +11047,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -12045,14 +11075,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12063,7 +11085,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12079,18 +11101,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12101,7 +11115,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -12114,20 +11128,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12157,7 +11160,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12166,23 +11169,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12208,7 +11207,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12216,16 +11215,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12235,11 +11233,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -12264,14 +11258,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12282,7 +11268,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12295,18 +11281,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12317,7 +11295,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -12327,20 +11305,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12370,7 +11337,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12379,23 +11346,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12424,7 +11387,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12432,16 +11395,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12451,11 +11413,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12480,14 +11438,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12498,7 +11448,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12511,18 +11461,10 @@ def begin_reimage( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12533,7 +11475,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set. @@ -12543,20 +11485,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12586,7 +11517,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12595,23 +11526,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12640,7 +11567,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12648,16 +11575,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12667,11 +11593,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -12697,14 +11619,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12715,7 +11629,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12729,18 +11643,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12751,7 +11657,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12762,20 +11668,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12805,7 +11700,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12814,20 +11709,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12840,7 +11731,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12859,22 +11749,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12886,16 +11775,12 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12920,7 +11805,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12946,7 +11831,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12955,16 +11840,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12984,10 +11868,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -13015,14 +11895,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13036,7 +11908,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13052,18 +11924,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13077,7 +11941,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -13090,20 +11954,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13137,7 +11990,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13147,17 +12000,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -13176,22 +12027,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13201,11 +12051,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13220,14 +12066,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13255,7 +12093,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13264,17 +12102,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13296,7 +12130,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -13315,23 +12148,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13343,13 +12175,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13361,7 +12189,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13385,17 +12212,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13406,14 +12232,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -13423,11 +12249,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13439,12 +12265,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13481,21 +12303,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13505,11 +12326,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13519,14 +12336,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13553,7 +12362,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13562,17 +12371,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13591,21 +12396,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13615,11 +12419,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -13633,14 +12433,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13667,7 +12459,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13676,17 +12468,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -13698,7 +12486,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -13717,21 +12504,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13743,13 +12529,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -13789,22 +12571,21 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13814,11 +12595,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage( @@ -13832,14 +12609,6 @@ def begin_reimage( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13867,7 +12636,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13876,17 +12645,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13905,22 +12670,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13930,11 +12694,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -13949,14 +12709,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13984,7 +12736,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13993,17 +12745,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14022,22 +12770,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14047,11 +12794,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -14067,14 +12810,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14102,7 +12837,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14111,24 +12846,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -14154,7 +12885,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14163,16 +12894,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14192,10 +12922,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -14222,14 +12948,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14243,7 +12961,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14259,18 +12977,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14284,7 +12994,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -14297,19 +13007,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14343,7 +13042,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14353,17 +13052,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14382,22 +13079,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14407,11 +13103,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14425,14 +13117,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14460,7 +13144,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14469,17 +13153,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14493,7 +13173,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -14512,22 +13191,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14539,13 +13217,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -14559,7 +13233,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14578,22 +13251,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14605,13 +13277,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14640,7 +13308,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14664,7 +13331,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -14672,12 +13339,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14688,14 +13354,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -14705,11 +13371,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14721,10 +13387,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -14742,22 +13404,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14767,11 +13428,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -14787,14 +13444,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14822,7 +13471,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14831,17 +13480,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14860,22 +13505,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14885,11 +13529,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -14903,14 +13543,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14938,7 +13570,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14947,17 +13579,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14976,22 +13604,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15001,11 +13628,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -15019,14 +13642,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15054,7 +13669,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15063,17 +13678,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15092,22 +13703,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15117,11 +13727,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -15136,14 +13742,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15171,7 +13769,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15180,17 +13778,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15209,22 +13803,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15234,11 +13827,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -15252,14 +13841,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15287,7 +13868,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15296,24 +13877,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -15339,7 +13916,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15348,16 +13925,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15371,13 +13947,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -15403,14 +13975,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -15423,7 +13987,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15437,18 +14001,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -15461,7 +14017,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -15473,19 +14029,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.RunCommandResult] @@ -15518,7 +14063,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15530,17 +14075,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -15564,7 +14107,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15589,23 +14132,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15619,13 +14161,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -15647,14 +14185,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15664,7 +14194,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15673,18 +14203,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15694,7 +14216,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15702,19 +14224,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15746,7 +14258,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15758,20 +14270,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15796,23 +14306,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15826,13 +14335,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -15853,14 +14358,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15870,7 +14367,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15878,18 +14375,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15899,7 +14388,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15907,19 +14396,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_04_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15951,7 +14429,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15963,17 +14441,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -16002,7 +14478,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -16026,16 +14501,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16046,14 +14520,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -16063,11 +14537,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16079,10 +14553,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -16091,7 +14561,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -16110,21 +14579,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16136,13 +14604,9 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class DisksOperations: @@ -16166,7 +14630,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -16191,7 +14655,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -16199,16 +14663,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16228,10 +14691,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -16255,14 +14714,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16273,7 +14724,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16287,18 +14738,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16306,7 +14749,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -16317,19 +14760,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_04_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_04_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16360,7 +14792,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16370,20 +14802,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -16408,7 +14838,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -16416,16 +14846,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16445,10 +14874,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -16472,14 +14897,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16490,7 +14907,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16504,18 +14921,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16523,7 +14932,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -16534,19 +14943,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_04_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_04_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16577,7 +14975,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16587,17 +14985,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -16609,7 +15005,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -16628,21 +15023,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16654,13 +15048,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -16679,21 +15069,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16703,11 +15092,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16719,14 +15104,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16753,7 +15130,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16762,17 +15139,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -16780,7 +15153,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16802,16 +15174,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16822,14 +15193,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -16839,11 +15210,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16855,15 +15226,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16885,15 +15251,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16904,14 +15269,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -16921,11 +15286,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16937,13 +15302,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -16969,7 +15332,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -16977,16 +15340,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17000,13 +15362,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -17032,14 +15390,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -17050,7 +15400,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17065,18 +15415,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -17087,7 +15429,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -17099,19 +15441,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -17142,7 +15473,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17154,17 +15485,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -17183,21 +15512,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17207,11 +15535,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17223,14 +15547,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17257,7 +15573,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -17268,17 +15584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -17302,7 +15614,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -17327,7 +15639,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -17335,16 +15647,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17364,10 +15675,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -17391,14 +15698,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17409,7 +15708,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17423,18 +15722,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17442,7 +15733,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -17453,19 +15744,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17496,7 +15776,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17506,20 +15786,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -17544,7 +15826,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -17552,16 +15834,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17581,10 +15862,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -17609,14 +15886,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17627,7 +15896,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17642,18 +15911,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17661,7 +15922,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -17672,19 +15937,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_04_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17715,7 +15969,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17725,17 +15979,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -17747,7 +15999,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_04_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -17766,21 +16017,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17792,13 +16042,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -17817,21 +16063,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17841,11 +16086,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17857,14 +16098,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17891,7 +16124,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17900,17 +16133,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -17918,7 +16147,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -17940,16 +16168,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17960,14 +16187,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -17977,11 +16204,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17993,15 +16220,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -18023,15 +16245,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18042,14 +16263,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -18059,11 +16280,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18075,13 +16296,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -18107,7 +16326,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -18115,16 +16334,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18138,13 +16356,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -18170,14 +16384,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -18188,7 +16394,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18203,18 +16409,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -18225,7 +16423,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -18237,19 +16435,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_04_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -18280,7 +16467,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18292,17 +16479,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -18321,21 +16506,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18345,11 +16529,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18361,14 +16541,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18395,7 +16567,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -18406,14 +16578,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_compute_management_client.py index d410dbb20c3a..07e06f9bc10c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -127,7 +129,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -189,7 +209,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-06-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -209,7 +229,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_configuration.py index a3c2be7fa478..2e004dce05ac 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-06-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_metadata.json index d89a4632065a..ad23d29439e8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_compute_management_client.py index 79280b40babb..1793ada87acd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -128,7 +130,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -190,7 +210,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-06-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -210,7 +232,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_configuration.py index 0902ede6d81d..4fba35c0877d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-06-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/operations/_operations.py index e13bf3dc4cf5..a2e97fcfb30f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -165,10 +164,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -197,7 +192,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -221,14 +215,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -239,14 +232,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -256,11 +249,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -272,8 +265,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -316,7 +307,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -327,7 +317,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -339,11 +329,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -354,7 +343,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -364,12 +353,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -397,7 +382,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -405,16 +390,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -426,13 +410,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -455,7 +435,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -466,7 +445,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -478,11 +457,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -493,7 +471,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -503,12 +481,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -536,7 +510,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -544,16 +518,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -565,13 +538,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -583,7 +552,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -602,21 +570,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -626,11 +593,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -640,7 +603,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -659,21 +621,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -685,13 +646,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -702,7 +659,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet] @@ -725,16 +681,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -745,14 +700,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -762,11 +717,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -778,17 +733,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet] @@ -811,16 +761,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -831,14 +780,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -848,11 +797,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -864,10 +813,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -879,7 +824,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] @@ -902,17 +846,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -923,14 +866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -940,11 +883,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -956,10 +899,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1003,7 +942,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1014,7 +952,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1027,11 +965,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1042,7 +979,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1052,12 +989,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1085,7 +1018,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1093,16 +1026,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1122,10 +1054,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1148,7 +1076,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1159,7 +1086,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1172,11 +1099,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1187,7 +1113,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1197,12 +1123,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1230,7 +1153,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1238,16 +1161,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1259,13 +1181,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1277,7 +1195,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1296,21 +1213,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1320,11 +1236,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1336,7 +1248,6 @@ async def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1355,21 +1266,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1381,19 +1291,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1417,15 +1322,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1436,14 +1340,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1453,11 +1357,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1469,10 +1373,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1481,7 +1381,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1505,16 +1404,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1525,14 +1423,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1542,11 +1440,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1558,10 +1456,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -1597,7 +1491,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1616,23 +1509,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1644,13 +1536,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -1662,7 +1550,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1681,21 +1568,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1707,13 +1593,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -1741,7 +1623,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1760,7 +1641,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -1769,16 +1650,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1790,13 +1670,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -1824,7 +1700,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1850,7 +1726,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1859,16 +1735,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1888,10 +1763,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1918,14 +1789,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1939,7 +1802,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1955,18 +1818,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1980,7 +1835,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1993,20 +1848,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2040,7 +1884,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2050,24 +1894,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2093,7 +1935,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2102,16 +1944,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2123,13 +1964,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2158,14 +1995,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2179,7 +2008,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2195,18 +2024,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2220,7 +2041,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -2233,20 +2054,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2280,7 +2090,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2290,17 +2100,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -2319,22 +2127,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2344,11 +2151,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2363,14 +2166,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2398,7 +2193,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2407,17 +2202,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2439,7 +2230,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -2458,23 +2248,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2486,13 +2275,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2506,7 +2291,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2525,22 +2309,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2552,13 +2335,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -2597,7 +2376,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2616,7 +2394,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2624,16 +2402,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2645,13 +2422,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2683,7 +2456,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2702,7 +2474,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2712,16 +2484,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2733,13 +2504,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -2751,7 +2518,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2770,21 +2536,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2796,13 +2561,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -2810,7 +2571,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2829,20 +2589,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2854,13 +2613,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -2874,7 +2629,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2893,22 +2647,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2920,13 +2673,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -2956,7 +2705,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2978,16 +2726,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2998,14 +2745,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -3015,11 +2762,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3031,8 +2778,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -3061,7 +2806,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -3084,16 +2828,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3104,14 +2847,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3121,11 +2864,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3137,15 +2880,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -3171,7 +2910,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3179,16 +2918,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3202,13 +2940,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -3232,14 +2966,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3252,7 +2978,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3265,18 +2991,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3289,7 +3007,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -3300,19 +3018,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3345,7 +3053,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3357,20 +3065,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3395,7 +3105,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3403,16 +3113,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3432,10 +3141,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -3457,14 +3162,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3477,7 +3174,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3489,18 +3186,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3510,7 +3199,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -3519,19 +3212,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3564,7 +3246,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3574,20 +3256,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3612,7 +3296,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3620,16 +3304,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3649,10 +3332,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -3674,14 +3353,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3694,7 +3365,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3706,18 +3377,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3727,7 +3390,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -3736,19 +3403,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3781,7 +3437,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3791,17 +3447,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3820,21 +3474,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3844,11 +3497,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -3858,14 +3507,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3892,7 +3533,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3901,17 +3542,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -3926,7 +3563,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -3945,22 +3581,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3972,13 +3607,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -3990,7 +3621,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -4009,21 +3639,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4035,13 +3664,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4060,21 +3685,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4084,11 +3708,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -4101,14 +3721,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4135,7 +3747,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4144,17 +3756,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4173,21 +3781,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4197,11 +3804,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4212,14 +3815,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4246,7 +3841,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4255,17 +3850,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -4277,7 +3868,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4296,21 +3886,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4320,11 +3909,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -4333,7 +3918,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -4356,16 +3940,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4376,14 +3959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4393,11 +3976,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4409,16 +3992,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -4441,15 +4019,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4460,14 +4037,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4477,11 +4054,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4493,8 +4070,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4506,7 +4081,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] @@ -4529,17 +4103,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4550,14 +4123,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4567,11 +4140,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4583,10 +4156,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -4604,21 +4173,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4628,11 +4196,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4643,14 +4207,6 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4677,7 +4233,7 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4686,17 +4242,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4715,21 +4267,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4739,11 +4290,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4753,14 +4300,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4787,7 +4326,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4796,17 +4335,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4825,21 +4360,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4849,11 +4383,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4863,14 +4393,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4897,7 +4419,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4906,17 +4428,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4935,21 +4453,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4959,11 +4476,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4973,14 +4486,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5007,7 +4512,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5016,23 +4521,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -5061,7 +4562,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5069,16 +4570,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5088,11 +4588,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -5116,14 +4612,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5134,7 +4622,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5147,18 +4635,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5169,7 +4649,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -5179,19 +4659,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -5221,7 +4691,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5230,17 +4700,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5259,21 +4725,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5283,11 +4748,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -5299,14 +4760,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5333,7 +4786,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5342,20 +4795,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -5380,7 +4833,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5388,16 +4841,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5411,13 +4863,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -5440,14 +4888,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5460,7 +4900,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5472,18 +4912,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5493,7 +4925,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -5502,19 +4938,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5547,7 +4972,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5559,17 +4984,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -5599,7 +5022,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] @@ -5622,16 +5044,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5642,14 +5063,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5659,11 +5080,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5675,8 +5096,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -5699,7 +5118,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5724,7 +5143,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5732,16 +5151,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5761,10 +5179,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -5786,14 +5200,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5804,7 +5210,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5816,18 +5222,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5835,7 +5233,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -5844,19 +5242,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5887,7 +5274,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5897,20 +5284,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5935,7 +5324,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5943,16 +5332,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5972,10 +5360,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -5997,14 +5381,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6015,7 +5391,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6027,18 +5403,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6046,7 +5414,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -6055,19 +5427,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6098,7 +5459,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6108,17 +5469,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -6137,21 +5496,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6161,11 +5519,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6175,14 +5529,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6209,7 +5555,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6218,17 +5564,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6242,7 +5584,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -6261,22 +5602,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6288,13 +5628,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -6302,7 +5638,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6324,16 +5659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6344,14 +5678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6361,11 +5695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6377,16 +5711,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6408,15 +5737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6427,14 +5755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6444,11 +5772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6460,8 +5788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6487,7 +5813,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6513,7 +5839,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6521,16 +5847,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6550,10 +5875,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -6575,14 +5896,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6595,7 +5908,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6607,18 +5920,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6631,7 +5936,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6640,20 +5945,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6686,7 +5980,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6696,23 +5990,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6738,7 +6030,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6746,16 +6038,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6767,13 +6058,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6796,14 +6083,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6816,7 +6095,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6828,18 +6107,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6852,7 +6123,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6861,20 +6132,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6907,7 +6168,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6917,17 +6178,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6946,21 +6205,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6970,11 +6228,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6986,14 +6240,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7020,7 +6266,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7029,17 +6275,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7051,7 +6293,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7070,21 +6311,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7096,19 +6336,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7137,7 +6373,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7145,16 +6381,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7164,11 +6399,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -7195,14 +6426,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7213,7 +6436,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7228,18 +6451,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7250,7 +6465,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7262,20 +6477,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7305,7 +6509,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7314,23 +6518,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -7356,7 +6556,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7364,16 +6564,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7383,11 +6582,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -7412,14 +6607,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7430,7 +6617,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7443,18 +6630,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7465,7 +6644,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -7475,20 +6654,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7518,7 +6686,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7527,17 +6695,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -7549,7 +6713,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7568,21 +6731,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7594,13 +6756,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -7608,7 +6766,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7632,16 +6789,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7652,14 +6808,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7669,11 +6825,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7685,17 +6841,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7719,15 +6870,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7738,14 +6888,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7755,11 +6905,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7771,8 +6921,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7784,7 +6932,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7808,17 +6955,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7829,14 +6975,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7846,11 +6992,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7862,10 +7008,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7876,7 +7018,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7900,17 +7041,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7921,14 +7061,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7938,11 +7078,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7954,15 +7094,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7991,7 +7127,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7999,16 +7135,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8018,11 +7153,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -8049,14 +7180,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8067,7 +7190,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8082,18 +7205,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8104,7 +7219,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -8116,20 +7231,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8159,7 +7263,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8168,23 +7272,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8213,7 +7313,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8221,16 +7321,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8240,11 +7339,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -8269,14 +7364,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8287,7 +7374,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8300,18 +7387,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8322,7 +7401,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8332,20 +7411,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8375,7 +7443,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8384,23 +7452,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8429,7 +7493,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8437,16 +7501,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8456,11 +7519,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -8485,14 +7544,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8503,7 +7554,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8516,18 +7567,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8538,7 +7581,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8548,20 +7591,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8591,7 +7623,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8600,23 +7632,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8645,7 +7673,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8653,16 +7681,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8672,11 +7699,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -8702,14 +7725,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8720,7 +7735,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8734,18 +7749,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8756,7 +7763,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8767,20 +7774,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8810,7 +7806,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8819,23 +7815,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8864,7 +7856,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8872,16 +7864,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8891,11 +7882,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -8923,14 +7910,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8941,7 +7920,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8957,18 +7936,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8979,7 +7950,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8992,20 +7963,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9035,7 +7995,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9044,23 +8004,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9086,7 +8042,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9094,16 +8050,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9113,11 +8068,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -9142,14 +8093,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9160,7 +8103,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9173,18 +8116,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9195,7 +8130,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9205,20 +8140,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9248,7 +8172,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9257,23 +8181,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9302,7 +8222,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9310,16 +8230,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9329,11 +8248,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -9359,14 +8274,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9377,7 +8284,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9391,18 +8298,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9413,7 +8312,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9425,20 +8324,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9468,7 +8356,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9477,23 +8365,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9522,7 +8406,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9530,16 +8414,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9549,11 +8432,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -9579,14 +8458,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9597,7 +8468,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9611,18 +8482,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9633,7 +8496,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9644,20 +8507,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9687,7 +8539,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9696,20 +8548,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -9722,7 +8570,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9741,22 +8588,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9768,16 +8614,12 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -9802,7 +8644,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -9828,7 +8670,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -9837,16 +8679,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9866,10 +8707,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9897,14 +8734,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9918,7 +8747,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9934,18 +8763,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9959,7 +8780,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -9972,20 +8793,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10019,7 +8829,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10029,17 +8839,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10058,22 +8866,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10083,11 +8890,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10102,14 +8905,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10137,7 +8932,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10146,17 +8941,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10178,7 +8969,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10197,23 +8987,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10225,13 +9014,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10243,7 +9028,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10267,17 +9051,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10288,14 +9071,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10305,11 +9088,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10321,12 +9104,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10363,21 +9142,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10387,11 +9165,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -10403,14 +9177,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10437,7 +9203,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10446,17 +9212,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10475,21 +9237,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10499,11 +9260,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -10517,14 +9274,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10551,7 +9300,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10560,17 +9309,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10589,21 +9334,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10613,11 +9357,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -10631,14 +9371,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10665,7 +9397,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10674,17 +9406,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -10696,7 +9424,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -10715,21 +9442,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10741,13 +9467,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -10775,7 +9497,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -10804,7 +9528,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -10813,16 +9537,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10832,11 +9555,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -10864,14 +9583,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10883,7 +9594,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10898,18 +9609,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10921,7 +9624,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -10933,20 +9638,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10977,7 +9672,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10986,17 +9681,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11015,22 +9706,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11040,11 +9730,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -11059,14 +9745,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11094,7 +9772,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11103,17 +9781,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11132,22 +9806,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11157,11 +9830,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -11177,14 +9846,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11212,7 +9873,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11221,24 +9882,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -11264,7 +9921,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11273,16 +9930,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11302,10 +9958,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -11332,14 +9984,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11353,7 +9997,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11369,18 +10013,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11394,7 +10030,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -11407,19 +10043,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11453,7 +10078,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11463,17 +10088,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11492,22 +10115,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11517,11 +10139,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -11535,14 +10153,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11570,7 +10180,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11579,17 +10189,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11603,7 +10209,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -11622,22 +10227,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11649,13 +10253,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -11669,7 +10269,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11688,22 +10287,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11715,13 +10313,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11750,7 +10344,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11774,7 +10367,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -11782,12 +10375,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11798,14 +10390,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -11815,11 +10407,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11831,10 +10423,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -11852,22 +10440,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11877,11 +10464,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11897,14 +10480,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11932,7 +10507,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11941,17 +10516,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11970,22 +10541,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11995,11 +10565,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -12013,14 +10579,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12048,7 +10606,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12057,17 +10615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12086,22 +10640,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12111,11 +10664,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -12129,14 +10678,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12164,7 +10705,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12173,17 +10714,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12202,22 +10739,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12227,11 +10763,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -12246,14 +10778,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12281,7 +10805,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12290,17 +10814,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12319,22 +10839,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12344,11 +10863,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -12362,14 +10877,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12397,7 +10904,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12406,24 +10913,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -12449,7 +10952,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12458,16 +10961,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12481,13 +10983,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12513,14 +11011,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12534,7 +11024,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12548,18 +11038,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12573,7 +11055,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -12585,19 +11067,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12631,7 +11102,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12643,17 +11114,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -12677,7 +11146,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12702,23 +11171,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12732,13 +11200,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -12760,14 +11224,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12777,7 +11233,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12786,18 +11242,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12807,7 +11255,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12815,19 +11263,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12859,7 +11297,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12872,20 +11310,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12910,23 +11346,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12940,13 +11375,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -12967,14 +11398,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12984,7 +11407,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12992,18 +11415,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -13013,7 +11428,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -13021,19 +11436,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -13065,7 +11469,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13078,17 +11482,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -13117,7 +11519,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -13141,16 +11542,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13161,14 +11561,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -13178,11 +11578,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13194,10 +11594,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -13206,7 +11602,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -13225,21 +11620,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13251,13 +11645,9 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -13281,7 +11671,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -13306,7 +11696,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -13314,16 +11704,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13346,10 +11735,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -13374,14 +11759,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] @@ -13393,7 +11770,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13408,18 +11785,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] @@ -13428,7 +11797,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -13439,19 +11808,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2018_06_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2018_06_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] @@ -13483,7 +11841,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13493,17 +11851,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -13513,7 +11869,6 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -13532,21 +11887,20 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13558,13 +11912,9 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -13583,21 +11933,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13607,11 +11956,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -13621,14 +11966,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13655,7 +11992,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13664,17 +12001,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -13682,7 +12015,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -13704,16 +12036,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13724,14 +12055,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -13741,11 +12072,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13757,15 +12088,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -13787,15 +12113,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13806,14 +12131,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -13823,11 +12148,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13839,8 +12164,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -13867,7 +12190,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -13893,7 +12216,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -13902,16 +12225,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13934,10 +12256,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -13966,14 +12284,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -13986,7 +12296,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14004,18 +12314,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -14028,7 +12330,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -14043,19 +12345,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -14088,7 +12379,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14098,17 +12389,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -14123,7 +12412,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14142,22 +12430,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14169,13 +12456,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -14194,22 +12477,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14219,11 +12501,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14238,14 +12516,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14273,7 +12543,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14282,17 +12552,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -14305,7 +12571,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -14328,17 +12593,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14349,14 +12613,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -14366,11 +12630,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14382,10 +12646,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -14413,7 +12673,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -14439,7 +12699,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -14449,16 +12709,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14481,10 +12740,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -14518,14 +12773,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -14540,7 +12787,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14562,18 +12809,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -14588,7 +12827,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -14607,19 +12846,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -14654,7 +12883,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14664,17 +12893,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -14703,7 +12930,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2018_06_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -14722,7 +12948,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -14730,16 +12956,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14751,13 +12976,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -14781,23 +13002,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14807,11 +13027,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14835,14 +13051,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14871,7 +13079,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14880,17 +13088,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -14906,7 +13110,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion] @@ -14929,18 +13132,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14951,14 +13153,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -14968,11 +13170,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14984,10 +13186,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class DisksOperations: """ @@ -15010,7 +13208,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -15035,7 +13233,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -15043,16 +13241,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15072,10 +13269,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -15099,14 +13292,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15117,7 +13302,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15131,18 +13316,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15150,7 +13327,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -15161,19 +13338,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_06_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_06_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15204,7 +13370,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15214,20 +13380,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -15252,7 +13416,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -15260,16 +13424,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15289,10 +13452,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -15316,14 +13475,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15334,7 +13485,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15348,18 +13499,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15367,7 +13510,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -15378,19 +13521,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_06_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_06_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15421,7 +13553,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15431,17 +13563,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -15453,7 +13583,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -15472,21 +13601,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15498,13 +13626,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -15523,21 +13647,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15547,11 +13670,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -15563,14 +13682,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15597,7 +13708,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15606,17 +13717,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -15624,7 +13731,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15646,16 +13752,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15666,14 +13771,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -15683,11 +13788,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15699,15 +13804,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -15729,15 +13829,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15748,14 +13847,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -15765,11 +13864,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15781,13 +13880,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -15813,7 +13910,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -15821,16 +13918,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15844,13 +13940,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -15876,14 +13968,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -15895,7 +13979,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15910,18 +13994,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -15933,7 +14009,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -15945,19 +14021,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -15989,7 +14054,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16001,17 +14066,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -16030,21 +14093,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16054,11 +14116,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -16072,14 +14130,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16106,7 +14156,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -16117,17 +14167,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -16151,7 +14197,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -16176,7 +14222,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -16184,16 +14230,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16213,10 +14258,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -16240,14 +14281,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16259,7 +14292,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16273,18 +14306,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16293,7 +14318,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -16304,19 +14329,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16348,7 +14362,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16358,20 +14372,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -16396,7 +14412,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -16404,16 +14420,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16433,10 +14448,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -16461,14 +14472,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16480,7 +14483,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16495,18 +14498,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16515,7 +14510,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -16526,19 +14525,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16570,7 +14558,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16580,17 +14568,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -16602,7 +14588,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -16621,21 +14606,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16647,13 +14631,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -16672,21 +14652,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16696,11 +14675,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -16712,14 +14687,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16746,7 +14713,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16755,17 +14722,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -16773,7 +14736,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16796,16 +14758,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16816,14 +14777,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -16833,11 +14794,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16849,15 +14810,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -16880,15 +14836,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16899,14 +14854,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -16916,11 +14871,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16932,13 +14887,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -16964,7 +14917,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -16972,16 +14925,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16995,13 +14947,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -17027,14 +14975,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -17046,7 +14986,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17061,18 +15001,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -17084,7 +15016,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -17096,19 +15028,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] @@ -17140,7 +15061,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17152,17 +15073,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -17181,21 +15100,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17205,11 +15123,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -17223,14 +15137,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17257,7 +15163,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -17268,14 +15174,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_compute_management_client_enums.py index 2f01f24e2e99..1190b8d02b4f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_compute_management_client_enums.py @@ -59,10 +59,9 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): EMPTY = "Empty" ATTACH = "Attach" FROM_IMAGE = "FromImage" - IMPORT = "Import" + IMPORT_ENUM = "Import" COPY = "Copy" RESTORE = "Restore" - IMPORT_ENUM = "Import" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_models_py3.py index 9159b3ca287e..15eaae41cede 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -273,7 +269,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -331,7 +327,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -430,7 +426,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet] @@ -697,10 +693,10 @@ def __init__(self, **kwargs: Any) -> None: class CreationData(_serialization.Model): """Data used when creating a disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :vartype create_option: str or ~azure.mgmt.compute.v2018_06_01.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -740,7 +736,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", and "Restore". :paramtype create_option: str or ~azure.mgmt.compute.v2018_06_01.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in @@ -766,7 +762,7 @@ def __init__( class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -983,7 +979,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1208,7 +1204,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.Disk] @@ -1403,7 +1399,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1474,7 +1470,7 @@ def __init__( class GalleryArtifactPublishingProfileBase(_serialization.Model): """Describes the basic gallery artifact publishing profile. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1515,7 +1511,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2018_06_01.models.ManagedArtifact @@ -1630,7 +1626,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1778,7 +1774,7 @@ def __init__( class GalleryImageIdentifier(_serialization.Model): """This is the gallery Image Definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery Image Definition publisher. Required. :vartype publisher: str @@ -1818,7 +1814,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -1854,7 +1850,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1929,7 +1925,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Image Versions. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion] @@ -1967,7 +1963,7 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -2068,7 +2064,7 @@ def __init__(self, **kwargs: Any) -> None: class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.Gallery] @@ -2111,25 +2107,11 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2018_06_01.models.HostCaching """ - _validation = { - "size_in_gb": {"readonly": True}, - "host_caching": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None" and "Read". :vartype access: str or ~azure.mgmt.compute.v2018_06_01.models.AccessLevel @@ -2164,7 +2146,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -2221,7 +2203,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -2278,7 +2260,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2345,7 +2327,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -2438,7 +2420,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -2476,7 +2458,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.Image] @@ -2510,7 +2492,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -2741,12 +2723,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_06_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -2771,12 +2753,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_06_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -2921,7 +2903,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_06_01.models.SourceVault @@ -2954,7 +2936,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_06_01.models.SourceVault @@ -2987,7 +2969,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3020,7 +3002,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -3106,7 +3088,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.Usage] @@ -3140,7 +3122,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3329,7 +3311,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3454,7 +3436,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -3595,7 +3577,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -3628,7 +3610,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -3639,10 +3621,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3655,16 +3637,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -3672,10 +3654,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_06_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2018_06_01.models.VaultSecretGroup] @@ -3715,7 +3697,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -3727,10 +3709,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3743,16 +3725,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -3760,10 +3742,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_06_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -3841,7 +3823,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3922,7 +3904,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup] @@ -3960,22 +3942,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -4148,7 +4119,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4453,7 +4424,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4516,7 +4487,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -4579,7 +4550,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -4653,7 +4624,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -4700,7 +4671,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -4733,7 +4704,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.RunCommandDocumentBase] @@ -4769,7 +4740,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -4878,7 +4849,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4984,7 +4955,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.Snapshot] @@ -5159,7 +5130,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -5177,7 +5148,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -5196,12 +5167,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_06_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.DataDisk] """ @@ -5228,12 +5199,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_06_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.DataDisk] """ super().__init__(**kwargs) @@ -5268,7 +5239,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -5302,7 +5273,7 @@ def __init__(self, *, name: str, regional_replica_count: Optional[int] = None, * class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5319,57 +5290,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -5405,7 +5325,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5555,7 +5475,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -5765,7 +5685,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5807,10 +5727,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -5828,7 +5748,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -5924,10 +5844,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -5941,7 +5861,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -6008,7 +5928,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -6098,7 +6018,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6216,7 +6136,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -6260,7 +6180,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6541,7 +6461,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_06_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_06_01.models.UserAssignedIdentitiesValue] """ @@ -6573,7 +6493,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2018_06_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_06_01.models.UserAssignedIdentitiesValue] """ @@ -6587,7 +6507,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6643,7 +6563,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6831,7 +6751,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -6892,7 +6812,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7051,7 +6971,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -7247,10 +7167,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension] @@ -7326,7 +7246,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_06_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -7365,7 +7285,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -7376,7 +7296,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -7441,7 +7361,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7468,7 +7388,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7622,10 +7542,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -7667,7 +7587,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet] @@ -7703,7 +7623,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetSku] @@ -7739,7 +7659,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet] @@ -7772,7 +7692,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -7802,10 +7722,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7892,7 +7812,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -7917,7 +7837,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2018_06_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -7942,7 +7862,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2018_06_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -7956,7 +7876,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -8096,10 +8016,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -8112,16 +8032,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -8129,10 +8049,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_06_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -8175,10 +8095,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -8191,16 +8111,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -8208,10 +8128,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_06_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -8227,10 +8147,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -8291,10 +8211,12 @@ def __init__( self.public_ip_prefix = public_ip_prefix -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -8321,7 +8243,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -8329,18 +8251,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -8459,12 +8369,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetDataDisk] """ @@ -8492,12 +8402,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetDataDisk] """ @@ -8603,7 +8513,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -8719,7 +8629,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -8804,7 +8714,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar network_interface_configurations: The list of network configurations. @@ -8955,7 +8865,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -8999,7 +8909,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -9113,7 +9023,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9166,10 +9076,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -9181,7 +9091,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -9273,10 +9183,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -9286,7 +9196,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -9310,7 +9220,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -9363,10 +9273,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -9502,7 +9412,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM] @@ -9565,7 +9475,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -9634,7 +9544,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -9806,10 +9716,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -9827,7 +9737,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -9910,10 +9820,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_06_01.models.SubResource @@ -9927,7 +9837,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/operations/_operations.py index b3a44cf50f84..55f29daab062 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_06_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +401,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -432,7 +429,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -462,7 +459,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -495,7 +492,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -526,7 +523,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -572,7 +569,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -607,7 +604,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -642,7 +639,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -668,7 +665,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -708,7 +705,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -775,7 +772,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -823,7 +820,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -854,7 +851,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -883,7 +880,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +939,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1004,7 +1003,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1134,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1165,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1191,7 +1190,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1216,7 +1215,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1293,7 +1292,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1455,7 +1454,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1480,7 +1479,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1667,7 +1666,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1721,7 +1720,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1755,7 +1754,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1789,7 +1788,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1814,7 +1813,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1845,7 +1844,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1876,7 +1875,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1907,7 +1906,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1938,7 +1937,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1968,7 +1967,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1994,7 +1995,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2025,7 +2026,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2056,7 +2057,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2087,7 +2088,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2118,7 +2119,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2149,7 +2150,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2180,7 +2181,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2211,7 +2212,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2242,7 +2243,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2273,7 +2274,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2304,7 +2305,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2341,7 +2342,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2376,7 +2377,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2402,7 +2403,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2442,7 +2443,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2473,7 +2474,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2498,7 +2499,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2523,7 +2524,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2548,7 +2549,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2579,7 +2580,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2611,7 +2612,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2637,7 +2638,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2663,7 +2664,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2698,7 +2699,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2724,7 +2725,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2756,7 +2757,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2788,7 +2789,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2834,7 +2835,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2860,7 +2861,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2886,7 +2887,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2912,7 +2913,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2938,7 +2939,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2964,7 +2965,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2999,7 +3000,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3032,7 +3033,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3065,7 +3066,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3092,7 +3095,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3219,7 +3222,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3273,7 +3276,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3372,7 +3375,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3403,7 +3406,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -3486,7 +3489,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -3524,7 +3527,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3680,7 +3683,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3917,7 +3920,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4054,7 +4057,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -4078,14 +4080,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4096,14 +4097,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -4113,11 +4114,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4129,8 +4130,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -4173,7 +4172,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4184,7 +4182,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4196,11 +4194,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4211,7 +4208,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -4221,12 +4218,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4254,7 +4247,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4262,16 +4255,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4283,13 +4275,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -4312,7 +4300,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4323,7 +4310,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4335,11 +4322,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4350,7 +4336,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -4360,12 +4346,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4393,7 +4375,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4401,16 +4383,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4422,13 +4403,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -4440,7 +4417,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4459,21 +4435,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4483,11 +4458,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -4497,7 +4468,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4516,21 +4486,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4542,13 +4511,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -4559,7 +4524,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4581,16 +4545,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4601,14 +4564,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4618,11 +4581,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4634,17 +4597,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4666,16 +4624,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4686,14 +4643,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4703,11 +4660,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4719,10 +4676,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -4734,7 +4687,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -4756,17 +4708,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4777,14 +4728,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4794,11 +4745,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4810,10 +4761,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -4857,7 +4804,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4868,7 +4814,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4881,11 +4827,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4896,7 +4841,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -4906,12 +4851,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4939,7 +4880,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4947,16 +4888,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4976,10 +4916,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -5002,7 +4938,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5013,7 +4948,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5026,11 +4961,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5041,7 +4975,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -5051,12 +4985,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5084,7 +5015,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5092,16 +5023,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5113,13 +5043,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5131,7 +5057,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5150,21 +5075,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5174,11 +5098,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -5190,7 +5110,6 @@ def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5209,21 +5128,20 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5235,19 +5153,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5271,15 +5184,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5290,14 +5202,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5307,11 +5219,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5323,10 +5235,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -5335,7 +5243,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5359,16 +5266,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5379,14 +5285,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5396,11 +5302,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5412,10 +5318,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -5451,7 +5353,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5470,23 +5371,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5498,13 +5398,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -5516,7 +5412,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5535,21 +5430,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5561,13 +5455,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -5595,7 +5485,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5614,7 +5503,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -5623,16 +5512,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5644,13 +5532,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -5678,7 +5562,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -5704,7 +5588,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -5713,16 +5597,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5742,10 +5625,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -5772,14 +5651,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5793,7 +5664,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5809,18 +5680,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5834,7 +5697,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -5847,20 +5710,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5894,7 +5746,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5904,24 +5756,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -5947,7 +5797,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -5956,16 +5806,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5977,13 +5826,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6012,14 +5857,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -6033,7 +5870,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6049,18 +5886,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -6074,7 +5903,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -6087,20 +5916,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -6134,7 +5952,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6144,17 +5962,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -6173,22 +5989,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6198,11 +6013,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6217,14 +6028,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6252,7 +6055,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6261,17 +6064,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6293,7 +6092,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6312,23 +6110,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6340,13 +6137,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6360,7 +6153,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6379,22 +6171,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6406,13 +6197,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -6451,7 +6238,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6470,7 +6256,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6478,16 +6264,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6499,13 +6284,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6537,7 +6318,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6556,7 +6336,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -6566,16 +6346,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6587,13 +6366,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -6605,7 +6380,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6624,21 +6398,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6650,13 +6423,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -6664,7 +6433,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6683,20 +6451,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6708,13 +6475,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -6728,7 +6491,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -6747,22 +6509,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6774,13 +6535,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -6810,7 +6567,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6832,16 +6588,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6852,14 +6607,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6869,11 +6624,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6885,8 +6640,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -6915,7 +6668,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -6937,16 +6689,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6957,14 +6708,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6974,11 +6725,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6990,15 +6741,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -7024,7 +6771,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7032,16 +6779,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7055,13 +6801,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -7085,14 +6827,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -7105,7 +6839,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7118,18 +6852,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -7142,7 +6868,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -7153,19 +6879,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -7198,7 +6914,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7210,20 +6926,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -7248,7 +6966,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7256,16 +6974,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7285,10 +7002,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -7310,14 +7023,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7329,7 +7034,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7341,18 +7046,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7361,7 +7058,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -7370,19 +7071,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7414,7 +7104,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7424,20 +7114,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -7462,7 +7154,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7470,16 +7162,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7499,10 +7190,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -7524,14 +7211,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7543,7 +7222,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7555,18 +7234,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7575,7 +7246,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -7584,19 +7259,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] @@ -7628,7 +7292,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7638,17 +7302,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7667,21 +7329,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7691,11 +7352,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7705,14 +7362,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7739,7 +7388,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7748,17 +7397,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7773,7 +7418,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -7792,22 +7436,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7819,13 +7462,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -7837,7 +7476,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7856,21 +7494,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7882,13 +7519,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7907,21 +7540,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7931,11 +7563,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7946,14 +7574,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7980,7 +7600,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7989,17 +7609,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8018,21 +7634,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8042,11 +7657,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8057,14 +7668,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8091,7 +7694,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8100,17 +7703,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -8122,7 +7721,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8141,21 +7739,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8165,11 +7762,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -8178,7 +7771,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -8200,16 +7792,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8220,14 +7811,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -8237,11 +7828,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8253,16 +7844,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -8284,15 +7870,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8303,14 +7888,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -8320,11 +7905,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8336,8 +7921,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8349,7 +7932,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -8371,17 +7953,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8392,14 +7973,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -8409,11 +7990,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8425,10 +8006,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -8446,21 +8023,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8470,11 +8046,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8485,14 +8057,6 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8519,7 +8083,7 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8528,17 +8092,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8557,21 +8117,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8581,11 +8140,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8595,14 +8150,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8629,7 +8176,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8638,17 +8185,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8667,21 +8210,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8691,11 +8233,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8705,14 +8243,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8739,7 +8269,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8748,17 +8278,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8777,21 +8303,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8801,11 +8326,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8815,14 +8336,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8849,7 +8362,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8858,23 +8371,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8903,7 +8412,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8911,16 +8420,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8930,11 +8438,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -8958,14 +8462,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8976,7 +8472,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8989,18 +8485,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9011,7 +8499,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -9021,19 +8509,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9063,7 +8541,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9072,17 +8550,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -9101,21 +8575,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9125,11 +8598,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9139,14 +8608,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9173,7 +8634,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9182,20 +8643,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -9220,7 +8681,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9228,16 +8689,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9251,13 +8711,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -9280,14 +8736,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -9299,7 +8747,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9311,18 +8759,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -9331,7 +8771,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -9340,19 +8784,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -9384,7 +8817,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9396,17 +8829,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -9436,7 +8867,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -9458,16 +8888,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9478,14 +8907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -9495,11 +8924,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9511,8 +8940,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -9535,7 +8962,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -9560,7 +8987,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -9568,16 +8995,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9597,10 +9023,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -9622,14 +9044,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9640,7 +9054,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9652,18 +9066,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9671,7 +9077,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -9680,19 +9086,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9723,7 +9118,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9733,20 +9128,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -9771,7 +9168,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -9779,16 +9176,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9808,10 +9204,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -9833,14 +9225,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9851,7 +9235,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9863,18 +9247,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9882,7 +9258,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -9891,19 +9271,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9934,7 +9303,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9944,17 +9313,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -9973,21 +9340,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9997,11 +9363,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10011,14 +9373,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10045,7 +9399,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10054,17 +9408,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10078,7 +9428,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -10097,22 +9446,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10124,13 +9472,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -10138,7 +9482,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -10160,16 +9503,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10180,14 +9522,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -10197,11 +9539,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10213,16 +9555,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -10244,15 +9581,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10263,14 +9599,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -10280,11 +9616,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10296,8 +9632,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -10323,7 +9657,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -10349,7 +9683,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10357,16 +9691,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10386,10 +9719,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -10411,14 +9740,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10431,7 +9752,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10443,18 +9764,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10467,7 +9780,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -10476,20 +9789,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10522,7 +9824,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10532,23 +9834,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -10574,7 +9874,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10582,16 +9882,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10603,13 +9902,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -10632,14 +9927,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10652,7 +9939,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10664,18 +9951,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10688,7 +9967,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -10697,20 +9976,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10743,7 +10012,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10753,17 +10022,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10782,21 +10049,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10806,11 +10072,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10820,14 +10082,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10854,7 +10108,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10863,17 +10117,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -10883,7 +10133,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -10902,21 +10151,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10928,19 +10176,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10969,7 +10213,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10977,16 +10221,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10996,11 +10239,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -11027,14 +10266,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11045,7 +10276,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11060,18 +10291,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11082,7 +10305,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -11094,20 +10317,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11137,7 +10349,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11146,23 +10358,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11188,7 +10396,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11196,16 +10404,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11215,11 +10422,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -11244,14 +10447,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11262,7 +10457,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11275,18 +10470,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11297,7 +10484,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -11307,20 +10494,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11350,7 +10526,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11359,17 +10535,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -11381,7 +10553,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11400,21 +10571,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11426,13 +10596,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -11440,7 +10606,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -11464,16 +10629,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11484,14 +10648,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -11501,11 +10665,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11517,17 +10681,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -11551,15 +10710,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11570,14 +10728,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -11587,11 +10745,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11603,8 +10761,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11616,7 +10772,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -11640,17 +10795,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11661,14 +10815,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -11678,11 +10832,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11694,10 +10848,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11708,7 +10858,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -11732,17 +10881,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11753,14 +10901,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -11770,11 +10918,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11786,15 +10934,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11823,7 +10967,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11831,16 +10975,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11850,11 +10993,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -11881,14 +11020,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11899,7 +11030,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11914,18 +11045,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11936,7 +11059,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -11948,20 +11071,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11991,7 +11103,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12000,23 +11112,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12045,7 +11153,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12053,16 +11161,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12072,11 +11179,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -12101,14 +11204,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12119,7 +11214,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12132,18 +11227,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12154,7 +11241,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -12164,20 +11251,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12207,7 +11283,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12216,23 +11292,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12261,7 +11333,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12269,16 +11341,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12288,11 +11359,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -12317,14 +11384,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12335,7 +11394,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12348,18 +11407,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12370,7 +11421,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -12380,20 +11431,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12423,7 +11463,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12432,23 +11472,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12477,7 +11513,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12485,16 +11521,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12504,11 +11539,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -12534,14 +11565,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12552,7 +11575,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12566,18 +11589,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12588,7 +11603,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -12599,20 +11614,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12642,7 +11646,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12651,23 +11655,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12696,7 +11696,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12704,16 +11704,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12723,11 +11722,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -12755,14 +11750,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12773,7 +11760,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12789,18 +11776,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12811,7 +11790,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -12824,20 +11803,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12867,7 +11835,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12876,23 +11844,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12918,7 +11882,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12926,16 +11890,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12945,11 +11908,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -12974,14 +11933,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12992,7 +11943,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13005,18 +11956,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13027,7 +11970,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -13037,20 +11980,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13080,7 +12012,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13089,23 +12021,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13134,7 +12062,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13142,16 +12070,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13161,11 +12088,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -13191,14 +12114,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13209,7 +12124,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13223,18 +12138,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13245,7 +12152,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -13257,20 +12164,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13300,7 +12196,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13309,23 +12205,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13354,7 +12246,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13362,16 +12254,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13381,11 +12272,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -13411,14 +12298,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13429,7 +12308,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13443,18 +12322,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13465,7 +12336,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -13476,20 +12347,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13519,7 +12379,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13528,20 +12388,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -13554,7 +12410,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -13573,22 +12428,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13600,16 +12454,12 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13634,7 +12484,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -13660,7 +12510,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -13669,16 +12519,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13698,10 +12547,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -13729,14 +12574,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13750,7 +12587,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13766,18 +12603,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13791,7 +12620,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -13804,20 +12633,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13851,7 +12669,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13861,17 +12679,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -13890,22 +12706,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13915,11 +12730,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13934,14 +12745,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13969,7 +12772,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13978,17 +12781,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14010,7 +12809,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14029,23 +12827,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14057,13 +12854,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14075,7 +12868,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -14099,17 +12891,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14120,14 +12911,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -14137,11 +12928,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14153,12 +12944,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -14195,21 +12982,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14219,11 +13005,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14233,14 +13015,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14267,7 +13041,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14276,17 +13050,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14305,21 +13075,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14329,11 +13098,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -14347,14 +13112,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14381,7 +13138,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14390,17 +13147,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14419,21 +13172,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14443,11 +13195,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -14461,14 +13209,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14495,7 +13235,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14504,17 +13244,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -14526,7 +13262,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -14545,21 +13280,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14571,13 +13305,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -14605,7 +13335,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -14634,7 +13366,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14643,16 +13375,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14662,11 +13393,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -14694,14 +13421,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14713,7 +13432,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14728,18 +13447,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14751,7 +13462,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -14763,20 +13476,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14807,7 +13510,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14816,17 +13519,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14845,22 +13544,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14870,11 +13568,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -14889,14 +13583,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14924,7 +13610,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14933,17 +13619,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14962,22 +13644,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14987,11 +13668,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -15007,14 +13684,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15042,7 +13711,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15051,24 +13720,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -15094,7 +13759,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15103,16 +13768,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15132,10 +13796,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -15162,14 +13822,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15183,7 +13835,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15199,18 +13851,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15224,7 +13868,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -15237,19 +13881,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15283,7 +13916,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15293,17 +13926,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15322,22 +13953,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15347,11 +13977,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15365,14 +13991,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15400,7 +14018,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15409,17 +14027,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15433,7 +14047,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -15452,22 +14065,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15479,13 +14091,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -15499,7 +14107,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15518,22 +14125,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15545,13 +14151,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15580,7 +14182,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15604,7 +14205,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -15612,12 +14213,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15628,14 +14228,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -15645,11 +14245,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15661,10 +14261,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -15682,22 +14278,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15707,11 +14302,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -15727,14 +14318,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15762,7 +14345,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15771,17 +14354,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15800,22 +14379,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15825,11 +14403,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -15843,14 +14417,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15878,7 +14444,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15887,17 +14453,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15916,22 +14478,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15941,11 +14502,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -15959,14 +14516,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15994,7 +14543,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16003,17 +14552,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -16032,22 +14577,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16057,11 +14601,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -16076,14 +14616,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16111,7 +14643,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16120,17 +14652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -16149,22 +14677,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16174,11 +14701,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -16192,14 +14715,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16227,7 +14742,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16236,24 +14751,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -16279,7 +14790,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -16288,16 +14799,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16311,13 +14821,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -16343,14 +14849,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -16363,7 +14861,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16377,18 +14875,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -16401,7 +14891,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -16413,19 +14903,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.RunCommandResult] @@ -16458,7 +14937,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16470,17 +14949,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -16504,7 +14981,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16529,23 +15006,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16559,13 +15035,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -16587,14 +15059,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16604,7 +15068,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16613,18 +15077,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16634,7 +15090,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16642,19 +15098,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16686,7 +15132,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16698,20 +15144,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16736,23 +15180,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16766,13 +15209,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -16793,14 +15232,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16810,7 +15241,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -16818,18 +15249,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16839,7 +15262,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -16847,19 +15270,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_06_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16891,7 +15303,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16903,17 +15315,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -16942,7 +15352,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -16966,16 +15375,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16986,14 +15394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -17003,11 +15411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17019,10 +15427,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -17031,7 +15435,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -17050,21 +15453,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17076,13 +15478,9 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -17106,7 +15504,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -17131,7 +15529,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -17139,16 +15537,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17171,10 +15568,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -17199,14 +15592,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17217,7 +15602,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17232,18 +15617,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17251,7 +15628,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -17262,19 +15639,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2018_06_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2018_06_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17305,7 +15671,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17315,17 +15681,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -17335,7 +15699,6 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -17354,21 +15717,20 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17380,13 +15742,9 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -17405,21 +15763,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17429,11 +15786,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17443,14 +15796,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17477,7 +15822,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17486,17 +15831,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -17504,7 +15845,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17526,16 +15866,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17546,14 +15885,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -17563,11 +15902,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17579,15 +15918,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17609,15 +15943,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17628,14 +15961,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -17645,11 +15978,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17661,8 +15994,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -17689,7 +16020,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -17715,7 +16046,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -17724,16 +16055,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17756,10 +16086,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -17788,14 +16114,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -17808,7 +16126,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17826,18 +16144,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -17850,7 +16160,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -17865,19 +16175,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] @@ -17910,7 +16209,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17920,17 +16219,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -17945,7 +16242,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -17964,22 +16260,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17991,13 +16286,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -18016,22 +16307,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18041,11 +16331,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18060,14 +16346,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18095,7 +16373,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18104,17 +16382,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -18127,7 +16401,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -18149,17 +16422,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18170,14 +16442,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -18187,11 +16459,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18203,10 +16475,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -18234,7 +16502,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -18260,7 +16528,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -18270,16 +16538,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18302,10 +16569,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -18339,14 +16602,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18361,7 +16616,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18383,18 +16638,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18409,7 +16656,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -18428,19 +16675,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18475,7 +16712,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18485,17 +16722,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -18524,7 +16759,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2018_06_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -18543,7 +16777,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -18551,16 +16785,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18572,13 +16805,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -18602,23 +16831,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18628,11 +16856,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18656,14 +16880,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18692,7 +16908,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18701,17 +16917,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -18727,7 +16939,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.GalleryImageVersion] @@ -18750,18 +16961,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18772,14 +16982,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -18789,11 +16999,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18805,10 +17015,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class DisksOperations: """ @@ -18831,7 +17037,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -18856,7 +17062,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -18864,16 +17070,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18893,10 +17098,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -18920,14 +17121,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18938,7 +17131,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18952,18 +17145,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18971,7 +17156,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -18982,19 +17167,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_06_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_06_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19025,7 +17199,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19035,20 +17209,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -19073,7 +17245,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -19081,16 +17253,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19110,10 +17281,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -19137,14 +17304,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19155,7 +17314,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19169,18 +17328,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19188,7 +17339,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -19199,19 +17350,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_06_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_06_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19242,7 +17382,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19252,17 +17392,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -19274,7 +17412,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -19293,21 +17430,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19319,13 +17455,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -19344,21 +17476,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19368,11 +17499,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19384,14 +17511,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19418,7 +17537,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19427,17 +17546,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -19445,7 +17560,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19467,16 +17581,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19487,14 +17600,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -19504,11 +17617,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19520,15 +17633,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -19550,15 +17658,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19569,14 +17676,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -19586,11 +17693,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19602,13 +17709,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -19634,7 +17739,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -19642,16 +17747,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19665,13 +17769,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -19697,14 +17797,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -19715,7 +17807,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19730,18 +17822,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -19752,7 +17836,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -19764,19 +17848,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -19807,7 +17880,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19819,17 +17892,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -19848,21 +17919,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19872,11 +17942,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19888,14 +17954,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19922,7 +17980,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -19933,17 +17991,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -19967,7 +18021,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -19992,7 +18046,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -20000,16 +18054,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20029,10 +18082,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -20056,14 +18105,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20074,7 +18115,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20088,18 +18129,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20107,7 +18140,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -20118,19 +18151,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20161,7 +18183,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20171,20 +18193,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -20209,7 +18233,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -20217,16 +18241,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20246,10 +18269,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -20274,14 +18293,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20292,7 +18303,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20307,18 +18318,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20326,7 +18329,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -20337,19 +18344,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_06_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20380,7 +18376,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20390,17 +18386,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -20412,7 +18406,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_06_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -20431,21 +18424,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20457,13 +18449,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -20482,21 +18470,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20506,11 +18493,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -20522,14 +18505,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20556,7 +18531,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20565,17 +18540,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -20583,7 +18554,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20605,16 +18575,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20625,14 +18594,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -20642,11 +18611,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20658,15 +18627,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_06_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -20688,15 +18652,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20707,14 +18670,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -20724,11 +18687,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20740,13 +18703,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -20772,7 +18733,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -20780,16 +18741,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20803,13 +18763,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -20835,14 +18791,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -20853,7 +18801,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20868,18 +18816,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -20890,7 +18830,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -20902,19 +18842,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_06_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_06_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -20945,7 +18874,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20957,17 +18886,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -20986,21 +18913,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21010,11 +18936,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -21026,14 +18948,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21060,7 +18974,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -21071,14 +18985,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_compute_management_client.py index 3da053892e37..a10a8c6dfde1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -53,7 +55,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -64,7 +84,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-09-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -84,7 +104,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_configuration.py index e31b117d171f..2e93434093a3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-09-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_metadata.json index c170577f6ea1..0bca69be66d1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_compute_management_client.py index 5599edb960d5..ba971e9424ef 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -53,7 +55,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -64,7 +84,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-09-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -84,7 +106,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_configuration.py index d072e089ba58..18c4fd0fb5b1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-09-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/operations/_operations.py index aa510efb2cf5..6fdc43a87667 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -75,7 +75,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -100,7 +100,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -108,16 +108,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -137,10 +136,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -164,14 +159,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -182,7 +169,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -196,18 +183,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -215,7 +194,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -226,19 +205,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_09_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_09_30.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -269,7 +237,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -279,20 +247,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -317,7 +283,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -325,16 +291,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -354,10 +319,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -381,14 +342,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -399,7 +352,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -413,18 +366,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -432,7 +377,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -443,19 +388,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_09_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_09_30.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -486,7 +420,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -496,17 +430,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -518,7 +450,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_09_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -537,21 +468,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -563,13 +493,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -588,21 +514,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -612,11 +537,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -628,14 +549,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -662,7 +575,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -671,17 +584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -689,7 +598,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -711,16 +619,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -731,14 +638,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -748,11 +655,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -764,15 +671,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -794,15 +696,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -813,14 +714,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -830,11 +731,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -846,13 +747,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -878,7 +777,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -886,16 +785,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -909,13 +807,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -941,14 +835,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -960,7 +846,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -975,18 +861,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -998,7 +876,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1010,19 +888,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -1054,7 +921,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1066,17 +933,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1095,21 +960,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1119,11 +983,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1137,14 +997,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1171,7 +1023,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1182,17 +1034,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1216,7 +1064,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1241,7 +1089,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1249,16 +1097,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1278,10 +1125,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1305,14 +1148,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1324,7 +1159,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1338,18 +1173,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1358,7 +1185,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1369,19 +1196,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1413,7 +1229,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1423,20 +1239,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1461,7 +1279,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1469,16 +1287,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1498,10 +1315,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1526,14 +1339,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1545,7 +1350,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1560,18 +1365,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1580,7 +1377,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1591,19 +1392,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1635,7 +1425,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1645,17 +1435,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1667,7 +1455,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_09_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1686,21 +1473,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1712,13 +1498,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1737,21 +1519,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,11 +1542,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1777,14 +1554,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1811,7 +1580,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1820,17 +1589,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1838,7 +1603,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1861,16 +1625,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1881,14 +1644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1898,11 +1661,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1914,15 +1677,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_09_30.models.Snapshot] @@ -1945,15 +1703,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1964,14 +1721,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1981,11 +1738,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1997,13 +1754,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2029,7 +1784,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2037,16 +1792,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2060,13 +1814,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2092,14 +1842,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -2111,7 +1853,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2126,18 +1868,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -2149,7 +1883,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2161,19 +1895,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] @@ -2205,7 +1928,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2217,17 +1940,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2246,21 +1967,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2270,11 +1990,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2288,14 +2004,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2322,7 +2030,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2333,14 +2041,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_compute_management_client_enums.py index 0e6723c88d9a..52629423164e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_compute_management_client_enums.py @@ -24,11 +24,10 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): EMPTY = "Empty" ATTACH = "Attach" FROM_IMAGE = "FromImage" - IMPORT = "Import" + IMPORT_ENUM = "Import" COPY = "Copy" RESTORE = "Restore" UPLOAD = "Upload" - IMPORT_ENUM = "Import" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_models_py3.py index 08f0d2dcf2a2..a70dfab68fe4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/models/_models_py3.py @@ -42,11 +42,10 @@ def __init__(self, **kwargs: Any) -> None: class CreationData(_serialization.Model): """Data used when creating a disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2018_09_30.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -86,8 +85,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2018_09_30.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in @@ -115,7 +113,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -164,7 +162,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -321,7 +319,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2018_09_30.models.Disk] @@ -475,7 +473,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -556,7 +554,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2018_09_30.models.AccessLevel @@ -589,7 +587,7 @@ def __init__(self, *, access: Union[str, "_models.AccessLevel"], duration_in_sec class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -628,7 +626,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_09_30.models.SourceVault @@ -661,7 +659,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2018_09_30.models.SourceVault @@ -696,7 +694,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -818,7 +816,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2018_09_30.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/operations/_operations.py index 162d7a161a63..d7a843610df2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_09_30/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -534,7 +534,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -559,7 +559,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -567,16 +567,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -596,10 +595,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -623,14 +618,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -641,7 +628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -655,18 +642,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -674,7 +653,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -685,19 +664,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_09_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_09_30.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -728,7 +696,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -738,20 +706,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -776,7 +742,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -784,16 +750,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -813,10 +778,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -840,14 +801,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -858,7 +811,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -872,18 +825,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -891,7 +836,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -902,19 +847,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2018_09_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2018_09_30.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -945,7 +879,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -955,17 +889,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -977,7 +909,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_09_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -996,21 +927,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1022,13 +952,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1047,21 +973,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1071,11 +996,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1087,14 +1008,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1121,7 +1034,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1130,17 +1043,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1148,7 +1057,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1170,16 +1078,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1190,14 +1097,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1207,11 +1114,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1223,15 +1130,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1253,15 +1155,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1272,14 +1173,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1289,11 +1190,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1305,13 +1206,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -1337,7 +1236,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1345,16 +1244,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1368,13 +1266,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -1400,14 +1294,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1418,7 +1304,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1433,18 +1319,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1455,7 +1333,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1467,19 +1345,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1510,7 +1377,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1522,17 +1389,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1551,21 +1416,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1575,11 +1439,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1591,14 +1451,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1625,7 +1477,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -1636,17 +1488,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1670,7 +1518,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1695,7 +1543,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1703,16 +1551,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1732,10 +1579,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -1759,14 +1602,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1777,7 +1612,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1791,18 +1626,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1810,7 +1637,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1821,19 +1648,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1864,7 +1680,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1874,20 +1690,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1912,7 +1730,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1920,16 +1738,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1949,10 +1766,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -1977,14 +1790,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1995,7 +1800,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2010,18 +1815,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2029,7 +1826,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2040,19 +1841,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2018_09_30.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2083,7 +1873,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2093,17 +1883,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2115,7 +1903,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_09_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2134,21 +1921,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2160,13 +1946,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2185,21 +1967,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2209,11 +1990,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2225,14 +2002,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2259,7 +2028,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2268,17 +2037,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -2286,7 +2051,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2308,16 +2072,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2328,14 +2091,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2345,11 +2108,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2361,15 +2124,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2391,15 +2149,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2410,14 +2167,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2427,11 +2184,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2443,13 +2200,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2475,7 +2230,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2483,16 +2238,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2506,13 +2260,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2538,14 +2288,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2556,7 +2298,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2571,18 +2313,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2593,7 +2327,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2605,19 +2339,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2018_09_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2648,7 +2371,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2660,17 +2383,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2689,21 +2410,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2713,11 +2433,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2729,14 +2445,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2763,7 +2471,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2774,14 +2482,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_compute_management_client.py index f258c77bafa9..2604177874bc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -111,7 +113,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -160,7 +180,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-10-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -180,7 +200,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_configuration.py index d40d1e1989cc..5b7907207829 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-10-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_metadata.json index 4c4db493cef2..28033ebb300d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_compute_management_client.py index c0d1344f491a..025671cb130b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -112,7 +114,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -161,7 +181,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2018-10-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -181,7 +203,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_configuration.py index 29a3be69ac12..3e63e63b0294 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2018-10-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/operations/_operations.py index b2600ebf857e..9ddeb727133d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -136,10 +135,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -168,7 +163,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -192,14 +186,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -210,14 +203,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -227,11 +220,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -243,8 +236,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -287,7 +278,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -298,7 +288,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -310,11 +300,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -325,7 +314,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -335,12 +324,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -368,7 +353,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -376,16 +361,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -397,13 +381,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -426,7 +406,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -437,7 +416,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -449,11 +428,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -464,7 +442,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -474,12 +452,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -507,7 +481,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -515,16 +489,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -536,13 +509,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -554,7 +523,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -573,21 +541,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -597,11 +564,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -611,7 +574,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -630,21 +592,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -656,13 +617,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -673,7 +630,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet] @@ -696,16 +652,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -716,14 +671,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -733,11 +688,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -749,17 +704,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet] @@ -782,16 +732,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -802,14 +751,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -819,11 +768,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -835,10 +784,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -850,7 +795,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] @@ -873,17 +817,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -894,14 +837,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -911,11 +854,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -927,10 +870,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -974,7 +913,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -985,7 +923,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -998,11 +936,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1013,7 +950,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1023,12 +960,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1056,7 +989,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1064,16 +997,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1093,10 +1025,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1119,7 +1047,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1130,7 +1057,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1143,11 +1070,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1158,7 +1084,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1168,12 +1094,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1201,7 +1124,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1209,16 +1132,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1230,13 +1152,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1248,7 +1166,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1267,21 +1184,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1291,11 +1207,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1307,7 +1219,6 @@ async def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1326,21 +1237,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1352,19 +1262,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1388,15 +1293,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1407,14 +1311,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1424,11 +1328,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1440,10 +1344,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1452,7 +1352,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1476,16 +1375,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1496,14 +1394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1513,11 +1411,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1529,10 +1427,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -1568,7 +1462,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1587,23 +1480,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1615,13 +1507,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -1633,7 +1521,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1652,21 +1539,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1678,13 +1564,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -1712,7 +1594,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -1731,7 +1612,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -1740,16 +1621,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1641,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -1795,7 +1671,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -1821,7 +1697,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -1830,16 +1706,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1859,10 +1734,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -1889,14 +1760,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1910,7 +1773,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1926,18 +1789,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -1951,7 +1806,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -1964,20 +1819,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2011,7 +1855,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2021,24 +1865,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -2064,7 +1906,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -2073,16 +1915,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2094,13 +1935,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2129,14 +1966,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2150,7 +1979,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2166,18 +1995,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2191,7 +2012,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -2204,20 +2025,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -2251,7 +2061,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2071,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -2290,22 +2098,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2315,11 +2122,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2334,14 +2137,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2369,7 +2164,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2378,17 +2173,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2410,7 +2201,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -2429,23 +2219,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2457,13 +2246,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2477,7 +2262,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2496,22 +2280,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2523,13 +2306,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -2568,7 +2347,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2587,7 +2365,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2595,16 +2373,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2616,13 +2393,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -2654,7 +2427,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2673,7 +2445,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -2683,16 +2455,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2704,13 +2475,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -2722,7 +2489,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2741,21 +2507,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2767,13 +2532,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -2781,7 +2542,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2800,20 +2560,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2825,13 +2584,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -2845,7 +2600,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -2864,22 +2618,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2891,13 +2644,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -2927,7 +2676,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2949,16 +2697,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2969,14 +2716,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -2986,11 +2733,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3002,8 +2749,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -3032,7 +2777,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -3055,16 +2799,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3075,14 +2818,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -3092,11 +2835,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3108,15 +2851,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -3142,7 +2881,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3150,16 +2889,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2911,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -3203,14 +2937,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3223,7 +2949,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3236,18 +2962,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3260,7 +2978,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -3271,19 +2989,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -3316,7 +3024,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3328,20 +3036,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3366,7 +3076,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3374,16 +3084,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3403,10 +3112,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -3428,14 +3133,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3448,7 +3145,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3460,18 +3157,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3481,7 +3170,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -3490,19 +3183,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3535,7 +3217,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3545,20 +3227,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -3583,7 +3267,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -3591,16 +3275,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,10 +3303,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -3645,14 +3324,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3665,7 +3336,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3677,18 +3348,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3698,7 +3361,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -3707,19 +3374,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -3752,7 +3408,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3762,17 +3418,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -3791,21 +3445,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3815,11 +3468,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -3829,14 +3478,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3863,7 +3504,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3872,17 +3513,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -3897,7 +3534,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -3916,22 +3552,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3943,13 +3578,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -3961,7 +3592,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -3980,21 +3610,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4006,13 +3635,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4031,21 +3656,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4055,11 +3679,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -4072,14 +3692,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4106,7 +3718,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4115,17 +3727,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4144,21 +3752,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4168,11 +3775,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4183,14 +3786,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4217,7 +3812,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4226,17 +3821,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -4248,7 +3839,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4267,21 +3857,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4291,11 +3880,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -4304,7 +3889,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -4327,16 +3911,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4347,14 +3930,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4364,11 +3947,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4380,16 +3963,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -4412,15 +3990,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4431,14 +4008,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4448,11 +4025,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4464,8 +4041,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4477,7 +4052,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] @@ -4500,17 +4074,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4521,14 +4094,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4538,11 +4111,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4554,10 +4127,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -4575,21 +4144,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4599,11 +4167,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4614,14 +4178,6 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4648,7 +4204,7 @@ async def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4657,17 +4213,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4686,21 +4238,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4710,11 +4261,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4724,14 +4271,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4758,7 +4297,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4767,17 +4306,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4796,21 +4331,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4820,11 +4354,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4834,14 +4364,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4868,7 +4390,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4877,17 +4399,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -4906,21 +4424,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4930,11 +4447,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4944,14 +4457,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4978,7 +4483,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4987,23 +4492,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -5032,7 +4533,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5040,16 +4541,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5059,11 +4559,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -5087,14 +4583,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5105,7 +4593,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -5118,18 +4606,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5140,7 +4620,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -5150,19 +4630,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -5192,7 +4662,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5201,17 +4671,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5230,21 +4696,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5254,11 +4719,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -5270,14 +4731,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5304,7 +4757,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5313,20 +4766,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -5351,7 +4804,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5359,16 +4812,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5382,13 +4834,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -5411,14 +4859,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5431,7 +4871,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5443,18 +4883,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5464,7 +4896,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -5473,19 +4909,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -5518,7 +4943,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5530,17 +4955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -5570,7 +4993,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] @@ -5593,16 +5015,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5613,14 +5034,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5630,11 +5051,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5646,8 +5067,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -5670,7 +5089,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5695,7 +5114,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5703,16 +5122,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5732,10 +5150,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -5757,14 +5171,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5775,7 +5181,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5787,18 +5193,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5806,7 +5204,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -5815,19 +5213,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5858,7 +5245,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5868,20 +5255,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -5906,7 +5295,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -5914,16 +5303,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5943,10 +5331,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -5968,14 +5352,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -5986,7 +5362,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5998,18 +5374,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6017,7 +5385,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -6026,19 +5398,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6069,7 +5430,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6079,17 +5440,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -6108,21 +5467,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6132,11 +5490,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6146,14 +5500,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6180,7 +5526,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6189,17 +5535,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6213,7 +5555,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -6232,22 +5573,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6259,13 +5599,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -6273,7 +5609,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6295,16 +5630,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6315,14 +5649,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6332,11 +5666,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6348,16 +5682,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -6379,15 +5708,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6398,14 +5726,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -6415,11 +5743,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6431,8 +5759,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6458,7 +5784,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6484,7 +5810,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6492,16 +5818,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6521,10 +5846,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -6546,14 +5867,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6566,7 +5879,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6578,18 +5891,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6602,7 +5907,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6611,20 +5916,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6657,7 +5951,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6667,23 +5961,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6709,7 +6001,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6717,16 +6009,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6738,13 +6029,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6767,14 +6054,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6787,7 +6066,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6799,18 +6078,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6823,7 +6094,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6832,20 +6103,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6878,7 +6139,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6888,17 +6149,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -6917,21 +6176,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6941,11 +6199,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6957,14 +6211,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6991,7 +6237,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7000,17 +6246,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7022,7 +6264,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7041,21 +6282,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7067,19 +6307,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7108,7 +6344,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7116,16 +6352,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7135,11 +6370,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -7166,14 +6397,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7184,7 +6407,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7199,18 +6422,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7221,7 +6436,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7233,20 +6448,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7276,7 +6480,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7285,23 +6489,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -7327,7 +6527,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7335,16 +6535,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7354,11 +6553,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -7383,14 +6578,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7401,7 +6588,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7414,18 +6601,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7436,7 +6615,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -7446,20 +6625,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7489,7 +6657,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7498,17 +6666,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -7520,7 +6684,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7539,21 +6702,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7565,13 +6727,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -7579,7 +6737,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7603,16 +6760,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7623,14 +6779,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7640,11 +6796,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7656,17 +6812,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7690,15 +6841,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7709,14 +6859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7726,11 +6876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7742,8 +6892,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7755,7 +6903,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7779,17 +6926,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7800,14 +6946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7817,11 +6963,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7833,10 +6979,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7847,7 +6989,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7871,17 +7012,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7892,14 +7032,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7909,11 +7049,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7925,15 +7065,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7962,7 +7098,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7970,16 +7106,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7989,11 +7124,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -8020,14 +7151,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8038,7 +7161,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8053,18 +7176,10 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8075,7 +7190,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -8087,20 +7202,9 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8130,7 +7234,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8139,23 +7243,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8184,7 +7284,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8192,16 +7292,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8211,11 +7310,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -8240,14 +7335,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8258,7 +7345,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8271,18 +7358,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8293,7 +7372,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8303,20 +7382,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8346,7 +7414,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8355,23 +7423,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8400,7 +7464,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8408,16 +7472,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8427,11 +7490,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -8456,14 +7515,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8474,7 +7525,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8487,18 +7538,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8509,7 +7552,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8519,20 +7562,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8562,7 +7594,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8571,23 +7603,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8616,7 +7644,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8624,16 +7652,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8643,11 +7670,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -8673,14 +7696,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8691,7 +7706,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8705,18 +7720,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8727,7 +7734,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8738,20 +7745,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8781,7 +7777,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8790,23 +7786,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8835,7 +7827,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8843,16 +7835,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8862,11 +7853,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -8894,14 +7881,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8912,7 +7891,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8928,18 +7907,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8950,7 +7921,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8963,20 +7934,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9006,7 +7966,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9015,23 +7975,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9057,7 +8013,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9065,16 +8021,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9084,11 +8039,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -9113,14 +8064,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9131,7 +8074,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9144,18 +8087,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9166,7 +8101,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9176,20 +8111,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9219,7 +8143,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9228,23 +8152,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9273,7 +8193,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9281,16 +8201,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9300,11 +8219,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -9330,14 +8245,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9348,7 +8255,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9362,18 +8269,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9384,7 +8283,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9396,20 +8295,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9439,7 +8327,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9448,23 +8336,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9493,7 +8377,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9501,16 +8385,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9520,11 +8403,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -9550,14 +8429,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9568,7 +8439,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9582,18 +8453,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9604,7 +8467,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9615,20 +8478,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9658,7 +8510,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9667,20 +8519,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -9693,7 +8541,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9712,22 +8559,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9739,16 +8585,12 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -9773,7 +8615,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -9799,7 +8641,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -9808,16 +8650,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9837,10 +8678,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9868,14 +8705,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9889,7 +8718,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9905,18 +8734,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9930,7 +8751,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -9943,20 +8764,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -9990,7 +8800,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10000,17 +8810,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10029,22 +8837,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10054,11 +8861,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10073,14 +8876,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10108,7 +8903,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10117,17 +8912,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10149,7 +8940,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10168,23 +8958,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10196,13 +8985,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10214,7 +8999,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10238,17 +9022,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10259,14 +9042,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10276,11 +9059,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10292,12 +9075,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10334,21 +9113,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10358,11 +9136,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -10374,14 +9148,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10408,7 +9174,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10417,17 +9183,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10446,21 +9208,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10470,11 +9231,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -10488,14 +9245,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10522,7 +9271,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10531,17 +9280,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10560,21 +9305,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10584,11 +9328,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -10602,14 +9342,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10636,7 +9368,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10645,17 +9377,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -10667,7 +9395,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -10686,21 +9413,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10712,13 +9438,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -10746,7 +9468,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -10775,7 +9499,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -10784,16 +9508,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10803,11 +9526,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -10835,14 +9554,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10854,7 +9565,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10869,18 +9580,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10892,7 +9595,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -10904,20 +9609,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10948,7 +9643,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10957,17 +9652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -10986,22 +9677,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11011,11 +9701,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -11030,14 +9716,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11065,7 +9743,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11074,17 +9752,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11103,22 +9777,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11128,11 +9801,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -11148,14 +9817,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11183,7 +9844,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11192,24 +9853,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -11235,7 +9892,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11244,16 +9901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11273,10 +9929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -11303,14 +9955,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11324,7 +9968,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11340,18 +9984,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11365,7 +10001,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -11378,19 +10014,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11424,7 +10049,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11434,17 +10059,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11463,22 +10086,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11488,11 +10110,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -11506,14 +10124,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11541,7 +10151,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11550,17 +10160,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11574,7 +10180,6 @@ async def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -11593,22 +10198,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11620,13 +10224,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -11640,7 +10240,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11659,22 +10258,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11686,13 +10284,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11721,7 +10315,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -11745,7 +10338,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -11753,12 +10346,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11769,14 +10361,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -11786,11 +10378,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11802,10 +10394,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -11823,22 +10411,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11848,11 +10435,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11868,14 +10451,6 @@ async def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11903,7 +10478,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11912,17 +10487,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -11941,22 +10512,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11966,11 +10536,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -11984,14 +10550,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12019,7 +10577,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12028,17 +10586,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12057,22 +10611,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12082,11 +10635,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -12100,14 +10649,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12135,7 +10676,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12144,17 +10685,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12173,22 +10710,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12198,11 +10734,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -12217,14 +10749,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12252,7 +10776,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12261,17 +10785,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12290,22 +10810,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12315,11 +10834,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -12333,14 +10848,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12368,7 +10875,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12377,24 +10884,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -12420,7 +10923,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12429,16 +10932,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12452,13 +10954,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12484,14 +10982,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12505,7 +10995,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12519,18 +11009,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12544,7 +11026,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -12556,19 +11038,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12602,7 +11073,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12614,17 +11085,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -12648,7 +11117,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12673,23 +11142,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12703,13 +11171,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -12731,14 +11195,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12748,7 +11204,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12757,18 +11213,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12778,7 +11226,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -12786,19 +11234,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12830,7 +11268,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12843,20 +11281,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -12881,23 +11317,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12911,13 +11346,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -12938,14 +11369,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12955,7 +11378,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12963,18 +11386,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -12984,7 +11399,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -12992,19 +11407,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -13036,7 +11440,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13049,17 +11453,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -13088,7 +11490,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -13112,16 +11513,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13132,14 +11532,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -13149,11 +11549,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13165,10 +11565,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -13177,7 +11573,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -13196,21 +11591,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13222,10 +11616,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/models/_models_py3.py index 61437a591dc0..8ce08c7020b0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -230,7 +226,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, recommendation is to set `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long to false. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -255,7 +251,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, recommendation is to set `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long to false. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -270,7 +266,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -336,7 +332,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -394,7 +390,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet] @@ -760,7 +756,7 @@ def __init__(self, **kwargs: Any) -> None: class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1041,7 +1037,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1098,7 +1094,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -1155,7 +1151,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1222,7 +1218,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1315,7 +1311,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.Image] @@ -1349,7 +1345,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -1541,12 +1537,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_10_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -1571,12 +1567,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_10_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -1720,7 +1716,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -1753,7 +1749,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -1839,7 +1835,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.Usage] @@ -1873,7 +1869,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -2161,7 +2157,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -2302,7 +2298,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -2335,7 +2331,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -2346,10 +2342,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2362,16 +2358,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2379,10 +2375,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_10_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2018_10_01.models.VaultSecretGroup] @@ -2422,7 +2418,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -2434,10 +2430,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -2450,16 +2446,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -2467,10 +2463,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_10_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -2548,7 +2544,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2629,7 +2625,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup] @@ -2667,22 +2663,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -2753,7 +2738,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3026,7 +3011,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3089,7 +3074,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3152,7 +3137,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -3226,7 +3211,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -3273,7 +3258,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3306,7 +3291,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.RunCommandDocumentBase] @@ -3342,7 +3327,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -3477,7 +3462,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -3495,7 +3480,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -3514,12 +3499,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_10_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.DataDisk] """ @@ -3546,12 +3531,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_10_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.DataDisk] """ super().__init__(**kwargs) @@ -3586,7 +3571,7 @@ def __init__(self, **kwargs: Any) -> None: class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3603,57 +3588,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -3689,7 +3623,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -3832,7 +3766,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -4042,7 +3976,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4084,10 +4018,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -4105,7 +4039,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -4201,10 +4135,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -4218,7 +4152,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -4285,7 +4219,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -4375,7 +4309,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4493,7 +4427,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -4537,7 +4471,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4818,7 +4752,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_10_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_10_01.models.UserAssignedIdentitiesValue] """ @@ -4850,7 +4784,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2018_10_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_10_01.models.UserAssignedIdentitiesValue] """ @@ -4864,7 +4798,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4920,7 +4854,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5121,7 +5055,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -5182,7 +5116,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5364,7 +5298,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -5560,10 +5494,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension] @@ -5639,7 +5573,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2018_10_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -5678,7 +5612,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -5689,7 +5623,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -5754,7 +5688,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5781,7 +5715,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5935,10 +5869,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -5980,7 +5914,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet] @@ -6016,7 +5950,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetSku] @@ -6052,7 +5986,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet] @@ -6085,7 +6019,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -6115,10 +6049,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6205,7 +6139,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -6230,7 +6164,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2018_10_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -6255,7 +6189,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2018_10_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -6269,7 +6203,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -6409,10 +6343,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6425,16 +6359,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6442,10 +6376,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2018_10_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6488,10 +6422,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -6504,16 +6438,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -6521,10 +6455,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2018_10_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -6540,10 +6474,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -6604,10 +6538,12 @@ def __init__( self.public_ip_prefix = public_ip_prefix -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -6634,7 +6570,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -6642,18 +6578,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -6772,12 +6696,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetDataDisk] """ @@ -6805,12 +6729,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetDataDisk] """ @@ -6939,7 +6863,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -7055,7 +6979,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -7140,7 +7064,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar network_interface_configurations: The list of network configurations. @@ -7291,7 +7215,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -7335,7 +7259,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -7449,7 +7373,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7502,10 +7426,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -7517,7 +7441,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str """ @@ -7609,10 +7533,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -7622,7 +7546,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7646,7 +7570,7 @@ def __init__( self.license_type = license_type -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7699,10 +7623,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -7838,7 +7762,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM] @@ -7901,7 +7825,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -7970,7 +7894,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -8142,10 +8066,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -8163,7 +8087,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -8246,10 +8170,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2018_10_01.models.SubResource @@ -8263,7 +8187,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/operations/_operations.py index b3bfffad96ce..64c135eae2a2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2018_10_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +401,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -432,7 +429,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -462,7 +459,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -495,7 +492,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -526,7 +523,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -572,7 +569,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -607,7 +604,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -642,7 +639,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -668,7 +665,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -708,7 +705,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -775,7 +772,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -823,7 +820,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -854,7 +851,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -883,7 +880,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +939,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1004,7 +1003,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1134,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1165,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1191,7 +1190,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1216,7 +1215,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1293,7 +1292,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1455,7 +1454,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1480,7 +1479,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1667,7 +1666,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1721,7 +1720,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1755,7 +1754,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1789,7 +1788,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1814,7 +1813,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1845,7 +1844,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1876,7 +1875,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1907,7 +1906,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1938,7 +1937,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1968,7 +1967,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1994,7 +1995,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2025,7 +2026,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2056,7 +2057,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2087,7 +2088,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2118,7 +2119,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2149,7 +2150,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2180,7 +2181,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2211,7 +2212,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2242,7 +2243,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2273,7 +2274,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2304,7 +2305,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2341,7 +2342,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2376,7 +2377,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2402,7 +2403,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2442,7 +2443,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2473,7 +2474,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2498,7 +2499,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2523,7 +2524,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2548,7 +2549,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2579,7 +2580,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2611,7 +2612,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2637,7 +2638,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2663,7 +2664,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2698,7 +2699,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2724,7 +2725,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2756,7 +2757,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2788,7 +2789,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -2834,7 +2835,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2860,7 +2861,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2886,7 +2887,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2912,7 +2913,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2938,7 +2939,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2964,7 +2965,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2999,7 +3000,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3032,7 +3033,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3065,7 +3066,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3092,7 +3095,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3147,7 +3150,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -3171,14 +3173,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3189,14 +3190,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -3206,11 +3207,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,8 +3223,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -3266,7 +3265,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3277,7 +3275,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3289,11 +3287,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3304,7 +3301,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -3314,12 +3311,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3347,7 +3340,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3355,16 +3348,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3376,13 +3368,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -3405,7 +3393,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3416,7 +3403,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3428,11 +3415,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3443,7 +3429,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -3453,12 +3439,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3486,7 +3468,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -3494,16 +3476,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3515,13 +3496,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -3533,7 +3510,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3552,21 +3528,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3576,11 +3551,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -3590,7 +3561,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -3609,21 +3579,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3635,13 +3604,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -3652,7 +3617,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3674,16 +3638,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3694,14 +3657,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3711,11 +3674,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3727,17 +3690,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3759,16 +3717,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3779,14 +3736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -3796,11 +3753,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3812,10 +3769,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -3827,7 +3780,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -3849,17 +3801,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3870,14 +3821,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -3887,11 +3838,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3903,10 +3854,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -3950,7 +3897,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -3961,7 +3907,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3974,11 +3920,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -3989,7 +3934,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -3999,12 +3944,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4032,7 +3973,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4040,16 +3981,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4069,10 +4009,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -4095,7 +4031,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4106,7 +4041,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4119,11 +4054,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4134,7 +4068,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -4144,12 +4078,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4177,7 +4108,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4185,16 +4116,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4206,13 +4136,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -4224,7 +4150,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4243,21 +4168,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4267,11 +4191,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -4283,7 +4203,6 @@ def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4302,21 +4221,20 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4328,19 +4246,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -4364,15 +4277,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4383,14 +4295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -4400,11 +4312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4416,10 +4328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -4428,7 +4336,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -4452,16 +4359,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4472,14 +4378,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -4489,11 +4395,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4505,10 +4411,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class VirtualMachineExtensionImagesOperations: """ @@ -4544,7 +4446,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4563,23 +4464,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4591,13 +4491,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -4609,7 +4505,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4628,21 +4523,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4654,13 +4548,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -4688,7 +4578,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4707,7 +4596,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -4716,16 +4605,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4737,13 +4625,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -4771,7 +4655,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4797,7 +4681,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4806,16 +4690,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4835,10 +4718,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -4865,14 +4744,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4886,7 +4757,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4902,18 +4773,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4927,7 +4790,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4940,20 +4803,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4987,7 +4839,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4997,24 +4849,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -5040,7 +4890,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -5049,16 +4899,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5070,13 +4919,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5105,14 +4950,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5126,7 +4963,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5142,18 +4979,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5167,7 +4996,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -5180,20 +5009,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -5227,7 +5045,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5237,17 +5055,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -5266,22 +5082,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5291,11 +5106,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5310,14 +5121,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5345,7 +5148,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5354,17 +5157,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -5386,7 +5185,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5405,23 +5203,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5433,13 +5230,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5453,7 +5246,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -5472,22 +5264,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5499,13 +5290,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -5544,7 +5331,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5563,7 +5349,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -5571,16 +5357,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5592,13 +5377,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5630,7 +5411,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5649,7 +5429,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -5659,16 +5439,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5680,13 +5459,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -5698,7 +5473,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5717,21 +5491,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5743,13 +5516,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -5757,7 +5526,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5776,20 +5544,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5801,13 +5568,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -5821,7 +5584,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5840,22 +5602,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5867,13 +5628,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5903,7 +5660,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5925,16 +5681,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5945,14 +5700,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5962,11 +5717,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5978,8 +5733,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -6008,7 +5761,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -6030,16 +5782,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6050,14 +5801,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6067,11 +5818,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6083,15 +5834,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -6117,7 +5864,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6125,16 +5872,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6148,13 +5894,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -6178,14 +5920,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6198,7 +5932,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6211,18 +5945,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6235,7 +5961,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -6246,19 +5972,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -6291,7 +6007,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6303,20 +6019,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6341,7 +6059,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6349,16 +6067,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6378,10 +6095,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -6403,14 +6116,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6422,7 +6127,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6434,18 +6139,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6454,7 +6151,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -6463,19 +6164,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6507,7 +6197,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6517,20 +6207,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6555,7 +6247,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6563,16 +6255,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6592,10 +6283,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -6617,14 +6304,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6636,7 +6315,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6648,18 +6327,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6668,7 +6339,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -6677,19 +6352,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] @@ -6721,7 +6385,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6731,17 +6395,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6760,21 +6422,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6784,11 +6445,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6798,14 +6455,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6832,7 +6481,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6841,17 +6490,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6866,7 +6511,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6885,22 +6529,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6912,13 +6555,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -6930,7 +6569,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6949,21 +6587,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6975,13 +6612,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7000,21 +6633,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7024,11 +6656,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7039,14 +6667,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7073,7 +6693,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7082,17 +6702,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7111,21 +6727,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7135,11 +6750,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7150,14 +6761,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7184,7 +6787,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7193,17 +6796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -7215,7 +6814,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7234,21 +6832,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7258,11 +6855,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -7271,7 +6864,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -7293,16 +6885,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7313,14 +6904,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7330,11 +6921,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7346,16 +6937,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -7377,15 +6963,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7396,14 +6981,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7413,11 +6998,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7429,8 +7014,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7442,7 +7025,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7464,17 +7046,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7485,14 +7066,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7502,11 +7083,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7518,10 +7099,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any ) -> None: @@ -7539,21 +7116,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7563,11 +7139,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7578,14 +7150,6 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7612,7 +7176,7 @@ def begin_power_off(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7621,17 +7185,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7650,21 +7210,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7674,11 +7233,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7688,14 +7243,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7722,7 +7269,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7731,17 +7278,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7760,21 +7303,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7784,11 +7326,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7798,14 +7336,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7832,7 +7362,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7841,17 +7371,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7870,21 +7396,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7894,11 +7419,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7908,14 +7429,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7942,7 +7455,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7951,23 +7464,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7996,7 +7505,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8004,16 +7513,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8023,11 +7531,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -8051,14 +7555,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8069,7 +7565,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8082,18 +7578,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8104,7 +7592,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -8114,19 +7602,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8156,7 +7634,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8165,17 +7643,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8194,21 +7668,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8218,11 +7691,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8232,14 +7701,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8266,7 +7727,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8275,20 +7736,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -8313,7 +7774,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8321,16 +7782,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8344,13 +7804,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -8373,14 +7829,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -8392,7 +7840,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8404,18 +7852,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -8424,7 +7864,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -8433,19 +7877,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -8477,7 +7910,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8489,17 +7922,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -8529,7 +7960,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -8551,16 +7981,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8571,14 +8000,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -8588,11 +8017,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8604,8 +8033,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -8628,7 +8055,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8653,7 +8080,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8661,16 +8088,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8690,10 +8116,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -8715,14 +8137,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8733,7 +8147,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8745,18 +8159,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8764,7 +8170,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -8773,19 +8179,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8816,7 +8211,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8826,20 +8221,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8864,7 +8261,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8872,16 +8269,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8901,10 +8297,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -8926,14 +8318,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8944,7 +8328,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8956,18 +8340,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8975,7 +8351,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -8984,19 +8364,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9027,7 +8396,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9037,17 +8406,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -9066,21 +8433,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9090,11 +8456,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9104,14 +8466,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9138,7 +8492,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9147,17 +8501,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9171,7 +8521,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -9190,22 +8539,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9217,13 +8565,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -9231,7 +8575,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9253,16 +8596,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9273,14 +8615,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -9290,11 +8632,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9306,16 +8648,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2018_10_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -9337,15 +8674,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9356,14 +8692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -9373,11 +8709,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9389,8 +8725,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -9416,7 +8750,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9442,7 +8776,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9450,16 +8784,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9479,10 +8812,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -9504,14 +8833,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9524,7 +8845,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9536,18 +8857,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9560,7 +8873,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -9569,20 +8882,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9615,7 +8917,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9625,23 +8927,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9667,7 +8967,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9675,16 +8975,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9696,13 +8995,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9725,14 +9020,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9745,7 +9032,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9757,18 +9044,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9781,7 +9060,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9790,20 +9069,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9836,7 +9105,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9846,17 +9115,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9875,21 +9142,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9899,11 +9165,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9913,14 +9175,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9947,7 +9201,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9956,17 +9210,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -9976,7 +9226,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9995,21 +9244,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10021,19 +9269,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10062,7 +9306,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10070,16 +9314,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10089,11 +9332,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -10120,14 +9359,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10138,7 +9369,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10153,18 +9384,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10175,7 +9398,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -10187,20 +9410,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10230,7 +9442,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10239,23 +9451,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10281,7 +9489,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10289,16 +9497,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10308,11 +9515,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -10337,14 +9540,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10355,7 +9550,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10368,18 +9563,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10390,7 +9577,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -10400,20 +9587,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10443,7 +9619,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10452,17 +9628,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -10474,7 +9646,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10493,21 +9664,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10519,13 +9689,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -10533,7 +9699,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10557,16 +9722,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10577,14 +9741,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10594,11 +9758,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10610,17 +9774,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10644,15 +9803,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10663,14 +9821,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10680,11 +9838,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10696,8 +9854,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10709,7 +9865,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10733,17 +9888,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10754,14 +9908,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10771,11 +9925,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10787,10 +9941,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10801,7 +9951,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10825,17 +9974,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10846,14 +9994,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10863,11 +10011,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10879,15 +10027,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10916,7 +10060,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10924,16 +10068,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10943,11 +10086,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -10974,14 +10113,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10992,7 +10123,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11007,18 +10138,10 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11029,7 +10152,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still @@ -11041,20 +10164,9 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11084,7 +10196,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11093,23 +10205,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11138,7 +10246,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11146,16 +10254,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11165,11 +10272,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -11194,14 +10297,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11212,7 +10307,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11225,18 +10320,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11247,7 +10334,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -11257,20 +10344,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11300,7 +10376,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11309,23 +10385,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11354,7 +10426,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11362,16 +10434,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11381,11 +10452,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -11410,14 +10477,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11428,7 +10487,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11441,18 +10500,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11463,7 +10514,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -11473,20 +10524,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11516,7 +10556,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11525,23 +10565,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11570,7 +10606,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11578,16 +10614,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11597,11 +10632,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -11627,14 +10658,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11645,7 +10668,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11659,18 +10682,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11681,7 +10696,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11692,20 +10707,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11735,7 +10739,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11744,23 +10748,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11789,7 +10789,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11797,16 +10797,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11816,11 +10815,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -11848,14 +10843,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11866,7 +10853,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11882,18 +10869,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11904,7 +10883,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11917,20 +10896,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11960,7 +10928,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11969,23 +10937,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12011,7 +10975,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12019,16 +10983,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12038,11 +11001,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -12067,14 +11026,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12085,7 +11036,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12098,18 +11049,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12120,7 +11063,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -12130,20 +11073,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12173,7 +11105,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12182,23 +11114,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12227,7 +11155,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12235,16 +11163,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12254,11 +11181,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12284,14 +11207,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12302,7 +11217,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12316,18 +11231,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12338,7 +11245,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -12350,20 +11257,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12393,7 +11289,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12402,23 +11298,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12447,7 +11339,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12455,16 +11347,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12474,11 +11365,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -12504,14 +11391,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12522,7 +11401,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12536,18 +11415,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12558,7 +11429,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12569,20 +11440,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12612,7 +11472,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12621,20 +11481,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12647,7 +11503,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12666,22 +11521,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12693,16 +11547,12 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12727,7 +11577,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12753,7 +11603,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12762,16 +11612,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12791,10 +11640,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -12822,14 +11667,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12843,7 +11680,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12859,18 +11696,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12884,7 +11713,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -12897,20 +11726,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12944,7 +11762,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12954,17 +11772,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -12983,22 +11799,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13008,11 +11823,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13027,14 +11838,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13062,7 +11865,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13071,17 +11874,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13103,7 +11902,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -13122,23 +11920,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13150,13 +11947,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13168,7 +11961,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13192,17 +11984,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13213,14 +12004,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -13230,11 +12021,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13246,12 +12037,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13288,21 +12075,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13312,11 +12098,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13326,14 +12108,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13360,7 +12134,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13369,17 +12143,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13398,21 +12168,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13422,11 +12191,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -13440,14 +12205,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13474,7 +12231,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13483,17 +12240,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13512,21 +12265,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13536,11 +12288,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -13554,14 +12302,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13588,7 +12328,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13597,17 +12337,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -13619,7 +12355,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -13638,21 +12373,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13664,13 +12398,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -13698,7 +12428,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -13727,7 +12459,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13736,16 +12468,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13755,11 +12486,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -13787,14 +12514,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13806,7 +12525,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13821,18 +12540,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13844,7 +12555,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -13856,20 +12569,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13900,7 +12603,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13909,17 +12612,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13938,22 +12637,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13963,11 +12661,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -13982,14 +12676,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14017,7 +12703,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14026,17 +12712,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14055,22 +12737,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14080,11 +12761,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -14100,14 +12777,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14135,7 +12804,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14144,24 +12813,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -14187,7 +12852,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14196,16 +12861,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14225,10 +12889,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -14255,14 +12915,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14276,7 +12928,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14292,18 +12944,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14317,7 +12961,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -14330,19 +12974,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14376,7 +13009,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14386,17 +13019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14415,22 +13046,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14440,11 +13070,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14458,14 +13084,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14493,7 +13111,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14502,17 +13120,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14526,7 +13140,6 @@ def get( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -14545,22 +13158,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14572,13 +13184,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -14592,7 +13200,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14611,22 +13218,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14638,13 +13244,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14673,7 +13275,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14697,7 +13298,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -14705,12 +13306,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14721,14 +13321,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -14738,11 +13338,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14754,10 +13354,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any ) -> None: @@ -14775,22 +13371,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14800,11 +13395,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -14820,14 +13411,6 @@ def begin_power_off( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14855,7 +13438,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14864,17 +13447,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14893,22 +13472,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14918,11 +13496,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -14936,14 +13510,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14971,7 +13537,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14980,17 +13546,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15009,22 +13571,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15034,11 +13595,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -15052,14 +13609,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15087,7 +13636,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15096,17 +13645,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15125,22 +13670,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15150,11 +13694,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -15169,14 +13709,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15204,7 +13736,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15213,17 +13745,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15242,22 +13770,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15267,11 +13794,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -15285,14 +13808,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15320,7 +13835,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15329,24 +13844,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -15372,7 +13883,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15381,16 +13892,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15404,13 +13914,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -15436,14 +13942,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -15456,7 +13954,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15470,18 +13968,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -15494,7 +13984,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -15506,19 +13996,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2018_10_01.models.RunCommandResult] @@ -15551,7 +14030,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15563,17 +14042,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -15597,7 +14074,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15622,23 +14099,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15652,13 +14128,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -15680,14 +14152,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15697,7 +14161,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15706,18 +14170,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15727,7 +14183,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15735,19 +14191,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15779,7 +14225,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15791,20 +14237,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15829,23 +14273,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15859,13 +14302,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -15886,14 +14325,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15903,7 +14334,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15911,18 +14342,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15932,7 +14355,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15940,19 +14363,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2018_10_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15984,7 +14396,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15996,17 +14408,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -16035,7 +14445,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -16059,16 +14468,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16079,14 +14487,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -16096,11 +14504,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16112,10 +14520,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -16124,7 +14528,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2018_10_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -16143,21 +14546,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2018-10-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16169,10 +14571,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_compute_management_client.py index 7f53b4f0fc38..6a64a4eee290 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -142,7 +144,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -216,7 +236,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -236,7 +256,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_configuration.py index 03880bca46ff..66be55a80f73 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_metadata.json index 48cfca550cfa..23e2a6048305 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_compute_management_client.py index b78a921430e7..1bcb84996ad1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -144,7 +146,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -218,7 +238,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -238,7 +260,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_configuration.py index 6ea6ee60e062..f4c216ecd24e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/operations/_operations.py index cf5359b22bd0..219def88fac5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -185,10 +184,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -217,7 +212,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -241,14 +235,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -259,14 +252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -276,11 +269,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -292,8 +285,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -336,7 +327,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -347,7 +337,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -359,11 +349,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -374,7 +363,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -384,12 +373,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -417,7 +402,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -425,16 +410,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -446,13 +430,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -475,7 +455,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -486,7 +465,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -498,11 +477,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -513,7 +491,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -523,12 +501,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -556,7 +530,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -564,16 +538,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +558,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -603,7 +572,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -622,21 +590,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -646,11 +613,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -660,7 +623,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -679,21 +641,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -705,13 +666,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -722,7 +679,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet] @@ -745,16 +701,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -765,14 +720,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -782,11 +737,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -798,17 +753,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet] @@ -831,16 +781,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -851,14 +800,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -868,11 +817,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -884,10 +833,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -899,7 +844,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] @@ -922,17 +866,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -943,14 +886,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -960,11 +903,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -976,10 +919,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1023,7 +962,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1034,7 +972,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1047,11 +985,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1062,7 +999,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1072,12 +1009,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1105,7 +1038,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1113,16 +1046,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1142,10 +1074,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1168,7 +1096,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1179,7 +1106,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1192,11 +1119,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1207,7 +1133,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1217,12 +1143,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1250,7 +1173,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1258,16 +1181,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1279,13 +1201,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1297,7 +1215,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1316,21 +1233,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1340,11 +1256,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1356,7 +1268,6 @@ async def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1375,21 +1286,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1401,19 +1311,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1437,15 +1342,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1456,14 +1360,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1473,11 +1377,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1489,10 +1393,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1501,7 +1401,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1525,16 +1424,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1545,14 +1443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1562,11 +1460,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1578,10 +1476,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1626,7 +1520,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1637,7 +1530,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1651,11 +1544,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1666,7 +1558,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1678,12 +1570,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1711,7 +1599,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1719,16 +1607,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1748,10 +1635,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1773,7 +1656,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1784,7 +1666,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1796,11 +1678,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1811,7 +1692,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1821,12 +1702,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1854,7 +1731,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1862,16 +1739,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1883,13 +1759,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1901,7 +1773,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1920,21 +1791,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1944,11 +1814,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -1958,7 +1824,6 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1977,21 +1842,20 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2003,13 +1867,9 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2020,7 +1880,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup] @@ -2043,16 +1902,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2063,14 +1921,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2080,11 +1938,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2096,16 +1954,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup] @@ -2128,15 +1981,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2147,14 +1999,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2164,11 +2016,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2180,8 +2032,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2208,7 +2058,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2234,7 +2084,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2243,16 +2093,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,10 +2121,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2300,14 +2145,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2321,7 +2158,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2335,18 +2172,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2360,7 +2189,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2372,19 +2201,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2418,7 +2236,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2428,24 +2246,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2471,7 +2287,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2480,16 +2296,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2501,13 +2316,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2533,14 +2344,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2554,7 +2357,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2568,18 +2371,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2593,7 +2388,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2605,19 +2400,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2651,7 +2435,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2661,17 +2445,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2690,22 +2472,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2715,11 +2496,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2733,14 +2510,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2768,7 +2537,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2777,17 +2546,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2810,7 +2575,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2829,23 +2593,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2857,13 +2620,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2876,7 +2635,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -2899,17 +2657,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2920,14 +2677,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -2937,11 +2694,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2953,10 +2710,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class VirtualMachineExtensionImagesOperations: """ @@ -2992,7 +2745,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3011,23 +2763,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3039,13 +2790,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3057,7 +2804,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3076,21 +2822,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3102,13 +2847,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3136,7 +2877,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3155,7 +2895,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3164,16 +2904,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3185,13 +2924,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3219,7 +2954,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3245,7 +2980,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3254,16 +2989,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3283,10 +3017,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -3313,14 +3043,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3334,7 +3056,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3350,18 +3072,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3375,7 +3089,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -3388,20 +3102,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3435,7 +3138,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3445,24 +3148,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3488,7 +3189,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3497,16 +3198,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3518,13 +3218,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3553,14 +3249,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3574,7 +3262,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3590,18 +3278,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3615,7 +3295,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -3628,20 +3308,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3675,7 +3344,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3685,17 +3354,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -3714,22 +3381,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3739,11 +3405,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3758,14 +3420,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3793,7 +3447,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3802,17 +3456,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -3834,7 +3484,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -3853,23 +3502,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3881,13 +3529,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -3901,7 +3545,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3920,22 +3563,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3947,13 +3589,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -3992,7 +3630,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4011,7 +3648,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4019,16 +3656,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4040,13 +3676,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4078,7 +3710,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4097,7 +3728,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4107,16 +3738,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4128,13 +3758,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4146,7 +3772,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4165,21 +3790,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4191,13 +3815,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4205,7 +3825,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4224,20 +3843,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4249,13 +3867,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4269,7 +3883,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4288,22 +3901,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4315,13 +3927,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -4351,7 +3959,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4373,16 +3980,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4393,14 +3999,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -4410,11 +4016,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4426,8 +4032,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -4456,7 +4060,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -4479,16 +4082,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4499,14 +4101,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4516,11 +4118,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4532,15 +4134,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -4566,7 +4164,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4574,16 +4172,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4597,13 +4194,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -4627,14 +4220,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4647,7 +4232,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4660,18 +4245,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4684,7 +4261,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -4695,19 +4272,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4740,7 +4307,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4752,20 +4319,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -4790,7 +4359,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4798,16 +4367,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4827,10 +4395,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -4852,14 +4416,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4872,7 +4428,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4884,18 +4440,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4905,7 +4453,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -4914,19 +4466,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4959,7 +4500,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4969,20 +4510,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5007,7 +4550,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5015,16 +4558,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5044,10 +4586,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -5069,14 +4607,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5089,7 +4619,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5101,18 +4631,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5122,7 +4644,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -5131,19 +4657,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5176,7 +4691,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5186,17 +4701,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5215,21 +4728,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5239,11 +4751,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5253,14 +4761,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5287,7 +4787,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5296,17 +4796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -5321,7 +4817,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -5340,22 +4835,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5367,13 +4861,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -5385,7 +4875,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -5404,21 +4893,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5430,13 +4918,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5455,21 +4939,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5479,11 +4962,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -5496,14 +4975,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5530,7 +5001,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5539,17 +5010,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5568,21 +5035,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5592,11 +5058,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5607,14 +5069,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5641,7 +5095,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5650,17 +5104,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -5672,7 +5122,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5691,21 +5140,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5715,11 +5163,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -5728,7 +5172,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -5751,16 +5194,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5771,14 +5213,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5788,11 +5230,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5804,16 +5246,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -5836,15 +5273,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5855,14 +5291,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5872,11 +5308,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5888,8 +5324,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5901,7 +5335,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] @@ -5924,17 +5357,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5945,14 +5377,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5962,11 +5394,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5978,10 +5410,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -5999,22 +5427,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6024,11 +5451,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -6045,14 +5468,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6080,7 +5495,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6089,17 +5504,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6118,21 +5529,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6142,11 +5552,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6156,14 +5562,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6190,7 +5588,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6199,17 +5597,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6228,21 +5622,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6252,11 +5645,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6266,14 +5655,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6300,7 +5681,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6309,17 +5690,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6338,21 +5715,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6362,11 +5738,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6376,14 +5748,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6410,7 +5774,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6419,23 +5783,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -6464,7 +5824,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6472,16 +5832,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6491,11 +5850,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6519,14 +5874,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6537,7 +5884,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6550,18 +5897,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6572,7 +5911,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -6582,19 +5921,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6624,7 +5953,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6633,17 +5962,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6662,21 +5987,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6686,11 +6010,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -6702,14 +6022,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6736,7 +6048,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6745,20 +6057,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -6783,7 +6095,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6791,16 +6103,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6814,13 +6125,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -6843,14 +6150,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -6863,7 +6162,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6875,18 +6174,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -6896,7 +6187,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -6905,19 +6200,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -6950,7 +6234,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6962,17 +6246,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -7002,7 +6284,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] @@ -7025,16 +6306,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7045,14 +6325,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7062,11 +6342,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7078,8 +6358,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -7102,7 +6380,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7127,7 +6405,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7135,16 +6413,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7164,10 +6441,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -7189,14 +6462,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7207,7 +6472,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7219,18 +6484,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7238,7 +6495,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -7247,19 +6504,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7290,7 +6536,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7300,20 +6546,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7338,7 +6586,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7346,16 +6594,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7375,10 +6622,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -7400,14 +6643,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7418,7 +6653,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7430,18 +6665,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7449,7 +6676,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -7458,19 +6689,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7501,7 +6721,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7511,17 +6731,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -7540,21 +6758,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7564,11 +6781,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7578,14 +6791,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7612,7 +6817,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7621,17 +6826,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7645,7 +6846,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -7664,22 +6864,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7691,13 +6890,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -7705,7 +6900,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7727,16 +6921,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7747,14 +6940,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7764,11 +6957,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7780,16 +6973,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7811,15 +6999,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7830,14 +7017,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7847,11 +7034,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7863,8 +7050,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -7890,7 +7075,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -7916,7 +7101,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7924,16 +7109,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7953,10 +7137,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -7978,14 +7158,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7998,7 +7170,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8010,18 +7182,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8034,7 +7198,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8043,20 +7207,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8089,7 +7242,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8099,23 +7252,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8141,7 +7292,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8149,16 +7300,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8170,13 +7320,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8199,14 +7345,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8219,7 +7357,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8231,18 +7369,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8255,7 +7385,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -8264,20 +7394,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8310,7 +7430,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8320,17 +7440,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8349,21 +7467,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8373,11 +7490,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8389,14 +7502,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8423,7 +7528,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8432,17 +7537,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8454,7 +7555,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -8473,21 +7573,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8499,19 +7598,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8540,7 +7635,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8548,16 +7643,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8567,11 +7661,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -8598,14 +7688,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8616,7 +7698,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8631,18 +7713,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8653,7 +7727,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -8665,20 +7739,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8708,7 +7771,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8717,23 +7780,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8759,7 +7818,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8767,16 +7826,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8786,11 +7844,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -8815,14 +7869,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8833,7 +7879,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8846,18 +7892,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8868,7 +7906,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -8878,20 +7916,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8921,7 +7948,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8930,17 +7957,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -8952,7 +7975,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -8971,21 +7993,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8997,13 +8018,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -9011,7 +8028,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9035,16 +8051,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9055,14 +8070,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -9072,11 +8087,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9088,17 +8103,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9122,15 +8132,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9141,14 +8150,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -9158,11 +8167,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9174,8 +8183,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9187,7 +8194,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -9211,17 +8217,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9232,14 +8237,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -9249,11 +8254,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9265,10 +8270,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9279,7 +8280,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -9303,17 +8303,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9324,14 +8323,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -9341,11 +8340,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9357,15 +8356,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -9396,7 +8391,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9405,16 +8400,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9424,11 +8418,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -9460,14 +8450,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9478,7 +8460,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -9494,7 +8476,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -9502,14 +8484,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9520,7 +8494,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -9534,24 +8508,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9582,7 +8545,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9591,23 +8554,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9636,7 +8595,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9644,16 +8603,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9663,11 +8621,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -9692,14 +8646,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9710,7 +8656,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9723,18 +8669,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9745,7 +8683,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -9755,20 +8693,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9798,7 +8725,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9807,23 +8734,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9852,7 +8775,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9860,16 +8783,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9879,11 +8801,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -9908,14 +8826,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9926,7 +8836,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9939,18 +8849,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9961,7 +8863,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -9971,20 +8873,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10014,7 +8905,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10023,23 +8914,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10068,7 +8955,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10076,16 +8963,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10095,11 +8981,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -10125,14 +9007,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10143,7 +9017,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10157,18 +9031,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10179,7 +9045,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -10190,20 +9056,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10233,7 +9088,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10242,23 +9097,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10287,7 +9138,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10295,16 +9146,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10314,11 +9164,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -10346,14 +9192,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10364,7 +9202,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10380,18 +9218,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10402,7 +9232,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -10415,20 +9245,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10458,7 +9277,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10467,23 +9286,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10509,7 +9324,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10517,16 +9332,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10536,11 +9350,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -10565,14 +9375,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10583,7 +9385,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10596,18 +9398,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10618,7 +9412,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -10628,20 +9422,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10671,7 +9454,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10680,23 +9463,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10725,7 +9504,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10733,16 +9512,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10752,11 +9530,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -10782,14 +9556,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10800,7 +9566,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10814,18 +9580,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10836,7 +9594,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -10848,20 +9606,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10891,7 +9638,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10900,23 +9647,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10945,7 +9688,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10953,16 +9696,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10972,11 +9714,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -11002,14 +9740,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11020,7 +9750,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11034,18 +9764,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11056,7 +9778,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -11067,20 +9789,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11110,7 +9821,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11119,20 +9830,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -11145,7 +9852,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -11164,22 +9870,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11191,13 +9896,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -11222,7 +9923,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11233,7 +9933,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11246,11 +9946,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11261,7 +9960,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -11272,13 +9971,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11306,7 +10002,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11314,16 +10010,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11333,14 +10028,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11365,7 +10056,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -11391,7 +10082,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -11400,16 +10091,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11429,10 +10119,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -11460,14 +10146,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11481,7 +10159,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11497,18 +10175,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11522,7 +10192,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -11535,20 +10205,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11582,7 +10241,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11592,17 +10251,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -11621,22 +10278,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11646,11 +10302,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -11665,14 +10317,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11700,7 +10344,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11709,17 +10353,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11741,7 +10381,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11760,23 +10399,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11788,13 +10426,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11806,7 +10440,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11830,17 +10463,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11851,14 +10483,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -11868,11 +10500,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11884,12 +10516,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11926,21 +10554,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11950,11 +10577,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -11966,14 +10589,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12000,7 +10615,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12009,17 +10624,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12038,21 +10649,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12062,11 +10672,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -12080,14 +10686,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12114,7 +10712,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12123,17 +10721,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12152,21 +10746,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12176,11 +10769,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -12194,14 +10783,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12228,7 +10809,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12237,17 +10818,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -12259,7 +10836,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -12278,21 +10854,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12304,13 +10879,9 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12338,7 +10909,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12367,7 +10940,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12376,16 +10949,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12395,11 +10967,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -12427,14 +10995,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12446,7 +11006,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12461,18 +11021,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12484,7 +11036,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12496,20 +11050,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12540,7 +11084,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12549,17 +11093,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12578,22 +11118,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12603,11 +11142,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -12622,14 +11157,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12657,7 +11184,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12666,17 +11193,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12695,22 +11218,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12720,11 +11242,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -12740,14 +11258,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12775,7 +11285,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12784,24 +11294,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12827,7 +11333,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12836,16 +11342,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12865,10 +11370,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -12895,14 +11396,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12916,7 +11409,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12932,18 +11425,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12957,7 +11442,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12970,19 +11455,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13016,7 +11490,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13026,17 +11500,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13055,22 +11527,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13080,11 +11551,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13098,14 +11565,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13133,7 +11592,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13142,17 +11601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13175,7 +11630,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -13194,23 +11648,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13222,13 +11675,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -13242,7 +11691,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13261,22 +11709,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13288,13 +11735,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13323,7 +11766,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13347,7 +11789,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13355,12 +11797,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13371,14 +11812,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13388,11 +11829,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13404,10 +11845,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13431,23 +11868,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13457,11 +11893,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -13487,14 +11919,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13523,7 +11947,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13532,17 +11956,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13561,22 +11981,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13586,11 +12005,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -13604,14 +12019,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13639,7 +12046,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13648,17 +12055,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13677,22 +12080,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13702,11 +12104,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -13720,14 +12118,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13755,7 +12145,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13764,17 +12154,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13793,22 +12179,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13818,11 +12203,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -13837,14 +12218,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13872,7 +12245,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13881,17 +12254,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13910,22 +12279,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13935,11 +12303,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -13953,14 +12317,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13988,7 +12344,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13997,24 +12353,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -14040,7 +12392,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14049,16 +12401,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14072,13 +12423,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -14104,14 +12451,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -14125,7 +12464,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14139,18 +12478,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -14164,7 +12495,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14176,19 +12507,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -14222,7 +12542,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14234,17 +12554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -14268,7 +12586,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -14293,23 +12611,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14323,13 +12640,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -14351,14 +12664,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14368,7 +12673,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -14377,18 +12682,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14398,7 +12695,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -14406,19 +12703,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14450,7 +12737,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14463,20 +12750,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -14501,23 +12786,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14531,13 +12815,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -14558,14 +12838,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14575,7 +12847,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -14583,18 +12855,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14604,7 +12868,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -14612,19 +12876,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -14656,7 +12909,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14669,17 +12922,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -14708,7 +12959,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -14732,16 +12982,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14752,14 +13001,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -14769,11 +13018,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14785,10 +13034,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -14797,7 +13042,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -14816,21 +13060,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14842,13 +13085,9 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -14872,7 +13111,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -14897,7 +13136,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -14905,16 +13144,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14937,10 +13175,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -14965,14 +13199,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] @@ -14984,7 +13210,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14999,18 +13225,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] @@ -15019,7 +13237,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -15030,19 +13248,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_03_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_03_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] @@ -15074,7 +13281,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15084,17 +13291,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -15104,7 +13309,6 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -15123,21 +13327,20 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15149,13 +13352,9 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -15174,21 +13373,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15198,11 +13396,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -15212,14 +13406,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15246,7 +13432,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15255,17 +13441,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -15273,7 +13455,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -15295,16 +13476,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15315,14 +13495,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -15332,11 +13512,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15348,15 +13528,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -15378,15 +13553,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15397,14 +13571,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -15414,11 +13588,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15430,8 +13604,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -15458,7 +13630,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -15484,7 +13656,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -15493,16 +13665,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15525,10 +13696,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -15557,14 +13724,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -15577,7 +13736,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15595,18 +13754,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -15619,7 +13770,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -15634,19 +13785,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -15679,7 +13819,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15689,17 +13829,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -15714,7 +13852,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -15733,22 +13870,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15760,13 +13896,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -15785,22 +13917,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15810,11 +13941,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15829,14 +13956,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15864,7 +13983,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15873,17 +13992,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -15896,7 +14011,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -15919,17 +14033,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15940,14 +14053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -15957,11 +14070,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15973,10 +14086,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -16004,7 +14113,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -16030,7 +14139,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -16040,16 +14149,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16072,10 +14180,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -16109,14 +14213,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -16131,7 +14227,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16153,18 +14249,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -16179,7 +14267,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -16198,19 +14286,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -16245,7 +14323,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16255,17 +14333,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -16294,7 +14370,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_03_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -16313,7 +14388,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -16321,16 +14396,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16342,13 +14416,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -16372,23 +14442,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16398,11 +14467,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16426,14 +14491,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16462,7 +14519,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16471,17 +14528,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -16497,7 +14550,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion] @@ -16520,18 +14572,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16542,14 +14593,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -16559,11 +14610,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16575,10 +14626,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -16605,7 +14652,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -16631,7 +14678,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -16640,16 +14687,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16672,10 +14718,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -16704,14 +14746,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -16725,7 +14759,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16743,18 +14777,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -16768,7 +14794,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -16783,19 +14809,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -16829,7 +14845,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16839,17 +14855,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -16865,7 +14879,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -16884,22 +14897,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16911,13 +14923,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -16936,22 +14944,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16961,11 +14968,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16981,14 +14984,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17016,7 +15011,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17025,17 +15020,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -17048,7 +15039,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryApplication] @@ -17071,17 +15061,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17092,14 +15081,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -17109,11 +15098,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17125,10 +15114,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -17156,7 +15141,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -17182,7 +15167,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -17192,16 +15177,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17224,10 +15208,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -17262,14 +15242,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -17284,7 +15256,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17306,18 +15278,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -17332,7 +15296,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -17351,21 +15315,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -17400,7 +15353,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17410,17 +15363,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -17449,7 +15400,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_03_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -17468,7 +15418,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -17476,16 +15426,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17497,13 +15446,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -17527,23 +15472,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17553,11 +15497,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17581,14 +15521,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17617,7 +15549,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17626,17 +15558,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -17652,7 +15580,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -17676,18 +15603,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17698,14 +15624,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -17715,11 +15641,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17731,10 +15657,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class DisksOperations: """ @@ -17757,7 +15679,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -17782,7 +15704,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -17790,16 +15712,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17819,10 +15740,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -17846,14 +15763,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -17864,7 +15773,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17878,18 +15787,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -17897,7 +15798,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -17908,19 +15809,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_03_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_03_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -17951,7 +15841,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17961,20 +15851,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -17999,7 +15887,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -18007,16 +15895,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18036,10 +15923,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -18063,14 +15946,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18081,7 +15956,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18095,18 +15970,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18114,7 +15981,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -18125,19 +15992,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_03_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_03_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18168,7 +16024,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18178,17 +16034,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -18200,7 +16054,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -18219,21 +16072,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18245,13 +16097,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -18270,21 +16118,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18294,11 +16141,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -18310,14 +16153,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18344,7 +16179,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18353,17 +16188,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -18371,7 +16202,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18393,16 +16223,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18413,14 +16242,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -18430,11 +16259,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18446,15 +16275,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -18476,15 +16300,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18495,14 +16318,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -18512,11 +16335,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18528,13 +16351,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -18560,7 +16381,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -18568,16 +16389,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18591,13 +16411,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -18623,14 +16439,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -18642,7 +16450,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18657,18 +16465,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -18680,7 +16480,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -18692,19 +16492,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -18736,7 +16525,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18748,17 +16537,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -18777,21 +16564,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18801,11 +16587,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -18819,14 +16601,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18853,7 +16627,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -18864,17 +16638,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -18898,7 +16668,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -18923,7 +16693,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -18931,16 +16701,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18960,10 +16729,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -18987,14 +16752,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19006,7 +16763,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19020,18 +16777,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19040,7 +16789,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -19051,19 +16800,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19095,7 +16833,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19105,20 +16843,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -19143,7 +16883,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -19151,16 +16891,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19180,10 +16919,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -19208,14 +16943,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19227,7 +16954,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19242,18 +16969,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19262,7 +16981,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -19273,19 +16996,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19317,7 +17029,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19327,17 +17039,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -19349,7 +17059,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -19368,21 +17077,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19394,13 +17102,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -19419,21 +17123,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19443,11 +17146,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -19459,14 +17158,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19493,7 +17184,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19502,17 +17193,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -19520,7 +17207,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19543,16 +17229,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19563,14 +17248,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -19580,11 +17265,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19596,15 +17281,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -19627,15 +17307,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19646,14 +17325,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -19663,11 +17342,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19679,13 +17358,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -19711,7 +17388,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -19719,16 +17396,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19742,13 +17418,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -19774,14 +17446,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -19793,7 +17457,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19808,18 +17472,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -19831,7 +17487,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -19843,19 +17499,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] @@ -19887,7 +17532,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19899,17 +17544,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -19928,21 +17571,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19952,11 +17594,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -19970,14 +17608,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20004,7 +17634,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -20015,14 +17645,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_compute_management_client_enums.py index c72d44925cdf..6f52af109a2d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_compute_management_client_enums.py @@ -75,20 +75,17 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference.""" - IMPORT = "Import" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -131,10 +128,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class GalleryApplicationVersionPropertiesProvisioningState(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_models_py3.py index 0c830a3f4479..ce00dae0feb6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -253,7 +249,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -278,7 +274,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -293,7 +289,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -359,7 +355,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -417,7 +413,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -516,7 +512,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet] @@ -828,11 +824,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2019_03_01.models.DiskCreateOption :ivar storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in a @@ -883,8 +878,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2019_03_01.models.DiskCreateOption :keyword storage_account_id: If createOption is Import, the Azure Resource Manager identifier of the storage account containing the blob to import as a disk. Required only if the blob is in @@ -916,7 +910,7 @@ def __init__( class DataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1073,7 +1067,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1250,7 +1244,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1322,7 +1316,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup] @@ -1457,7 +1451,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -1652,7 +1646,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1903,7 +1897,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.Disk] @@ -2057,7 +2051,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -2151,7 +2145,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2225,7 +2219,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2326,7 +2320,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.GalleryApplication] @@ -2365,7 +2359,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2437,7 +2431,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion] @@ -2546,12 +2540,12 @@ def __init__( self.storage_account_type = storage_account_type -class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): +class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): # pylint: disable=name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -2651,7 +2645,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2019_03_01.models.ManagedArtifact @@ -2766,7 +2760,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2916,7 +2910,7 @@ def __init__( class GalleryImageIdentifier(_serialization.Model): """This is the gallery Image Definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery Image Definition publisher. Required. :vartype publisher: str @@ -2956,7 +2950,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -2992,7 +2986,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3067,7 +3061,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Image Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion] @@ -3105,7 +3099,7 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -3219,7 +3213,7 @@ def __init__(self, **kwargs: Any) -> None: class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.Gallery] @@ -3262,25 +3256,11 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2019_03_01.models.HostCaching """ - _validation = { - "size_in_gb": {"readonly": True}, - "host_caching": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__(self, **kwargs: Any) -> None: - """ """ - super().__init__(**kwargs) - class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2019_03_01.models.AccessLevel @@ -3315,7 +3295,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -3372,7 +3352,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -3429,7 +3409,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3507,7 +3487,7 @@ def __init__( class ImageDataDisk(_serialization.Model): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -3600,7 +3580,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -3638,7 +3618,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.Image] @@ -3672,7 +3652,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(_serialization.Model): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from a custom image. :code:`
    `:code:`
    ` Possible values are: @@ -3903,12 +3883,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_03_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -3933,12 +3913,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_03_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4094,7 +4074,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_03_01.models.SourceVault @@ -4127,7 +4107,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_03_01.models.SourceVault @@ -4160,7 +4140,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -4193,7 +4173,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4279,7 +4259,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.Usage] @@ -4313,7 +4293,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4502,7 +4482,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -4627,7 +4607,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -4768,7 +4748,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -4801,7 +4781,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." :code:`
    `:code:`
    ` **Disallowed values:** @@ -4812,10 +4792,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -4828,16 +4808,16 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -4845,10 +4825,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_03_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2019_03_01.models.VaultSecretGroup] @@ -4893,7 +4873,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` **Windows-only restriction:** Cannot end in "." @@ -4905,10 +4885,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -4921,16 +4901,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -4938,10 +4918,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_03_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -5023,7 +5003,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5104,7 +5084,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup] @@ -5142,22 +5122,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5330,7 +5299,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5635,7 +5604,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5698,7 +5667,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -5761,7 +5730,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -5835,7 +5804,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -5882,7 +5851,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -5915,7 +5884,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.RunCommandDocumentBase] @@ -5951,7 +5920,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6139,7 +6108,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6280,7 +6249,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.Snapshot] @@ -6462,7 +6431,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -6480,7 +6449,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -6499,12 +6468,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_03_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.DataDisk] """ @@ -6531,12 +6500,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_03_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.DataDisk] """ super().__init__(**kwargs) @@ -6571,7 +6540,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -6655,7 +6624,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6672,57 +6641,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -6758,7 +6676,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -6901,7 +6819,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -6974,7 +6892,7 @@ def __init__(self, *, value: Optional[str] = None, localized_value: Optional[str class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -7021,7 +6939,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. @@ -7194,7 +7112,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7236,10 +7154,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :code:`
    `:code:`
    `This property cannot exist along with a non-null properties.virtualMachineScaleSet reference. @@ -7280,7 +7198,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -7386,10 +7304,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :code:`
    `:code:`
    `This property cannot exist along with a non-null properties.virtualMachineScaleSet reference. @@ -7427,7 +7345,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7499,7 +7417,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -7589,7 +7507,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7707,7 +7625,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -7751,7 +7669,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8032,7 +7950,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_03_01.models.UserAssignedIdentitiesValue] """ @@ -8064,7 +7982,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2019_03_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_03_01.models.UserAssignedIdentitiesValue] """ @@ -8078,7 +7996,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8134,7 +8052,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8355,7 +8273,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -8416,7 +8334,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8621,7 +8539,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -8817,10 +8735,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension] @@ -8896,7 +8814,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -8935,7 +8853,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -8946,7 +8864,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -9011,7 +8929,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -9038,7 +8956,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9192,10 +9110,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -9237,7 +9155,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet] @@ -9273,7 +9191,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetSku] @@ -9309,7 +9227,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet] @@ -9342,7 +9260,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -9372,10 +9290,10 @@ def __init__( self.storage_account_type = storage_account_type -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9462,7 +9380,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9487,7 +9405,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -9512,7 +9430,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -9526,7 +9444,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -9666,10 +9584,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -9682,16 +9600,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -9699,10 +9617,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_03_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -9745,10 +9663,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -9761,16 +9679,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -9778,10 +9696,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_03_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -9797,10 +9715,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -9861,10 +9779,12 @@ def __init__( self.public_ip_prefix = public_ip_prefix -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -9891,7 +9811,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -9899,18 +9819,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -10029,12 +9937,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetDataDisk] """ @@ -10062,12 +9970,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetDataDisk] """ @@ -10219,7 +10127,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -10335,7 +10243,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -10420,12 +10328,12 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -10452,7 +10360,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -10582,7 +10490,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -10626,7 +10534,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -10758,7 +10666,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10815,10 +10723,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2019_03_01.models.SubResource @@ -10830,7 +10738,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar model_definition_applied: Specifies whether the model applied to the virtual machine is @@ -10941,10 +10849,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2019_03_01.models.SubResource @@ -10954,7 +10862,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword protection_policy: Specifies the protection policy of the virtual machine. @@ -10984,7 +10892,7 @@ def __init__( # pylint: disable=too-many-locals self.protection_policy = protection_policy -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11037,10 +10945,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -11176,7 +11084,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM] @@ -11209,7 +11117,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -11265,7 +11173,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -11337,7 +11245,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -11554,10 +11462,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :code:`
    `:code:`
    `This property cannot exist along with a non-null properties.virtualMachineScaleSet reference. @@ -11598,7 +11506,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -11691,10 +11599,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :code:`
    `:code:`
    `This property cannot exist along with a non-null properties.virtualMachineScaleSet reference. @@ -11732,7 +11640,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -11759,7 +11667,7 @@ def __init__( self.vm_id = None -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/operations/_operations.py index 86f576a2da2f..2fbdb6648ad1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_03_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +401,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -432,7 +429,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -462,7 +459,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -496,7 +493,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -530,7 +527,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -586,7 +583,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -616,7 +613,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -640,7 +639,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -776,7 +775,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -807,7 +806,7 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -840,7 +839,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -871,7 +870,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -917,7 +916,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -952,7 +951,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -987,7 +986,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1013,7 +1012,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1053,7 +1052,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1120,7 +1119,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1168,7 +1167,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1199,7 +1198,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1228,7 +1227,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1287,7 +1286,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1349,7 +1350,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1480,7 +1481,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1511,7 +1512,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1536,7 +1537,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1561,7 +1562,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1638,7 +1639,7 @@ def build_virtual_machines_list_all_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1802,7 +1803,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1827,7 +1828,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2014,7 +2015,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2068,7 +2069,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2102,7 +2103,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2136,7 +2137,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2161,7 +2162,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2192,7 +2193,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2223,7 +2224,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2254,7 +2255,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2285,7 +2286,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2315,7 +2316,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2341,7 +2344,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2372,7 +2375,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2403,7 +2406,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2441,7 +2444,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2472,7 +2475,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2503,7 +2506,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2534,7 +2537,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2565,7 +2568,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2596,7 +2599,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2627,7 +2630,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2658,7 +2661,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2695,7 +2698,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2726,7 +2729,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2761,7 +2764,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2787,7 +2790,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2827,7 +2830,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2858,7 +2861,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2883,7 +2886,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2908,7 +2911,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2933,7 +2936,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2964,7 +2967,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2996,7 +2999,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3022,7 +3025,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3048,7 +3051,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3083,7 +3086,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3109,7 +3112,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3149,7 +3152,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3181,7 +3184,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -3227,7 +3230,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3261,7 +3264,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3287,7 +3290,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3313,7 +3316,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3339,7 +3342,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3365,7 +3368,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3400,7 +3403,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3433,7 +3436,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3466,7 +3469,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3493,7 +3498,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3620,7 +3625,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3674,7 +3679,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3773,7 +3778,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3804,7 +3809,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -3887,7 +3892,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -3925,7 +3930,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3957,7 +3962,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4024,7 +4029,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4056,7 +4061,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4087,7 +4092,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4130,7 +4135,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4174,7 +4179,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4214,7 +4219,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4370,7 +4375,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4607,7 +4612,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4744,7 +4749,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -4768,14 +4772,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4786,14 +4789,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -4803,11 +4806,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4819,8 +4822,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -4863,7 +4864,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4874,7 +4874,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4886,11 +4886,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4901,7 +4900,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -4911,12 +4910,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4944,7 +4939,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4952,16 +4947,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4973,13 +4967,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -5002,7 +4992,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5013,7 +5002,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5025,11 +5014,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5040,7 +5028,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -5050,12 +5038,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5083,7 +5067,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5091,16 +5075,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5112,13 +5095,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5130,7 +5109,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5149,21 +5127,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5173,11 +5150,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -5187,7 +5160,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5206,21 +5178,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5232,13 +5203,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -5249,7 +5216,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5271,16 +5237,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5291,14 +5256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5308,11 +5273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5324,17 +5289,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5356,16 +5316,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5376,14 +5335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5393,11 +5352,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5409,10 +5368,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -5424,7 +5379,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -5446,17 +5400,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5467,14 +5420,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5484,11 +5437,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5500,10 +5453,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -5547,7 +5496,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5558,7 +5506,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5571,11 +5519,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5586,7 +5533,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -5596,12 +5543,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5629,7 +5572,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5637,16 +5580,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5666,10 +5608,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -5692,7 +5630,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5703,7 +5640,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5716,11 +5653,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5731,7 +5667,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -5741,12 +5677,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5774,7 +5707,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5782,16 +5715,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5803,13 +5735,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5821,7 +5749,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5840,21 +5767,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5864,11 +5790,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -5880,7 +5802,6 @@ def get( :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5899,21 +5820,20 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5925,19 +5845,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5961,15 +5876,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5980,14 +5894,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5997,11 +5911,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6013,10 +5927,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -6025,7 +5935,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6049,16 +5958,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6069,14 +5977,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6086,11 +5994,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6102,10 +6010,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -6150,7 +6054,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6161,7 +6064,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6175,11 +6078,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6190,7 +6092,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -6202,12 +6104,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6235,7 +6133,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6243,16 +6141,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6272,10 +6169,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -6297,7 +6190,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6308,7 +6200,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6320,11 +6212,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6335,7 +6226,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -6345,12 +6236,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6378,7 +6265,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6386,16 +6273,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6407,13 +6293,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6425,7 +6307,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6444,21 +6325,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6468,11 +6348,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -6482,7 +6358,6 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6501,21 +6376,20 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6527,13 +6401,9 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -6542,7 +6412,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6564,16 +6433,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6584,14 +6452,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6601,11 +6469,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6617,16 +6485,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6648,15 +6511,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6667,14 +6529,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6684,11 +6546,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6700,8 +6562,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -6728,7 +6588,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -6754,7 +6614,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -6763,16 +6623,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6792,10 +6651,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -6820,14 +6675,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -6840,7 +6687,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6854,18 +6701,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -6878,7 +6717,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -6890,19 +6729,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -6935,7 +6763,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6945,24 +6773,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -6988,7 +6814,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -6997,16 +6823,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7018,13 +6843,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7050,14 +6871,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -7070,7 +6883,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7084,18 +6897,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -7108,7 +6913,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -7120,19 +6925,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] @@ -7165,7 +6959,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7175,17 +6969,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -7204,22 +6996,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7229,11 +7020,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -7247,14 +7034,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7282,7 +7061,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7291,17 +7070,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7324,7 +7099,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -7343,23 +7117,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7371,13 +7144,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -7390,7 +7159,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -7412,17 +7180,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7433,14 +7200,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -7450,11 +7217,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7466,10 +7233,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class VirtualMachineExtensionImagesOperations: """ @@ -7505,7 +7268,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -7524,23 +7286,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7552,13 +7313,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -7570,7 +7327,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7589,21 +7345,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7615,13 +7370,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -7649,7 +7400,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7668,7 +7418,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -7677,16 +7427,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7698,13 +7447,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -7732,7 +7477,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -7758,7 +7503,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -7767,16 +7512,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7796,10 +7540,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -7826,14 +7566,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7847,7 +7579,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7863,18 +7595,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7888,7 +7612,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -7901,20 +7625,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7948,7 +7661,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7958,24 +7671,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8001,7 +7712,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8010,16 +7721,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8031,13 +7741,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8066,14 +7772,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8087,7 +7785,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8103,18 +7801,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8128,7 +7818,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8141,20 +7831,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8188,7 +7867,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8198,17 +7877,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8227,22 +7904,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8252,11 +7928,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -8271,14 +7943,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8306,7 +7970,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8315,17 +7979,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -8347,7 +8007,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -8366,23 +8025,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8394,13 +8052,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8414,7 +8068,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8433,22 +8086,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8460,13 +8112,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -8505,7 +8153,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8524,7 +8171,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -8532,16 +8179,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8553,13 +8199,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8591,7 +8233,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8610,7 +8251,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -8620,16 +8261,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8641,13 +8281,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -8659,7 +8295,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8678,21 +8313,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8704,13 +8338,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -8718,7 +8348,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8737,20 +8366,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8762,13 +8390,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -8782,7 +8406,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8801,22 +8424,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8828,13 +8450,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -8864,7 +8482,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8886,16 +8503,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8906,14 +8522,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -8923,11 +8539,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8939,8 +8555,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -8969,7 +8583,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -8991,16 +8604,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9011,14 +8623,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9028,11 +8640,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9044,15 +8656,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9078,7 +8686,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9086,16 +8694,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9109,13 +8716,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -9139,14 +8742,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9159,7 +8754,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9172,18 +8767,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9196,7 +8783,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9207,19 +8794,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9252,7 +8829,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9264,20 +8841,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9302,7 +8881,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9310,16 +8889,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9339,10 +8917,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -9364,14 +8938,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9383,7 +8949,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9395,18 +8961,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9415,7 +8973,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. @@ -9424,19 +8986,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9468,7 +9019,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9478,20 +9029,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9516,7 +9069,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9524,16 +9077,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9553,10 +9105,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -9578,14 +9126,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9597,7 +9137,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9609,18 +9149,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9629,7 +9161,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -9638,19 +9174,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] @@ -9682,7 +9207,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9692,17 +9217,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -9721,21 +9244,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9745,11 +9267,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9759,14 +9277,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9793,7 +9303,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9802,17 +9312,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9827,7 +9333,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -9846,22 +9351,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9873,13 +9377,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -9891,7 +9391,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9910,21 +9409,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9936,13 +9434,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -9961,21 +9455,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9985,11 +9478,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10000,14 +9489,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10034,7 +9515,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10043,17 +9524,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10072,21 +9549,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10096,11 +9572,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10111,14 +9583,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10145,7 +9609,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10154,17 +9618,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -10176,7 +9636,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10195,21 +9654,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10219,11 +9677,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -10232,7 +9686,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -10254,16 +9707,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10274,14 +9726,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10291,11 +9743,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10307,16 +9759,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in the response to get the next page of virtual machines. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -10338,15 +9785,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10357,14 +9803,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10374,11 +9820,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10390,8 +9836,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10403,7 +9847,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -10425,17 +9868,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10446,14 +9888,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10463,11 +9905,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10479,10 +9921,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10500,22 +9938,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10525,11 +9962,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -10546,14 +9979,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10581,7 +10006,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10590,17 +10015,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10619,21 +10040,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10643,11 +10063,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10657,14 +10073,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10691,7 +10099,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10700,17 +10108,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10729,21 +10133,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10753,11 +10156,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10767,14 +10166,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10801,7 +10192,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10810,17 +10201,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10839,21 +10226,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10863,11 +10249,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10877,14 +10259,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10911,7 +10285,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10920,23 +10294,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10965,7 +10335,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10973,16 +10343,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10992,11 +10361,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -11020,14 +10385,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11038,7 +10395,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11051,18 +10408,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11073,7 +10422,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11083,19 +10432,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11125,7 +10464,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11134,17 +10473,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11163,21 +10498,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11187,11 +10521,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11201,14 +10531,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11235,7 +10557,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11244,20 +10566,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -11282,7 +10604,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11290,16 +10612,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11313,13 +10634,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -11342,14 +10659,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -11361,7 +10670,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11373,18 +10682,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -11393,7 +10694,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -11402,19 +10707,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -11446,7 +10740,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11458,17 +10752,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -11498,7 +10790,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -11520,16 +10811,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11540,14 +10830,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -11557,11 +10847,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11573,8 +10863,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -11597,7 +10885,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -11622,7 +10910,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -11630,16 +10918,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11659,10 +10946,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -11684,14 +10967,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11702,7 +10977,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11714,18 +10989,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11733,7 +11000,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -11742,19 +11009,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11785,7 +11041,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11795,20 +11051,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -11833,7 +11091,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -11841,16 +11099,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11870,10 +11127,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -11895,14 +11148,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11913,7 +11158,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11925,18 +11170,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11944,7 +11181,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -11953,19 +11194,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -11996,7 +11226,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12006,17 +11236,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -12035,21 +11263,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12059,11 +11286,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -12073,14 +11296,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12107,7 +11322,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12116,17 +11331,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12140,7 +11351,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -12159,22 +11369,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12186,13 +11395,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -12200,7 +11405,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12222,16 +11426,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12242,14 +11445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -12259,11 +11462,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12275,16 +11478,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12306,15 +11504,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12325,14 +11522,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -12342,11 +11539,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12358,8 +11555,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -12385,7 +11580,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -12411,7 +11606,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12419,16 +11614,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12448,10 +11642,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -12473,14 +11663,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12493,7 +11675,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12505,18 +11687,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12529,7 +11703,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -12538,20 +11712,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12584,7 +11747,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12594,23 +11757,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -12636,7 +11797,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12644,16 +11805,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12665,13 +11825,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -12694,14 +11850,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12714,7 +11862,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12726,18 +11874,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12750,7 +11890,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -12759,20 +11899,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12805,7 +11935,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12815,17 +11945,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12844,21 +11972,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12868,11 +11995,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -12882,14 +12005,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12916,7 +12031,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12925,17 +12040,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -12945,7 +12056,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -12964,21 +12074,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12990,19 +12099,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13031,7 +12136,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13039,16 +12144,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13058,11 +12162,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -13089,14 +12189,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13107,7 +12199,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13122,18 +12214,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13144,7 +12228,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -13156,20 +12240,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13199,7 +12272,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13208,23 +12281,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -13250,7 +12319,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13258,16 +12327,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13277,11 +12345,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -13306,14 +12370,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13324,7 +12380,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13337,18 +12393,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13359,7 +12407,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -13369,20 +12417,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13412,7 +12449,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13421,17 +12458,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -13443,7 +12476,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13462,21 +12494,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13488,13 +12519,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -13502,7 +12529,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13526,16 +12552,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13546,14 +12571,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -13563,11 +12588,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13579,17 +12604,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13613,15 +12633,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13632,14 +12651,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -13649,11 +12668,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13665,8 +12684,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13678,7 +12695,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -13702,17 +12718,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13723,14 +12738,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -13740,11 +12755,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13756,10 +12771,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13770,7 +12781,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -13794,17 +12804,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13815,14 +12824,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -13832,11 +12841,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13848,15 +12857,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -13887,7 +12892,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13896,16 +12901,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13915,11 +12919,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -13951,14 +12951,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13969,7 +12961,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -13985,7 +12977,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -13993,14 +12985,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14011,7 +12995,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14025,24 +13009,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14073,7 +13046,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14082,23 +13055,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14127,7 +13096,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14135,16 +13104,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14154,11 +13122,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -14183,14 +13147,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14201,7 +13157,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14214,18 +13170,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14236,7 +13184,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -14246,20 +13194,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14289,7 +13226,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14298,23 +13235,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14343,7 +13276,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14351,16 +13284,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14370,11 +13302,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -14399,14 +13327,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14417,7 +13337,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14430,18 +13350,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14452,7 +13364,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -14462,20 +13374,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14505,7 +13406,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14514,23 +13415,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14559,7 +13456,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14567,16 +13464,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14586,11 +13482,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -14616,14 +13508,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14634,7 +13518,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14648,18 +13532,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14670,7 +13546,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -14681,20 +13557,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14724,7 +13589,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14733,23 +13598,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14778,7 +13639,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14786,16 +13647,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14805,11 +13665,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -14837,14 +13693,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14855,7 +13703,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14871,18 +13719,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14893,7 +13733,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -14906,20 +13746,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14949,7 +13778,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14958,23 +13787,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -15000,7 +13825,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15008,16 +13833,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15027,11 +13851,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -15056,14 +13876,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15074,7 +13886,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15087,18 +13899,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15109,7 +13913,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -15119,20 +13923,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15162,7 +13955,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15171,23 +13964,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15216,7 +14005,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15224,16 +14013,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15243,11 +14031,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -15273,14 +14057,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15291,7 +14067,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15305,18 +14081,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15327,7 +14095,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -15339,20 +14107,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15382,7 +14139,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15391,23 +14148,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15436,7 +14189,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15444,16 +14197,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15463,11 +14215,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -15493,14 +14241,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15511,7 +14251,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15525,18 +14265,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15547,7 +14279,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -15558,20 +14290,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15601,7 +14322,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15610,20 +14331,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -15636,7 +14353,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -15655,22 +14371,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15682,13 +14397,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -15713,7 +14424,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15724,7 +14434,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15737,11 +14447,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15752,7 +14461,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -15763,13 +14472,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15797,7 +14503,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15805,16 +14511,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15824,14 +14529,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -15856,7 +14557,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -15882,7 +14583,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -15891,16 +14592,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15920,10 +14620,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -15951,14 +14647,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -15972,7 +14660,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15988,18 +14676,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16013,7 +14693,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -16026,20 +14706,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16073,7 +14742,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16083,17 +14752,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -16112,22 +14779,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16137,11 +14803,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -16156,14 +14818,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16191,7 +14845,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16200,17 +14854,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -16232,7 +14882,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -16251,23 +14900,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16279,13 +14927,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -16297,7 +14941,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16321,17 +14964,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16342,14 +14984,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -16359,11 +15001,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16375,12 +15017,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16417,21 +15055,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16441,11 +15078,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16455,14 +15088,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16489,7 +15114,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16498,17 +15123,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16527,21 +15148,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16551,11 +15171,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -16569,14 +15185,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16603,7 +15211,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16612,17 +15220,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16641,21 +15245,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16665,11 +15268,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -16683,14 +15282,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16717,7 +15308,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16726,17 +15317,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -16748,7 +15335,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -16767,21 +15353,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16793,13 +15378,9 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -16827,7 +15408,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -16856,7 +15439,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -16865,16 +15448,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16884,11 +15466,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -16916,14 +15494,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16935,7 +15505,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16950,18 +15520,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16973,7 +15535,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -16985,20 +15549,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17029,7 +15583,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17038,17 +15592,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -17067,22 +15617,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17092,11 +15641,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -17111,14 +15656,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17146,7 +15683,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17155,17 +15692,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -17184,22 +15717,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17209,11 +15741,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -17229,14 +15757,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17264,7 +15784,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17273,24 +15793,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -17316,7 +15832,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17325,16 +15841,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17354,10 +15869,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -17384,14 +15895,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -17405,7 +15908,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17421,18 +15924,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -17446,7 +15941,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -17459,19 +15954,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -17505,7 +15989,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17515,17 +15999,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -17544,22 +16026,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17569,11 +16050,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -17587,14 +16064,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17622,7 +16091,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17631,17 +16100,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17664,7 +16129,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -17683,23 +16147,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17711,13 +16174,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -17731,7 +16190,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -17750,22 +16208,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17777,13 +16234,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -17812,7 +16265,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -17836,7 +16288,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -17844,12 +16296,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17860,14 +16311,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -17877,11 +16328,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17893,10 +16344,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -17920,23 +16367,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17946,11 +16392,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -17976,14 +16418,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18012,7 +16446,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18021,17 +16455,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18050,22 +16480,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18075,11 +16504,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -18093,14 +16518,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18128,7 +16545,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18137,17 +16554,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18166,22 +16579,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18191,11 +16603,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -18209,14 +16617,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18244,7 +16644,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18253,17 +16653,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18282,22 +16678,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18307,11 +16702,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -18326,14 +16717,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18361,7 +16744,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18370,17 +16753,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18399,22 +16778,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18424,11 +16802,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -18442,14 +16816,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18477,7 +16843,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18486,24 +16852,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -18529,7 +16891,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18538,16 +16900,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18561,13 +16922,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -18593,14 +16950,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -18613,7 +16962,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18627,18 +16976,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -18651,7 +16992,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -18663,19 +17004,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.RunCommandResult] @@ -18708,7 +17038,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18720,17 +17050,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -18754,7 +17082,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -18779,23 +17107,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18809,13 +17136,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -18837,14 +17160,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -18854,7 +17169,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -18863,18 +17178,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -18884,7 +17191,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -18892,19 +17199,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -18936,7 +17233,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18948,20 +17245,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -18986,23 +17281,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19016,13 +17310,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -19043,14 +17333,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -19060,7 +17342,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -19068,18 +17350,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -19089,7 +17363,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -19097,19 +17371,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -19141,7 +17404,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19153,17 +17416,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -19192,7 +17453,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -19216,16 +17476,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19236,14 +17495,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -19253,11 +17512,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19269,10 +17528,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -19281,7 +17536,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -19300,21 +17554,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19326,13 +17579,9 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -19356,7 +17605,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -19381,7 +17630,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -19389,16 +17638,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19421,10 +17669,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -19449,14 +17693,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19467,7 +17703,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19482,18 +17718,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19501,7 +17729,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -19512,19 +17740,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_03_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_03_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19555,7 +17772,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19565,17 +17782,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -19585,7 +17800,6 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -19604,21 +17818,20 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19630,13 +17843,9 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -19655,21 +17864,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19679,11 +17887,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19693,14 +17897,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19727,7 +17923,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19736,17 +17932,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -19754,7 +17946,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19776,16 +17967,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19796,14 +17986,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -19813,11 +18003,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19829,15 +18019,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19859,15 +18044,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19878,14 +18062,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -19895,11 +18079,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19911,8 +18095,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -19939,7 +18121,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -19965,7 +18147,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -19974,16 +18156,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20006,10 +18187,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -20038,14 +18215,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -20058,7 +18227,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20076,18 +18245,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -20100,7 +18261,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -20115,19 +18276,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] @@ -20160,7 +18310,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20170,17 +18320,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -20195,7 +18343,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -20214,22 +18361,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20241,13 +18387,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -20266,22 +18408,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20291,11 +18432,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20310,14 +18447,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20345,7 +18474,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20354,17 +18483,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -20377,7 +18502,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -20399,17 +18523,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20420,14 +18543,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -20437,11 +18560,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20453,10 +18576,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -20484,7 +18603,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -20510,7 +18629,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -20520,16 +18639,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20552,10 +18670,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -20589,14 +18703,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20611,7 +18717,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20633,18 +18739,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20659,7 +18757,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -20678,19 +18776,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20725,7 +18813,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20735,17 +18823,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -20774,7 +18860,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_03_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -20793,7 +18878,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -20801,16 +18886,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20822,13 +18906,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20852,23 +18932,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20878,11 +18957,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20906,14 +18981,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20942,7 +19009,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20951,17 +19018,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -20977,7 +19040,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryImageVersion] @@ -21000,18 +19062,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21022,14 +19083,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -21039,11 +19100,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21055,10 +19116,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -21085,7 +19142,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -21111,7 +19168,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -21120,16 +19177,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21152,10 +19208,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -21184,14 +19236,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21205,7 +19249,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21223,18 +19267,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21248,7 +19284,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -21263,19 +19299,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21309,7 +19335,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21319,17 +19345,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -21345,7 +19369,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -21364,22 +19387,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21391,13 +19413,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -21416,22 +19434,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21441,11 +19458,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21461,14 +19474,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21496,7 +19501,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21505,17 +19510,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -21528,7 +19529,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -21550,17 +19550,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21571,14 +19570,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -21588,11 +19587,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21604,10 +19603,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -21635,7 +19630,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -21661,7 +19656,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -21671,16 +19666,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21703,10 +19697,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -21741,14 +19731,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -21763,7 +19745,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21785,18 +19767,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -21811,7 +19785,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -21830,21 +19804,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -21879,7 +19842,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21889,17 +19852,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -21928,7 +19889,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_03_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -21947,7 +19907,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -21955,16 +19915,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21976,13 +19935,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -22006,23 +19961,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22032,11 +19986,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -22060,14 +20010,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22096,7 +20038,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22105,17 +20047,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -22131,7 +20069,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22155,18 +20092,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22177,14 +20113,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -22194,11 +20130,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22210,10 +20146,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class DisksOperations: """ @@ -22236,7 +20168,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -22261,7 +20193,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -22269,16 +20201,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22298,10 +20229,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -22325,14 +20252,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22343,7 +20262,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22357,18 +20276,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22376,7 +20287,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -22387,19 +20298,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_03_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_03_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22430,7 +20330,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22440,20 +20340,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -22478,7 +20376,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -22486,16 +20384,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22515,10 +20412,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -22542,14 +20435,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22560,7 +20445,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22574,18 +20459,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22593,7 +20470,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -22604,19 +20481,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_03_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_03_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22647,7 +20513,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22657,17 +20523,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -22679,7 +20543,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -22698,21 +20561,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22724,13 +20586,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -22749,21 +20607,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22773,11 +20630,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22789,14 +20642,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22823,7 +20668,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22832,17 +20677,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -22850,7 +20691,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22872,16 +20712,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22892,14 +20731,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -22909,11 +20748,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22925,15 +20764,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22955,15 +20789,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22974,14 +20807,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -22991,11 +20824,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23007,13 +20840,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -23039,7 +20870,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -23047,16 +20878,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23070,13 +20900,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -23102,14 +20928,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23120,7 +20938,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23135,18 +20953,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23157,7 +20967,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -23169,19 +20979,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23212,7 +21011,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23224,17 +21023,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -23253,21 +21050,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23277,11 +21073,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23293,14 +21085,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23327,7 +21111,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -23338,17 +21122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -23372,7 +21152,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -23397,7 +21177,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -23405,16 +21185,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23434,10 +21213,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -23461,14 +21236,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23479,7 +21246,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23493,18 +21260,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23512,7 +21271,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -23523,19 +21282,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23566,7 +21314,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23576,20 +21324,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -23614,7 +21364,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -23622,16 +21372,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23651,10 +21400,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -23679,14 +21424,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23697,7 +21434,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23712,18 +21449,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23731,7 +21460,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -23742,19 +21475,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_03_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -23785,7 +21507,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23795,17 +21517,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -23817,7 +21537,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_03_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -23836,21 +21555,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23862,13 +21580,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -23887,21 +21601,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23911,11 +21624,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23927,14 +21636,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23961,7 +21662,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23970,17 +21671,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -23988,7 +21685,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -24010,16 +21706,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24030,14 +21725,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -24047,11 +21742,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24063,15 +21758,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_03_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -24093,15 +21783,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24112,14 +21801,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -24129,11 +21818,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24145,13 +21834,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -24177,7 +21864,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -24185,16 +21872,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24208,13 +21894,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -24240,14 +21922,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -24258,7 +21932,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24273,18 +21947,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -24295,7 +21961,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -24307,19 +21973,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_03_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_03_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -24350,7 +22005,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24362,17 +22017,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -24391,21 +22044,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24415,11 +22067,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -24431,14 +22079,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24465,7 +22105,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -24476,14 +22116,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_compute_management_client.py index e62070a482b7..75055f2928ca 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -49,7 +51,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -59,7 +79,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -79,7 +99,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_configuration.py index 6d1681cbd64d..0d4d2b2b1112 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_metadata.json index d2ead15d140f..a78195586ba5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": false, + "has_public_lro_operations": false, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "resource_skus": "ResourceSkusOperations" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_compute_management_client.py index 40c446e5edc5..95bb79207e69 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -49,7 +51,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -59,7 +79,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -79,7 +101,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_configuration.py index 7feef2f0b560..97b9385ea917 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/operations/_operations.py index 4117ff029398..bd1b2e3703da 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -60,7 +60,6 @@ def list(self, *, filter: Optional[str] = None, **kwargs: Any) -> AsyncIterable[ :keyword filter: The filter to apply on the operation. Only **location** filter is supported currently. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_04_01.models.ResourceSku] @@ -83,16 +82,15 @@ def list(self, *, filter: Optional[str] = None, **kwargs: Any) -> AsyncIterable[ def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -103,14 +101,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -120,11 +118,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -135,5 +133,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/models/_models_py3.py index 11430b1392fd..5cce9bacd6c4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/models/_models_py3.py @@ -308,7 +308,7 @@ def __init__(self, **kwargs: Any) -> None: class ResourceSkusResult(_serialization.Model): """The List Resource Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the subscription. Required. :vartype value: list[~azure.mgmt.compute.v2019_04_01.models.ResourceSku] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/operations/_operations.py index 87369bc81eb4..bfbaef890b8d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_04_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -91,7 +91,6 @@ def list(self, *, filter: Optional[str] = None, **kwargs: Any) -> Iterable["_mod :keyword filter: The filter to apply on the operation. Only **location** filter is supported currently. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_04_01.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: @@ -113,16 +112,15 @@ def list(self, *, filter: Optional[str] = None, **kwargs: Any) -> Iterable["_mod def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -133,14 +131,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -150,11 +148,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -165,5 +163,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_compute_management_client.py index a70e7d8d8432..a66f22919e55 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -151,7 +153,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -251,7 +271,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_configuration.py index e23a6ea520fa..a3aa7b78671b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_metadata.json index af3532030456..b5b7fac2b25d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_compute_management_client.py index 31f57f76932e..83c79e008369 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -153,7 +155,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -233,7 +253,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -253,7 +275,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_configuration.py index 1befe24e78e0..2061af368492 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/operations/_operations.py index 4485009d93ed..ff4f6212d105 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -203,10 +202,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -235,7 +230,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -259,14 +253,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -277,14 +270,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -294,11 +287,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -310,8 +303,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -354,7 +345,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -365,7 +355,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -377,11 +367,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -392,7 +381,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -402,12 +391,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -435,7 +420,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -443,16 +428,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -464,13 +448,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -493,7 +473,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -504,7 +483,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -516,11 +495,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -531,7 +509,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -541,12 +519,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -574,7 +548,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -582,16 +556,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -603,13 +576,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -621,7 +590,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -640,21 +608,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -664,11 +631,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -678,7 +641,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -697,21 +659,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -723,13 +684,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -740,7 +697,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet] @@ -763,16 +719,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -783,14 +738,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -800,11 +755,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -816,17 +771,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet] @@ -849,16 +799,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -869,14 +818,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -886,11 +835,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -902,10 +851,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -917,7 +862,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] @@ -940,17 +884,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -961,14 +904,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -978,11 +921,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -994,10 +937,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1041,7 +980,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1052,7 +990,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1065,11 +1003,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1080,7 +1017,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1090,12 +1027,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1123,7 +1056,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1131,16 +1064,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1160,10 +1092,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1186,7 +1114,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1197,7 +1124,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1210,11 +1137,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1225,7 +1151,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1235,12 +1161,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1268,7 +1191,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1276,16 +1199,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1297,13 +1219,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1315,7 +1233,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1334,21 +1251,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1358,11 +1274,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1382,7 +1294,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1401,22 +1312,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1428,19 +1338,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1464,15 +1369,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1483,14 +1387,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1500,11 +1404,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1516,10 +1420,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1528,7 +1428,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1552,16 +1451,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1572,14 +1470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1589,11 +1487,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1605,10 +1503,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1653,7 +1547,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1664,7 +1557,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1678,11 +1571,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1693,7 +1585,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1705,12 +1597,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1738,7 +1626,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1746,16 +1634,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1775,10 +1662,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1800,7 +1683,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1811,7 +1693,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1823,11 +1705,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1838,7 +1719,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1848,12 +1729,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1881,7 +1758,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1889,16 +1766,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1910,13 +1786,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1928,7 +1800,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1947,21 +1818,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1971,11 +1841,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -1985,7 +1851,6 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -2004,21 +1869,20 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2030,13 +1894,9 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2047,7 +1907,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup] @@ -2070,16 +1929,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2090,14 +1948,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2107,11 +1965,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2123,16 +1981,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup] @@ -2155,15 +2008,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2174,14 +2026,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2191,11 +2043,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2207,8 +2059,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2235,7 +2085,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2261,7 +2111,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2270,16 +2120,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2299,10 +2148,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2327,14 +2172,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2348,7 +2185,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2362,18 +2199,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2387,7 +2216,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2399,19 +2228,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2445,7 +2263,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2455,24 +2273,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2498,7 +2314,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2507,16 +2323,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2528,13 +2343,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2560,14 +2371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2581,7 +2384,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2595,18 +2398,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2620,7 +2415,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2632,19 +2427,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2678,7 +2462,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2688,17 +2472,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2717,22 +2499,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2742,11 +2523,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2760,14 +2537,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2795,7 +2564,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2804,17 +2573,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2837,7 +2602,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2856,23 +2620,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2884,13 +2647,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2903,7 +2662,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -2926,17 +2684,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2947,14 +2704,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -2964,11 +2721,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2980,10 +2737,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class VirtualMachineExtensionImagesOperations: """ @@ -3019,7 +2772,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3038,23 +2790,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3066,13 +2817,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3084,7 +2831,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3103,21 +2849,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3129,13 +2874,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3163,7 +2904,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3182,7 +2922,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3191,16 +2931,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3212,13 +2951,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3246,7 +2981,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3272,7 +3007,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3281,16 +3016,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3310,10 +3044,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -3340,14 +3070,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3361,7 +3083,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3377,18 +3099,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3402,7 +3116,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -3415,20 +3129,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3462,7 +3165,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3472,24 +3175,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3515,7 +3216,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3524,16 +3225,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3545,13 +3245,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3580,14 +3276,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3601,7 +3289,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3617,18 +3305,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3642,7 +3322,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -3655,20 +3335,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -3702,7 +3371,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3712,17 +3381,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -3741,22 +3408,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3766,11 +3432,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3785,14 +3447,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3820,7 +3474,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3829,17 +3483,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -3861,7 +3511,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -3880,23 +3529,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3908,13 +3556,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -3928,7 +3572,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3947,22 +3590,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3974,13 +3616,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4019,7 +3657,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4038,7 +3675,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4046,16 +3683,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4067,13 +3703,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4105,7 +3737,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4124,7 +3755,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4134,16 +3765,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4155,13 +3785,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4173,7 +3799,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4192,21 +3817,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4218,13 +3842,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4232,7 +3852,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4251,20 +3870,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4276,13 +3894,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4296,7 +3910,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4315,22 +3928,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4342,13 +3954,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -4378,7 +3986,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -4400,16 +4007,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4420,14 +4026,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -4437,11 +4043,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4453,8 +4059,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -4483,7 +4087,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -4506,16 +4109,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4526,14 +4128,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -4543,11 +4145,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4559,15 +4161,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -4593,7 +4191,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4601,16 +4199,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4624,13 +4221,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -4654,14 +4247,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4674,7 +4259,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4687,18 +4272,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4711,7 +4288,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -4722,19 +4299,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -4767,7 +4334,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4779,20 +4346,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -4817,7 +4386,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -4825,16 +4394,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4854,10 +4422,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -4880,14 +4444,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4900,7 +4456,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4913,18 +4469,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4934,7 +4482,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -4944,19 +4496,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -4989,7 +4530,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4999,20 +4540,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5037,7 +4580,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5045,16 +4588,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5074,10 +4616,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -5099,14 +4637,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5119,7 +4649,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5131,18 +4661,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5152,7 +4674,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -5161,19 +4687,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5206,7 +4721,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5216,17 +4731,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5245,21 +4758,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5269,11 +4781,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5283,14 +4791,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5317,7 +4817,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5326,17 +4826,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -5351,7 +4847,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -5370,22 +4865,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5397,13 +4891,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -5415,7 +4905,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -5434,21 +4923,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5460,13 +4948,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5485,21 +4969,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5509,11 +4992,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -5530,14 +5009,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5564,7 +5035,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5573,17 +5044,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5602,21 +5069,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5626,11 +5092,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5641,14 +5103,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5675,7 +5129,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5684,17 +5138,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -5711,7 +5161,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5730,21 +5179,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5754,11 +5202,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -5767,7 +5211,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -5790,16 +5233,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5810,14 +5252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5827,11 +5269,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5843,10 +5285,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -5855,7 +5293,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -5878,16 +5315,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5898,14 +5334,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5915,11 +5351,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5931,8 +5367,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5944,7 +5378,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] @@ -5967,17 +5400,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5988,14 +5420,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6005,11 +5437,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6021,10 +5453,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -6042,22 +5470,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6067,11 +5494,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -6088,14 +5511,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6123,7 +5538,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6132,17 +5547,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6161,21 +5572,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6185,11 +5595,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6199,14 +5605,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6233,7 +5631,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6242,17 +5640,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6271,21 +5665,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6295,11 +5688,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6309,14 +5698,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6343,7 +5724,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6352,17 +5733,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6381,21 +5758,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6405,11 +5781,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6419,14 +5791,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6453,7 +5817,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6462,17 +5826,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6491,21 +5851,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6515,11 +5874,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6529,14 +5884,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6563,7 +5910,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6572,23 +5919,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -6617,7 +5960,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6625,16 +5968,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6644,11 +5986,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6672,14 +6010,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6690,7 +6020,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6703,18 +6033,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6725,7 +6047,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -6735,19 +6057,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6777,7 +6089,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6786,17 +6098,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6815,21 +6123,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6839,11 +6146,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -6856,14 +6159,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6890,7 +6185,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6899,20 +6194,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -6937,7 +6232,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6945,16 +6240,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6968,13 +6262,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -6997,14 +6287,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7017,7 +6299,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7029,18 +6311,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7050,7 +6324,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -7059,19 +6337,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7104,7 +6371,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7116,17 +6383,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -7156,7 +6421,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] @@ -7179,16 +6443,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7199,14 +6462,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7216,11 +6479,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7232,8 +6495,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -7256,7 +6517,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7281,7 +6542,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7289,16 +6550,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7318,10 +6578,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -7343,14 +6599,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7361,7 +6609,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7373,18 +6621,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7392,7 +6632,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -7401,19 +6641,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7444,7 +6673,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7454,20 +6683,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7492,7 +6723,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -7500,16 +6731,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7529,10 +6759,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -7554,14 +6780,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7572,7 +6790,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7584,18 +6802,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7603,7 +6813,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -7612,19 +6826,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7655,7 +6858,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7665,17 +6868,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -7694,21 +6895,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7718,11 +6918,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7732,14 +6928,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7766,7 +6954,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7775,17 +6963,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7799,7 +6983,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -7818,22 +7001,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7845,13 +7027,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -7859,7 +7037,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7881,16 +7058,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7901,14 +7077,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -7918,11 +7094,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7934,16 +7110,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -7965,15 +7136,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7984,14 +7154,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8001,11 +7171,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8017,8 +7187,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -8044,7 +7212,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8070,7 +7238,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8078,16 +7246,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8107,10 +7274,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -8132,14 +7295,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8152,7 +7307,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8164,18 +7319,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8188,7 +7335,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8197,20 +7344,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8243,7 +7379,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8253,23 +7389,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8295,7 +7429,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8303,16 +7437,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8324,13 +7457,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8353,14 +7482,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8373,7 +7494,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8385,18 +7506,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8409,7 +7522,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -8418,20 +7531,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8464,7 +7567,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8474,17 +7577,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8503,21 +7604,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8527,11 +7627,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8543,14 +7639,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8577,7 +7665,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8586,17 +7674,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8608,7 +7692,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -8627,21 +7710,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8653,19 +7735,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8694,7 +7772,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8702,16 +7780,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8721,11 +7798,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -8752,14 +7825,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8770,7 +7835,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8785,18 +7850,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8807,7 +7864,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -8819,20 +7876,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8862,7 +7908,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8871,23 +7917,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8913,7 +7955,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8921,16 +7963,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8940,11 +7981,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -8969,14 +8006,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8987,7 +8016,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9000,18 +8029,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9022,7 +8043,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -9032,20 +8053,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9075,7 +8085,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9084,17 +8094,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -9106,7 +8112,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9125,21 +8130,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9151,13 +8155,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -9165,7 +8165,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9189,16 +8188,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9209,14 +8207,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -9226,11 +8224,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9242,17 +8240,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9276,15 +8269,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9295,14 +8287,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -9312,11 +8304,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9328,8 +8320,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9341,7 +8331,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -9365,17 +8354,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9386,14 +8374,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -9403,11 +8391,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9419,10 +8407,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9433,7 +8417,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -9457,17 +8440,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9478,14 +8460,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -9495,11 +8477,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9511,15 +8493,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -9550,7 +8528,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9559,16 +8537,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9578,11 +8555,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -9614,14 +8587,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9632,7 +8597,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -9648,7 +8613,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -9656,14 +8621,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9674,7 +8631,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -9688,24 +8645,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9736,7 +8682,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9745,23 +8691,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9790,7 +8732,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9798,16 +8740,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9817,11 +8758,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -9846,14 +8783,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9864,7 +8793,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9877,18 +8806,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9899,7 +8820,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -9909,20 +8830,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9952,7 +8862,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9961,23 +8871,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10006,7 +8912,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10014,16 +8920,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10033,11 +8938,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10062,14 +8963,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10080,7 +8973,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10093,18 +8986,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10115,7 +9000,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -10125,20 +9010,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10168,7 +9042,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10177,23 +9051,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10222,7 +9092,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10230,16 +9100,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10249,11 +9118,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -10279,14 +9144,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10297,7 +9154,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10311,18 +9168,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10333,7 +9182,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -10344,20 +9193,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10387,7 +9225,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10396,23 +9234,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10441,7 +9275,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10449,16 +9283,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10468,11 +9301,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -10500,14 +9329,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10518,7 +9339,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10534,18 +9355,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10556,7 +9369,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -10569,20 +9382,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10612,7 +9414,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10621,23 +9423,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10663,7 +9461,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10671,16 +9469,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10690,11 +9487,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -10719,14 +9512,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10737,7 +9522,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10750,18 +9535,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10772,7 +9549,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -10782,20 +9559,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10825,7 +9591,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10834,23 +9600,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10879,7 +9641,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10887,16 +9649,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10906,11 +9667,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -10936,14 +9693,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10954,7 +9703,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10968,18 +9717,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10990,7 +9731,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11002,20 +9743,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11045,7 +9775,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11054,23 +9784,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11099,7 +9825,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11107,16 +9833,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11126,11 +9851,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -11156,14 +9877,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11174,7 +9887,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11188,18 +9901,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11210,7 +9915,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -11221,20 +9926,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11264,7 +9958,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11273,20 +9967,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -11299,7 +9989,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -11318,22 +10007,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11345,13 +10033,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -11376,7 +10060,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11387,7 +10070,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11400,11 +10083,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11415,7 +10097,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -11426,13 +10108,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11460,7 +10139,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11468,16 +10147,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11487,14 +10165,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11519,7 +10193,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -11545,7 +10219,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -11554,16 +10228,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11583,10 +10256,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -11614,14 +10283,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11635,7 +10296,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11651,18 +10312,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11676,7 +10329,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -11689,20 +10342,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11736,7 +10378,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11746,24 +10388,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -11789,7 +10429,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -11798,16 +10438,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11827,10 +10466,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -11858,14 +10493,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11879,7 +10506,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11895,18 +10522,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11920,7 +10539,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -11933,20 +10552,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11980,7 +10589,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11990,17 +10599,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -12019,22 +10626,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12044,11 +10650,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -12063,14 +10665,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12098,7 +10692,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12107,17 +10701,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -12139,7 +10729,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -12158,23 +10747,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12186,13 +10774,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12204,7 +10788,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12228,17 +10811,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12249,14 +10831,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -12266,11 +10848,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12282,12 +10864,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12324,21 +10902,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12348,11 +10925,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -12364,14 +10937,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12398,7 +10963,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12407,17 +10972,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12436,21 +10997,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12460,11 +11020,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -12478,14 +11034,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12512,7 +11060,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12521,17 +11069,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -12550,21 +11094,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12574,11 +11117,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -12592,14 +11131,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12626,7 +11157,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12635,17 +11166,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -12657,7 +11184,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -12676,21 +11202,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12702,16 +11227,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12737,7 +11258,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -12763,7 +11284,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12773,16 +11294,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12802,10 +11322,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -12834,14 +11350,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -12856,7 +11364,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12873,18 +11381,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -12899,7 +11399,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the VMSS VM extension. @@ -12913,20 +11413,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -12961,7 +11450,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12971,17 +11460,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -12989,7 +11476,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -13015,7 +11502,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13025,16 +11512,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13046,13 +11532,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -13083,14 +11565,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13105,7 +11579,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13122,18 +11596,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13148,7 +11614,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the VMSS VM extension. @@ -13162,20 +11628,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13210,7 +11665,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13220,17 +11675,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -13249,23 +11702,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13275,11 +11727,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13295,14 +11743,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13331,7 +11771,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13340,17 +11780,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13375,7 +11811,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -13394,7 +11829,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13402,16 +11837,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13423,13 +11857,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -13451,7 +11881,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13470,23 +11899,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13498,13 +11926,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -13532,7 +11956,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -13561,7 +11987,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13570,16 +11996,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13589,11 +12014,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -13621,14 +12042,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13640,7 +12053,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13655,18 +12068,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13678,7 +12083,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -13690,20 +12097,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13734,7 +12131,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13743,17 +12140,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13772,22 +12165,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13797,11 +12189,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -13816,14 +12204,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13851,7 +12231,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13860,17 +12240,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13889,22 +12265,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13914,11 +12289,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -13934,14 +12305,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13969,7 +12332,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13978,24 +12341,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -14021,7 +12380,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14030,16 +12389,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14059,10 +12417,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -14089,14 +12443,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14110,7 +12456,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14126,18 +12472,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14151,7 +12489,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -14164,19 +12502,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14210,7 +12537,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14220,17 +12547,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14249,22 +12574,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14274,11 +12598,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14292,14 +12612,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14327,7 +12639,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14336,17 +12648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14369,7 +12677,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -14388,23 +12695,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14416,13 +12722,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -14436,7 +12738,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14455,22 +12756,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14482,13 +12782,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14517,7 +12813,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14541,7 +12836,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -14549,12 +12844,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14565,14 +12859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -14582,11 +12876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14598,10 +12892,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -14625,23 +12915,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14651,11 +12940,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -14681,14 +12966,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14717,7 +12994,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14726,17 +13003,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14755,22 +13028,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14780,11 +13052,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -14798,14 +13066,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14833,7 +13093,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14842,17 +13102,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14871,22 +13127,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14896,11 +13151,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -14914,14 +13165,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14949,7 +13192,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14958,17 +13201,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14987,22 +13226,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15012,11 +13250,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -15031,14 +13265,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15066,7 +13292,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15075,17 +13301,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15104,22 +13326,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15129,11 +13350,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -15148,14 +13365,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15183,7 +13392,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15192,24 +13401,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -15235,7 +13440,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15244,16 +13449,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15267,13 +13471,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -15299,14 +13499,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -15320,7 +13512,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15334,18 +13526,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -15359,7 +13543,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -15371,19 +13555,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -15417,7 +13590,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15429,17 +13602,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -15463,7 +13634,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15488,23 +13659,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15518,13 +13688,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -15546,14 +13712,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15563,7 +13721,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15572,18 +13730,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15593,7 +13743,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -15601,19 +13751,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15645,7 +13785,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15658,20 +13798,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -15696,23 +13834,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15726,13 +13863,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -15753,14 +13886,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15770,7 +13895,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15778,18 +13903,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15799,7 +13916,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -15807,19 +13924,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -15851,7 +13957,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15864,17 +13970,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -15898,7 +14002,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -15923,7 +14027,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -15931,16 +14035,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15960,10 +14063,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -15987,14 +14086,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16005,7 +14096,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16019,18 +14110,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16038,7 +14121,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -16049,19 +14132,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_07_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_07_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16092,7 +14164,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16102,20 +14174,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -16140,7 +14210,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -16148,16 +14218,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16177,10 +14246,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -16204,14 +14269,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16222,7 +14279,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16236,18 +14293,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16255,7 +14304,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -16266,19 +14315,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_07_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_07_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16309,7 +14347,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16319,17 +14357,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -16341,7 +14377,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -16360,21 +14395,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16386,13 +14420,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -16411,21 +14441,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16435,11 +14464,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -16451,14 +14476,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16485,7 +14502,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16494,17 +14511,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -16512,7 +14525,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16534,16 +14546,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16554,14 +14565,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -16571,11 +14582,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16587,15 +14598,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -16617,15 +14623,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16636,14 +14641,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -16653,11 +14658,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16669,13 +14674,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -16701,7 +14704,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -16709,16 +14712,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16732,13 +14734,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -16764,14 +14762,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -16783,7 +14773,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16798,18 +14788,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -16821,7 +14803,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -16833,19 +14815,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -16877,7 +14848,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16889,17 +14860,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -16918,21 +14887,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16942,11 +14910,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -16960,14 +14924,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16994,7 +14950,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -17005,17 +14961,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -17039,7 +14991,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -17064,7 +15016,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -17072,16 +15024,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17101,10 +15052,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -17128,14 +15075,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17147,7 +15086,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17161,18 +15100,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17181,7 +15112,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -17192,19 +15123,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17236,7 +15156,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17246,20 +15166,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -17284,7 +15206,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -17292,16 +15214,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17321,10 +15242,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -17349,14 +15266,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17368,7 +15277,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17383,18 +15292,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17403,7 +15304,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -17414,19 +15319,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17458,7 +15352,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17468,17 +15362,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -17490,7 +15382,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -17509,21 +15400,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17535,13 +15425,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -17560,21 +15446,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17584,11 +15469,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17600,14 +15481,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17634,7 +15507,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17643,17 +15516,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -17661,7 +15530,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17684,16 +15552,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17704,14 +15571,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -17721,11 +15588,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17737,15 +15604,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -17768,15 +15630,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17787,14 +15648,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -17804,11 +15665,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17820,13 +15681,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -17852,7 +15711,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -17860,16 +15719,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17883,13 +15741,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -17915,14 +15769,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -17934,7 +15780,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17949,18 +15795,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -17972,7 +15810,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -17984,19 +15822,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] @@ -18028,7 +15855,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18040,17 +15867,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -18069,21 +15894,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18093,11 +15917,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -18111,14 +15931,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18145,7 +15957,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -18156,17 +15968,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -18193,7 +16001,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -18219,7 +16027,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -18227,16 +16035,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18256,10 +16063,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -18284,14 +16087,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18304,7 +16099,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18319,18 +16114,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18343,7 +16130,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -18355,19 +16142,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18400,7 +16177,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18410,23 +16187,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -18452,7 +16227,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -18460,16 +16235,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18489,10 +16263,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -18517,14 +16287,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18537,7 +16299,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18552,18 +16314,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18576,7 +16330,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -18588,19 +16342,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -18633,7 +16378,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18643,17 +16388,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -18667,7 +16410,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -18686,21 +16428,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18712,13 +16453,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -18737,21 +16474,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18761,11 +16497,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18779,14 +16511,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18813,7 +16537,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18822,17 +16546,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -18842,7 +16562,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -18865,16 +16584,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18885,14 +16603,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -18902,11 +16620,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18918,15 +16636,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -18949,15 +16662,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18968,14 +16680,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -18985,11 +16697,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19001,8 +16713,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - class GalleriesOperations: """ @@ -19025,7 +16735,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -19050,7 +16760,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -19058,16 +16768,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19090,10 +16799,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -19118,14 +16823,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19137,7 +16834,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19152,18 +16849,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19172,7 +16861,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -19183,19 +16872,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_07_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_07_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19227,7 +16905,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19237,20 +16915,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -19275,7 +16955,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -19283,16 +16963,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19304,13 +16983,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19335,14 +17010,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19354,7 +17021,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19368,18 +17035,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19388,7 +17047,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -19399,19 +17062,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_07_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_07_01.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -19443,7 +17095,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19453,17 +17105,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -19473,7 +17123,6 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -19492,21 +17141,20 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19518,13 +17166,9 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -19543,21 +17187,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19567,11 +17210,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -19581,14 +17220,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19615,7 +17246,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19624,17 +17255,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -19642,7 +17269,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19664,16 +17290,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19684,14 +17309,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -19701,11 +17326,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19717,15 +17342,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -19747,15 +17367,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19766,14 +17385,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -19783,11 +17402,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19799,8 +17418,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -19827,7 +17444,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -19853,7 +17470,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -19862,16 +17479,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19894,10 +17510,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -19926,14 +17538,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -19946,7 +17550,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19964,18 +17568,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -19988,7 +17584,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -20003,19 +17599,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -20048,7 +17633,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20058,24 +17643,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -20101,7 +17684,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -20110,16 +17693,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20131,13 +17713,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20166,14 +17744,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -20186,7 +17756,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20203,18 +17773,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -20227,7 +17789,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery Image Definition. @@ -20242,19 +17804,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -20287,7 +17838,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20297,17 +17848,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -20322,7 +17871,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -20341,22 +17889,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20368,13 +17915,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -20393,22 +17936,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20418,11 +17960,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20437,14 +17975,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20472,7 +18002,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20481,17 +18011,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -20504,7 +18030,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -20527,17 +18052,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20548,14 +18072,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -20565,11 +18089,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20581,10 +18105,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -20612,7 +18132,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -20638,7 +18158,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -20648,16 +18168,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20680,10 +18199,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -20717,14 +18232,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20739,7 +18246,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20761,18 +18268,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20787,7 +18286,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -20806,19 +18305,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -20853,7 +18342,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20863,17 +18352,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -20881,7 +18368,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -20907,7 +18394,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -20917,16 +18404,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20938,13 +18424,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20979,14 +18461,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -21001,7 +18475,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21023,18 +18497,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -21049,7 +18515,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery Image Version. @@ -21068,20 +18534,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -21116,7 +18571,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21126,17 +18581,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -21165,7 +18618,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -21184,7 +18636,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -21192,16 +18644,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21213,13 +18664,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -21243,23 +18690,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21269,11 +18715,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21297,14 +18739,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21333,7 +18767,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21342,17 +18776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -21368,7 +18798,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion] @@ -21391,18 +18820,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21413,14 +18841,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -21430,11 +18858,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21446,10 +18874,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -21476,7 +18900,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -21502,7 +18926,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -21511,16 +18935,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21543,10 +18966,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -21575,14 +18994,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21596,7 +19007,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21614,18 +19025,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21639,7 +19042,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -21654,19 +19057,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21700,7 +19093,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21710,24 +19103,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -21753,7 +19144,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -21762,16 +19153,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21783,13 +19173,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21819,14 +19205,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21840,7 +19218,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21858,18 +19236,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21883,7 +19253,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -21898,20 +19268,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -21945,7 +19304,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21955,17 +19314,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -21981,7 +19338,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -22000,22 +19356,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22027,13 +19382,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -22052,22 +19403,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22077,11 +19427,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22097,14 +19443,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22132,7 +19470,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22141,17 +19479,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -22164,7 +19498,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryApplication] @@ -22187,17 +19520,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22208,14 +19540,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -22225,11 +19557,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22241,10 +19573,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -22272,7 +19600,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -22298,7 +19626,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -22308,16 +19636,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22340,10 +19667,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -22378,14 +19701,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22400,7 +19715,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22422,18 +19737,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22448,7 +19755,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -22467,21 +19774,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22516,7 +19812,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22526,17 +19822,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -22544,7 +19838,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -22570,7 +19864,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -22580,16 +19874,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22601,13 +19894,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22643,14 +19932,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22665,7 +19946,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22687,18 +19968,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22713,7 +19986,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -22732,20 +20005,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -22780,7 +20043,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22790,17 +20053,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -22829,7 +20090,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -22848,7 +20108,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -22856,16 +20116,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22877,13 +20136,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -22907,23 +20162,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22933,11 +20187,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22961,14 +20211,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22997,7 +20239,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23006,17 +20248,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -23032,7 +20270,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -23056,18 +20293,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23078,14 +20314,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -23095,11 +20331,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23111,10 +20347,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class VirtualMachineRunCommandsOperations: """ @@ -23142,7 +20374,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -23166,16 +20397,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23186,14 +20416,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -23203,11 +20433,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23219,10 +20449,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -23231,7 +20457,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -23250,21 +20475,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23276,10 +20500,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_compute_management_client_enums.py index fed75dea21ad..af2ed91cd7ac 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_compute_management_client_enums.py @@ -75,20 +75,17 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference.""" - IMPORT = "Import" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -137,10 +134,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_models_py3.py index ee456e61f57c..e27fd01d2391 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -253,7 +249,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -278,7 +274,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -293,7 +289,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -359,7 +355,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -417,7 +413,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -516,7 +512,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet] @@ -828,11 +824,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2019_07_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -882,8 +877,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2019_07_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -916,7 +910,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1087,7 +1081,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1264,7 +1258,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1336,7 +1330,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup] @@ -1471,7 +1465,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -1666,7 +1660,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1844,7 +1838,7 @@ class DiskEncryptionSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1920,7 +1914,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -1983,17 +1977,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2116,7 +2099,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.Disk] @@ -2279,7 +2262,7 @@ def __init__( class Encryption(_serialization.Model): """Encryption at rest settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: ResourceId of the disk encryption set to use for enabling encryption at rest. @@ -2361,7 +2344,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -2455,7 +2438,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2529,7 +2512,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2630,7 +2613,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.GalleryApplication] @@ -2746,7 +2729,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2818,7 +2801,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion] @@ -2927,12 +2910,12 @@ def __init__( self.storage_account_type = storage_account_type -class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): +class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): # pylint: disable=name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -3085,7 +3068,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2019_07_01.models.ManagedArtifact @@ -3179,7 +3162,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -3258,7 +3241,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3417,7 +3400,7 @@ def __init__( class GalleryImageIdentifier(_serialization.Model): """This is the gallery Image Definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery Image Definition publisher. Required. :vartype publisher: str @@ -3457,7 +3440,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -3633,7 +3616,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3711,7 +3694,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Image Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion] @@ -3769,57 +3752,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) :vartype storage_account_type: str or ~azure.mgmt.compute.v2019_07_01.models.StorageAccountType """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2019_07_01.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery Image Version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS" and "Standard_ZRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2019_07_01.models.StorageAccountType - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -3922,7 +3854,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.Gallery] @@ -3967,32 +3899,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2019_07_01.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2019_07_01.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2019_07_01.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryUpdate(UpdateResource): """Specifies information about the Shared Image Gallery that you want to update. @@ -4049,7 +3955,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2019_07_01.models.AccessLevel @@ -4084,7 +3990,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -4141,7 +4047,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -4198,7 +4104,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4363,7 +4269,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2019_07_01.models.SubResource @@ -4469,7 +4375,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -4507,7 +4413,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.Image] @@ -4541,7 +4447,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2019_07_01.models.SubResource @@ -4777,12 +4683,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_07_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4807,12 +4713,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_07_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4968,7 +4874,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_07_01.models.SourceVault @@ -5001,7 +4907,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_07_01.models.SourceVault @@ -5034,7 +4940,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -5067,7 +4973,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -5153,7 +5059,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.Usage] @@ -5187,7 +5093,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5376,7 +5282,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -5512,7 +5418,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -5653,7 +5559,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5687,7 +5593,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. :code:`
    `:code:`
    ` @@ -5699,10 +5605,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5715,10 +5621,10 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5729,7 +5635,7 @@ class OSProfile(_serialization.Model): `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5737,10 +5643,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_07_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2019_07_01.models.VaultSecretGroup] @@ -5785,7 +5691,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. @@ -5798,10 +5704,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5814,10 +5720,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5828,7 +5734,7 @@ def __init__( `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5836,10 +5742,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_07_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -5921,7 +5827,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6015,7 +5921,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup] @@ -6053,22 +5959,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -6241,7 +6136,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6546,7 +6441,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6609,7 +6504,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6672,7 +6567,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6746,7 +6641,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6793,7 +6688,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6826,7 +6721,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.RunCommandDocumentBase] @@ -6862,7 +6757,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7050,7 +6945,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7200,7 +7095,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.Snapshot] @@ -7391,7 +7286,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -7409,7 +7304,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -7428,12 +7323,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_07_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.DataDisk] """ @@ -7460,12 +7355,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_07_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.DataDisk] """ super().__init__(**kwargs) @@ -7533,7 +7428,7 @@ def __init__( class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -7617,7 +7512,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7634,57 +7529,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -7720,7 +7564,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7863,7 +7707,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -7936,7 +7780,7 @@ def __init__(self, *, value: Optional[str] = None, localized_value: Optional[str class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -7983,7 +7827,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. @@ -8156,7 +8000,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8199,10 +8043,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -8247,7 +8091,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -8354,10 +8198,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -8399,7 +8243,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -8471,7 +8315,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8561,7 +8405,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8679,7 +8523,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8723,7 +8567,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9004,7 +8848,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_07_01.models.UserAssignedIdentitiesValue] """ @@ -9036,7 +8880,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2019_07_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_07_01.models.UserAssignedIdentitiesValue] """ @@ -9050,7 +8894,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9106,7 +8950,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9327,7 +9171,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -9388,7 +9232,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9593,7 +9437,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -9818,10 +9662,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension] @@ -10007,7 +9851,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -10046,7 +9890,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -10057,7 +9901,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -10122,7 +9966,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.statuses = statuses -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -10149,7 +9993,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10303,10 +10147,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -10348,7 +10192,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet] @@ -10384,7 +10228,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetSku] @@ -10420,7 +10264,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet] @@ -10453,7 +10297,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -10497,10 +10341,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10587,7 +10431,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -10612,7 +10456,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -10637,7 +10481,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -10651,7 +10495,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10791,10 +10635,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -10807,16 +10651,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -10824,10 +10668,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_07_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -10870,10 +10714,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -10886,16 +10730,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -10903,10 +10747,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_07_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -10922,10 +10766,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10997,10 +10841,12 @@ def __init__( self.public_ip_address_version = public_ip_address_version -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -11027,7 +10873,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -11035,18 +10881,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -11165,12 +10999,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetDataDisk] """ @@ -11198,12 +11032,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetDataDisk] """ @@ -11355,7 +11189,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. :ivar id: Resource Id. @@ -11471,7 +11305,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -11556,12 +11390,12 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -11588,7 +11422,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11718,7 +11552,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -11762,7 +11596,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -11894,7 +11728,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11951,10 +11785,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2019_07_01.models.SubResource @@ -11966,7 +11800,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar model_definition_applied: Specifies whether the model applied to the virtual machine is @@ -12077,10 +11911,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2019_07_01.models.SubResource @@ -12090,7 +11924,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword protection_policy: Specifies the protection policy of the virtual machine. @@ -12120,7 +11954,7 @@ def __init__( # pylint: disable=too-many-locals self.protection_policy = protection_policy -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -12173,10 +12007,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -12312,7 +12146,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM] @@ -12345,7 +12179,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -12401,7 +12235,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -12475,7 +12309,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -12695,10 +12529,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -12743,7 +12577,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -12837,10 +12671,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -12882,7 +12716,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -12909,7 +12743,7 @@ def __init__( self.vm_id = None -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/operations/_operations.py index f0cb1dc46307..4f77c17c51a5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_07_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -415,7 +410,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -441,7 +438,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -471,7 +468,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -505,7 +502,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -539,7 +536,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -595,7 +592,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -625,7 +622,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -649,7 +648,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -785,7 +784,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -816,7 +815,7 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -849,7 +848,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -880,7 +879,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -926,7 +925,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -961,7 +960,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -996,7 +995,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1022,7 +1021,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1062,7 +1061,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1129,7 +1128,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1177,7 +1176,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1208,7 +1207,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1237,7 +1236,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1296,7 +1295,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1358,7 +1359,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1489,7 +1490,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1520,7 +1521,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1545,7 +1546,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1570,7 +1571,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1651,7 +1652,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1846,7 +1847,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1871,7 +1872,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2058,7 +2059,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2112,7 +2113,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2146,7 +2147,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2180,7 +2181,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2205,7 +2206,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2236,7 +2237,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2267,7 +2268,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2298,7 +2299,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2329,7 +2330,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2359,7 +2360,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2385,7 +2388,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2416,7 +2419,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2447,7 +2450,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2485,7 +2488,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2516,7 +2519,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2547,7 +2550,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2578,7 +2581,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2609,7 +2612,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2640,7 +2643,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2671,7 +2674,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2702,7 +2705,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2739,7 +2742,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2770,7 +2773,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2805,7 +2808,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2840,7 +2843,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2866,7 +2869,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -2906,7 +2909,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2937,7 +2940,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2962,7 +2965,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2987,7 +2990,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3012,7 +3015,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3043,7 +3046,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3084,7 +3087,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3125,7 +3128,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3163,7 +3166,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3205,7 +3208,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3245,7 +3248,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3277,7 +3280,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3303,7 +3306,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3329,7 +3332,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3364,7 +3367,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3390,7 +3393,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3430,7 +3433,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3462,7 +3465,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -3508,7 +3511,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3542,7 +3545,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3568,7 +3571,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3594,7 +3597,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3620,7 +3623,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3646,7 +3649,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3681,7 +3684,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3714,7 +3717,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3871,7 +3874,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4108,7 +4111,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4221,7 +4224,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4255,7 +4258,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4320,7 +4323,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4351,7 +4354,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4535,7 +4538,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4589,7 +4592,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4723,7 +4726,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4754,7 +4757,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4795,7 +4798,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4878,7 +4881,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4916,7 +4919,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4948,7 +4951,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4983,7 +4986,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5050,7 +5053,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5082,7 +5085,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5113,7 +5116,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -5156,7 +5159,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -5199,7 +5202,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -5243,7 +5246,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -5283,7 +5286,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5315,7 +5318,9 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5342,7 +5347,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5397,7 +5402,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5421,14 +5425,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5439,14 +5442,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5456,11 +5459,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5472,8 +5475,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -5516,7 +5517,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5527,7 +5527,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5539,11 +5539,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5554,7 +5553,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -5564,12 +5563,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5597,7 +5592,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5605,16 +5600,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5626,13 +5620,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -5655,7 +5645,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5666,7 +5655,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5678,11 +5667,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5693,7 +5681,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -5703,12 +5691,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5736,7 +5720,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5744,16 +5728,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5765,13 +5748,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5783,7 +5762,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5802,21 +5780,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5826,11 +5803,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -5840,7 +5813,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5859,21 +5831,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5885,13 +5856,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -5902,7 +5869,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5924,16 +5890,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5944,14 +5909,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5961,11 +5926,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5977,17 +5942,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -6009,16 +5969,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6029,14 +5988,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -6046,11 +6005,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6062,10 +6021,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -6077,7 +6032,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6099,17 +6053,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6120,14 +6073,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6137,11 +6090,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6153,10 +6106,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -6200,7 +6149,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6211,7 +6159,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6224,11 +6172,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6239,7 +6186,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -6249,12 +6196,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6282,7 +6225,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6290,16 +6233,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6319,10 +6261,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -6345,7 +6283,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6356,7 +6293,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6369,11 +6306,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6384,7 +6320,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -6394,12 +6330,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6427,7 +6360,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6435,16 +6368,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6456,13 +6388,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6474,7 +6402,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6493,21 +6420,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6517,11 +6443,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -6541,7 +6463,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6560,22 +6481,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6587,19 +6507,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6623,15 +6538,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6642,14 +6556,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6659,11 +6573,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6675,10 +6589,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -6687,7 +6597,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6711,16 +6620,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6731,14 +6639,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6748,11 +6656,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6764,10 +6672,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -6812,7 +6716,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6823,7 +6726,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6837,11 +6740,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6852,7 +6754,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -6864,12 +6766,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6897,7 +6795,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6905,16 +6803,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6934,10 +6831,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -6959,7 +6852,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6970,7 +6862,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6982,11 +6874,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6997,7 +6888,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -7007,12 +6898,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7040,7 +6927,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -7048,16 +6935,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7069,13 +6955,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7087,7 +6969,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7106,21 +6987,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7130,11 +7010,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -7144,7 +7020,6 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7163,21 +7038,20 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7189,13 +7063,9 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -7204,7 +7074,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -7226,16 +7095,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7246,14 +7114,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7263,11 +7131,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7279,16 +7147,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -7310,15 +7173,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7329,14 +7191,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7346,11 +7208,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7362,8 +7224,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -7390,7 +7250,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7416,7 +7276,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7425,16 +7285,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7454,10 +7313,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -7482,14 +7337,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7502,7 +7349,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7516,18 +7363,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7540,7 +7379,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -7552,19 +7391,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7597,7 +7425,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7607,24 +7435,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7650,7 +7476,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7659,16 +7485,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7680,13 +7505,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7712,14 +7533,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7732,7 +7545,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7746,18 +7559,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7770,7 +7575,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -7782,19 +7587,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] @@ -7827,7 +7621,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7837,17 +7631,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -7866,22 +7658,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7891,11 +7682,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -7909,14 +7696,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7944,7 +7723,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7953,17 +7732,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7986,7 +7761,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -8005,23 +7779,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8033,13 +7806,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -8052,7 +7821,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -8074,17 +7842,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8095,14 +7862,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -8112,11 +7879,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8128,10 +7895,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class VirtualMachineExtensionImagesOperations: """ @@ -8167,7 +7930,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8186,23 +7948,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8214,13 +7975,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -8232,7 +7989,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8251,21 +8007,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8277,13 +8032,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -8311,7 +8062,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8330,7 +8080,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -8339,16 +8089,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8360,13 +8109,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -8394,7 +8139,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8420,7 +8165,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8429,16 +8174,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8458,10 +8202,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -8488,14 +8228,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8509,7 +8241,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8525,18 +8257,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8550,7 +8274,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8563,20 +8287,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8610,7 +8323,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8620,24 +8333,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8663,7 +8374,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8672,16 +8383,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8693,13 +8403,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8728,14 +8434,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8749,7 +8447,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8765,18 +8463,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8790,7 +8480,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8803,20 +8493,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8850,7 +8529,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8860,17 +8539,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8889,22 +8566,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8914,11 +8590,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -8933,14 +8605,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8968,7 +8632,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8977,17 +8641,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9009,7 +8669,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9028,23 +8687,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9056,13 +8714,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9076,7 +8730,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9095,22 +8748,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9122,13 +8774,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -9167,7 +8815,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9186,7 +8833,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9194,16 +8841,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9215,13 +8861,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9253,7 +8895,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9272,7 +8913,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9282,16 +8923,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9303,13 +8943,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -9321,7 +8957,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9340,21 +8975,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9366,13 +9000,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -9380,7 +9010,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9399,20 +9028,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9424,13 +9052,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -9444,7 +9068,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9463,22 +9086,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9490,13 +9112,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -9526,7 +9144,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9548,16 +9165,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9568,14 +9184,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -9585,11 +9201,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9601,8 +9217,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -9631,7 +9245,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -9653,16 +9266,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9673,14 +9285,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9690,11 +9302,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9706,15 +9318,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9740,7 +9348,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9748,16 +9356,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9771,13 +9378,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -9801,14 +9404,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9821,7 +9416,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9834,18 +9429,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9858,7 +9445,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9869,19 +9456,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9914,7 +9491,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9926,20 +9503,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9964,7 +9543,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9972,16 +9551,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10001,10 +9579,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -10027,14 +9601,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10046,7 +9612,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10059,18 +9625,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10079,7 +9637,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -10089,19 +9651,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10133,7 +9684,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10143,20 +9694,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -10181,7 +9734,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10189,16 +9742,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10218,10 +9770,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -10243,14 +9791,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10262,7 +9802,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10274,18 +9814,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10294,7 +9826,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -10303,19 +9839,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] @@ -10347,7 +9872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10357,17 +9882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10386,21 +9909,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10410,11 +9932,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10424,14 +9942,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10458,7 +9968,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10467,17 +9977,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10492,7 +9998,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10511,22 +10016,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10538,13 +10042,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -10556,7 +10056,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10575,21 +10074,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10601,13 +10099,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10626,21 +10120,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10650,11 +10143,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10669,14 +10158,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10703,7 +10184,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10712,17 +10193,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10741,21 +10218,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10765,11 +10241,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10780,14 +10252,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10814,7 +10278,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10823,17 +10287,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -10850,7 +10310,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10869,21 +10328,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10893,11 +10351,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -10906,7 +10360,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -10928,16 +10381,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10948,14 +10400,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10965,11 +10417,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10981,10 +10433,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -10993,7 +10441,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -11015,16 +10462,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11035,14 +10481,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11052,11 +10498,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11068,8 +10514,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11081,7 +10525,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -11103,17 +10546,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11124,14 +10566,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -11141,11 +10583,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11157,10 +10599,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -11178,22 +10616,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11203,11 +10640,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -11224,14 +10657,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11259,7 +10684,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11268,17 +10693,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11297,21 +10718,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11321,11 +10741,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11335,14 +10751,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11369,7 +10777,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11378,17 +10786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11407,21 +10811,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11431,11 +10834,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11445,14 +10844,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11479,7 +10870,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11488,17 +10879,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11517,21 +10904,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11541,11 +10927,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11555,14 +10937,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11589,7 +10963,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11598,17 +10972,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11627,21 +10997,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11651,11 +11020,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11665,14 +11030,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11699,7 +11056,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11708,23 +11065,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11753,7 +11106,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11761,16 +11114,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11780,11 +11132,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -11808,14 +11156,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11826,7 +11166,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11839,18 +11179,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11861,7 +11193,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11871,19 +11203,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11913,7 +11235,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11922,17 +11244,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11951,21 +11269,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11975,11 +11292,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11990,14 +11303,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12024,7 +11329,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12033,20 +11338,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12071,7 +11376,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12079,16 +11384,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12102,13 +11406,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -12131,14 +11431,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -12150,7 +11442,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12162,18 +11454,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -12182,7 +11466,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12191,19 +11479,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -12235,7 +11512,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12247,17 +11524,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -12287,7 +11562,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -12309,16 +11583,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12329,14 +11602,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12346,11 +11619,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12362,8 +11635,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12386,7 +11657,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12411,7 +11682,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12419,16 +11690,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12448,10 +11718,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -12473,14 +11739,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12491,7 +11749,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12503,18 +11761,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12522,7 +11772,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -12531,19 +11781,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12574,7 +11813,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12584,20 +11823,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12622,7 +11863,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12630,16 +11871,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12659,10 +11899,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -12684,14 +11920,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12702,7 +11930,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12714,18 +11942,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12733,7 +11953,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -12742,19 +11966,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12785,7 +11998,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12795,17 +12008,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -12824,21 +12035,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12848,11 +12058,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -12862,14 +12068,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12896,7 +12094,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12905,17 +12103,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12929,7 +12123,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -12948,22 +12141,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12975,13 +12167,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -12989,7 +12177,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13011,16 +12198,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13031,14 +12217,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13048,11 +12234,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13064,16 +12250,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13095,15 +12276,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13114,14 +12294,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13131,11 +12311,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13147,8 +12327,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -13174,7 +12352,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13200,7 +12378,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13208,16 +12386,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13237,10 +12414,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -13262,14 +12435,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13282,7 +12447,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13294,18 +12459,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13318,7 +12475,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -13327,20 +12484,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13373,7 +12519,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13383,23 +12529,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13425,7 +12569,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13433,16 +12577,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13454,13 +12597,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -13483,14 +12622,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13503,7 +12634,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13515,18 +12646,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13539,7 +12662,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -13548,20 +12671,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13594,7 +12707,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13604,17 +12717,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13633,21 +12744,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13657,11 +12767,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13671,14 +12777,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13705,7 +12803,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13714,17 +12812,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -13734,7 +12828,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -13753,21 +12846,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13779,19 +12871,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13820,7 +12908,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13828,16 +12916,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13847,11 +12934,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -13878,14 +12961,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13896,7 +12971,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13911,18 +12986,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13933,7 +13000,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -13945,20 +13012,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13988,7 +13044,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13997,23 +13053,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -14039,7 +13091,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14047,16 +13099,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14066,11 +13117,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -14095,14 +13142,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14113,7 +13152,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14126,18 +13165,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14148,7 +13179,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -14158,20 +13189,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14201,7 +13221,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14210,17 +13230,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -14232,7 +13248,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14251,21 +13266,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14277,13 +13291,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -14291,7 +13301,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14315,16 +13324,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14335,14 +13343,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -14352,11 +13360,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14368,17 +13376,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14402,15 +13405,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14421,14 +13423,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -14438,11 +13440,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14454,8 +13456,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14467,7 +13467,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -14491,17 +13490,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14512,14 +13510,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -14529,11 +13527,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14545,10 +13543,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14559,7 +13553,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -14583,17 +13576,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14604,14 +13596,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -14621,11 +13613,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14637,15 +13629,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14676,7 +13664,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14685,16 +13673,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14704,11 +13691,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -14740,14 +13723,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14758,7 +13733,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -14774,7 +13749,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -14782,14 +13757,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14800,7 +13767,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14814,24 +13781,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14862,7 +13818,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14871,23 +13827,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14916,7 +13868,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14924,16 +13876,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14943,11 +13894,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -14972,14 +13919,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14990,7 +13929,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15003,18 +13942,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15025,7 +13956,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -15035,20 +13966,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15078,7 +13998,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15087,23 +14007,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15132,7 +14048,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15140,16 +14056,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15159,11 +14074,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -15188,14 +14099,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15206,7 +14109,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15219,18 +14122,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15241,7 +14136,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -15251,20 +14146,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15294,7 +14178,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15303,23 +14187,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15348,7 +14228,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15356,16 +14236,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15375,11 +14254,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -15405,14 +14280,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15423,7 +14290,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15437,18 +14304,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15459,7 +14318,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -15470,20 +14329,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15513,7 +14361,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15522,23 +14370,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15567,7 +14411,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15575,16 +14419,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15594,11 +14437,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -15626,14 +14465,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15644,7 +14475,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15660,18 +14491,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15682,7 +14505,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -15695,20 +14518,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15738,7 +14550,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15747,23 +14559,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -15789,7 +14597,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15797,16 +14605,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15816,11 +14623,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -15845,14 +14648,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15863,7 +14658,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15876,18 +14671,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15898,7 +14685,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -15908,20 +14695,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15951,7 +14727,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15960,23 +14736,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16005,7 +14777,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16013,16 +14785,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16032,11 +14803,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -16062,14 +14829,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16080,7 +14839,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16094,18 +14853,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16116,7 +14867,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -16128,20 +14879,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16171,7 +14911,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16180,23 +14920,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16225,7 +14961,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16233,16 +14969,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16252,11 +14987,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -16282,14 +15013,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16300,7 +15023,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16314,18 +15037,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16336,7 +15051,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -16347,20 +15062,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16390,7 +15094,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16399,20 +15103,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -16425,7 +15125,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -16444,22 +15143,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16471,13 +15169,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -16502,7 +15196,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16513,7 +15206,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16526,11 +15219,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16541,7 +15233,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -16552,13 +15244,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16586,7 +15275,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16594,16 +15283,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16613,14 +15301,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16645,7 +15329,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16671,7 +15355,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16680,16 +15364,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16709,10 +15392,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -16740,14 +15419,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16761,7 +15432,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16777,18 +15448,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16802,7 +15465,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -16815,20 +15478,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16862,7 +15514,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16872,24 +15524,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16915,7 +15565,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16924,16 +15574,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16953,10 +15602,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -16984,14 +15629,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17005,7 +15642,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17021,18 +15658,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17046,7 +15675,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -17059,20 +15688,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17106,7 +15725,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17116,17 +15735,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -17145,22 +15762,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17170,11 +15786,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -17189,14 +15801,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17224,7 +15828,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17233,17 +15837,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17265,7 +15865,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -17284,23 +15883,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17312,13 +15910,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -17330,7 +15924,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17354,17 +15947,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17375,14 +15967,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -17392,11 +15984,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17408,12 +16000,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17450,21 +16038,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17474,11 +16061,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17488,14 +16071,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17522,7 +16097,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17531,17 +16106,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17560,21 +16131,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17584,11 +16154,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -17602,14 +16168,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17636,7 +16194,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17645,17 +16203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17674,21 +16228,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17698,11 +16251,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -17716,14 +16265,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17750,7 +16291,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17759,17 +16300,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -17781,7 +16318,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -17800,21 +16336,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17826,16 +16361,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17861,7 +16392,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -17887,7 +16418,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17897,16 +16428,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17926,10 +16456,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -17958,14 +16484,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -17980,7 +16498,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17997,18 +16515,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18023,7 +16533,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the VMSS VM extension. @@ -18037,20 +16547,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18085,7 +16584,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18095,17 +16594,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -18113,7 +16610,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -18139,7 +16636,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18149,16 +16646,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18170,13 +16666,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -18207,14 +16699,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18229,7 +16713,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18246,18 +16730,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18272,7 +16748,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the VMSS VM extension. @@ -18286,20 +16762,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18334,7 +16799,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18344,17 +16809,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -18373,23 +16836,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18399,11 +16861,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18419,14 +16877,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18455,7 +16905,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18464,17 +16914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -18499,7 +16945,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -18518,7 +16963,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18526,16 +16971,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18547,13 +16991,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18575,7 +17015,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18594,23 +17033,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18622,13 +17060,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -18656,7 +17090,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -18685,7 +17121,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18694,16 +17130,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18713,11 +17148,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -18745,14 +17176,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18764,7 +17187,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18779,18 +17202,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18802,7 +17217,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -18814,20 +17231,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18858,7 +17265,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18867,17 +17274,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18896,22 +17299,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18921,11 +17323,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -18940,14 +17338,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18975,7 +17365,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18984,17 +17374,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19013,22 +17399,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19038,11 +17423,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -19058,14 +17439,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19093,7 +17466,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19102,24 +17475,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -19145,7 +17514,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19154,16 +17523,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19183,10 +17551,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -19213,14 +17577,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19234,7 +17590,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19250,18 +17606,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19275,7 +17623,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -19288,19 +17636,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19334,7 +17671,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19344,17 +17681,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19373,22 +17708,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19398,11 +17732,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -19416,14 +17746,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19451,7 +17773,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19460,17 +17782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19493,7 +17811,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -19512,23 +17829,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19540,13 +17856,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -19560,7 +17872,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -19579,22 +17890,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19606,13 +17916,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19641,7 +17947,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19665,7 +17970,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -19673,12 +17978,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19689,14 +17993,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -19706,11 +18010,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19722,10 +18026,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -19749,23 +18049,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19775,11 +18074,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -19805,14 +18100,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19841,7 +18128,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19850,17 +18137,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19879,22 +18162,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19904,11 +18186,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -19922,14 +18200,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19957,7 +18227,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19966,17 +18236,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19995,22 +18261,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20020,11 +18285,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -20038,14 +18299,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20073,7 +18326,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20082,17 +18335,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20111,22 +18360,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20136,11 +18384,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -20155,14 +18399,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20190,7 +18426,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20199,17 +18435,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20228,22 +18460,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20253,11 +18484,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -20272,14 +18499,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20307,7 +18526,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20316,24 +18535,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -20359,7 +18574,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20368,16 +18583,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20391,13 +18605,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -20423,14 +18633,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -20443,7 +18645,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20457,18 +18659,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -20481,7 +18675,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -20493,19 +18687,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.RunCommandResult] @@ -20538,7 +18721,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20550,17 +18733,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -20584,7 +18765,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20609,23 +18790,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20639,13 +18819,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -20667,14 +18843,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20684,7 +18852,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20693,18 +18861,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20714,7 +18874,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20722,19 +18882,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20766,7 +18916,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20778,20 +18928,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20816,23 +18964,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20846,13 +18993,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -20873,14 +19016,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20890,7 +19025,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20898,18 +19033,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20919,7 +19046,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20927,19 +19054,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20971,7 +19087,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20983,17 +19099,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class DisksOperations: @@ -21017,7 +19131,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -21042,7 +19156,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -21050,16 +19164,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21079,10 +19192,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -21106,14 +19215,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21124,7 +19225,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21138,18 +19239,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21157,7 +19250,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -21168,19 +19261,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_07_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_07_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21211,7 +19293,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21221,20 +19303,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -21259,7 +19339,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -21267,16 +19347,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21296,10 +19375,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -21323,14 +19398,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21341,7 +19408,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21355,18 +19422,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21374,7 +19433,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -21385,19 +19444,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_07_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_07_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21428,7 +19476,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21438,17 +19486,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -21460,7 +19506,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -21479,21 +19524,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21505,13 +19549,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -21530,21 +19570,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21554,11 +19593,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -21570,14 +19605,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21604,7 +19631,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21613,17 +19640,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -21631,7 +19654,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21653,16 +19675,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21673,14 +19694,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -21690,11 +19711,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21706,15 +19727,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -21736,15 +19752,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21755,14 +19770,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -21772,11 +19787,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21788,13 +19803,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -21820,7 +19833,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -21828,16 +19841,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21851,13 +19863,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -21883,14 +19891,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -21901,7 +19901,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21916,18 +19916,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -21938,7 +19930,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -21950,19 +19942,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -21993,7 +19974,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22005,17 +19986,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -22034,21 +20013,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22058,11 +20036,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22074,14 +20048,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22108,7 +20074,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -22119,17 +20085,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -22153,7 +20115,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -22178,7 +20140,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -22186,16 +20148,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22215,10 +20176,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -22242,14 +20199,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22260,7 +20209,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22274,18 +20223,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22293,7 +20234,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -22304,19 +20245,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22347,7 +20277,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22357,20 +20287,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -22395,7 +20327,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -22403,16 +20335,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22432,10 +20363,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -22460,14 +20387,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22478,7 +20397,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22493,18 +20412,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22512,7 +20423,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -22523,19 +20438,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_07_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22566,7 +20470,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22576,17 +20480,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -22598,7 +20500,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -22617,21 +20518,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22643,13 +20543,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -22668,21 +20564,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22692,11 +20587,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22708,14 +20599,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22742,7 +20625,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22751,17 +20634,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -22769,7 +20648,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22791,16 +20669,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22811,14 +20688,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -22828,11 +20705,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22844,15 +20721,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -22874,15 +20746,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22893,14 +20764,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -22910,11 +20781,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22926,13 +20797,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -22958,7 +20827,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -22966,16 +20835,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22989,13 +20857,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -23021,14 +20885,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23039,7 +20895,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23054,18 +20910,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23076,7 +20924,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -23088,19 +20936,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_07_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -23131,7 +20968,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23143,17 +20980,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -23172,21 +21007,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23196,11 +21030,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23212,14 +21042,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23246,7 +21068,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -23257,17 +21079,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -23294,7 +21112,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -23320,7 +21138,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -23328,16 +21146,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23357,10 +21174,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -23385,14 +21198,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23404,7 +21209,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23419,18 +21224,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23442,7 +21239,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -23454,19 +21251,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23498,7 +21285,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23508,23 +21295,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -23550,7 +21335,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -23558,16 +21343,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23587,10 +21371,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -23615,14 +21395,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23634,7 +21406,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23649,18 +21421,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23672,7 +21436,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -23684,19 +21448,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] @@ -23728,7 +21483,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23738,17 +21493,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -23760,7 +21513,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -23779,21 +21531,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23805,13 +21556,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -23830,21 +21577,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23854,11 +21600,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23870,14 +21612,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23904,7 +21638,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23913,17 +21647,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -23931,7 +21661,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -23953,16 +21682,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23973,14 +21701,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -23990,11 +21718,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24006,15 +21734,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -24036,15 +21759,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24055,14 +21777,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -24072,11 +21794,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24088,8 +21810,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - class GalleriesOperations: """ @@ -24112,7 +21832,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -24137,7 +21857,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -24145,16 +21865,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24177,10 +21896,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -24205,14 +21920,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24223,7 +21930,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24238,18 +21945,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24257,7 +21956,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -24268,19 +21967,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_07_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_07_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24311,7 +21999,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24321,20 +22009,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -24359,7 +22049,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -24367,16 +22057,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24388,13 +22077,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -24419,14 +22104,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24437,7 +22114,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24451,18 +22128,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24470,7 +22139,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -24481,19 +22154,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_07_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_07_01.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24524,7 +22186,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24534,17 +22196,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -24554,7 +22214,6 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -24573,21 +22232,20 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24599,13 +22257,9 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -24624,21 +22278,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24648,11 +22301,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -24662,14 +22311,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24696,7 +22337,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24705,17 +22346,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -24723,7 +22360,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24745,16 +22381,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24765,14 +22400,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -24782,11 +22417,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24798,15 +22433,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -24828,15 +22458,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24847,14 +22476,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -24864,11 +22493,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24880,8 +22509,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -24908,7 +22535,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -24934,7 +22561,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -24943,16 +22570,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24975,10 +22601,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -25007,14 +22629,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25027,7 +22641,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25045,18 +22659,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25069,7 +22675,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -25084,19 +22690,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25129,7 +22724,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25139,24 +22734,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -25182,7 +22775,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -25191,16 +22784,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25212,13 +22804,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25247,14 +22835,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25267,7 +22847,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25284,18 +22864,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25308,7 +22880,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery Image Definition. @@ -25323,19 +22895,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] @@ -25368,7 +22929,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25378,17 +22939,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -25403,7 +22962,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -25422,22 +22980,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25449,13 +23006,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -25474,22 +23027,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25499,11 +23051,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25518,14 +23066,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25553,7 +23093,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25562,17 +23102,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -25585,7 +23121,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -25607,17 +23142,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25628,14 +23162,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -25645,11 +23179,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25661,10 +23195,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -25692,7 +23222,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -25718,7 +23248,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -25728,16 +23258,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25760,10 +23289,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -25797,14 +23322,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -25819,7 +23336,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25841,18 +23358,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -25867,7 +23376,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -25886,19 +23395,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -25933,7 +23432,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25943,17 +23442,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -25961,7 +23458,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -25987,7 +23484,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -25997,16 +23494,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26018,13 +23514,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26059,14 +23551,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -26081,7 +23565,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26103,18 +23587,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -26129,7 +23605,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery Image Version. @@ -26148,20 +23624,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -26196,7 +23661,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26206,17 +23671,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -26245,7 +23708,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -26264,7 +23726,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -26272,16 +23734,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26293,13 +23754,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -26323,23 +23780,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26349,11 +23805,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26377,14 +23829,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26413,7 +23857,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26422,17 +23866,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -26448,7 +23888,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryImageVersion] @@ -26471,18 +23910,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26493,14 +23931,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -26510,11 +23948,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26526,10 +23964,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -26556,7 +23990,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -26582,7 +24016,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -26591,16 +24025,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26623,10 +24056,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -26655,14 +24084,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -26676,7 +24097,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26694,18 +24115,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -26719,7 +24132,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -26734,19 +24147,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -26780,7 +24183,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26790,24 +24193,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -26833,7 +24234,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -26842,16 +24243,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26863,13 +24263,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26899,14 +24295,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -26920,7 +24308,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26938,18 +24326,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -26963,7 +24343,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -26978,20 +24358,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -27025,7 +24394,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27035,17 +24404,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -27061,7 +24428,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -27080,22 +24446,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27107,13 +24472,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -27132,22 +24493,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27157,11 +24517,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27177,14 +24533,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27212,7 +24560,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27221,17 +24569,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -27244,7 +24588,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_07_01.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -27266,17 +24609,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27287,14 +24629,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -27304,11 +24646,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27320,10 +24662,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -27351,7 +24689,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -27377,7 +24715,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -27387,16 +24725,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27419,10 +24756,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -27457,14 +24790,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27479,7 +24804,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27501,18 +24826,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27527,7 +24844,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -27546,21 +24863,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27595,7 +24901,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27605,17 +24911,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27623,7 +24927,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -27649,7 +24953,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -27659,16 +24963,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27680,13 +24983,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27722,14 +25021,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27744,7 +25035,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27766,18 +25057,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27792,7 +25075,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -27811,20 +25094,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -27859,7 +25132,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27869,17 +25142,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -27908,7 +25179,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -27927,7 +25197,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -27935,16 +25205,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27956,13 +25225,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -27986,23 +25251,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28012,11 +25276,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28040,14 +25300,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28076,7 +25328,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28085,17 +25337,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -28111,7 +25359,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -28135,18 +25382,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28157,14 +25403,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -28174,11 +25420,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28190,10 +25436,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class VirtualMachineRunCommandsOperations: """ @@ -28221,7 +25463,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -28245,16 +25486,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28265,14 +25505,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -28282,11 +25522,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28298,10 +25538,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -28310,7 +25546,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -28329,21 +25564,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28355,10 +25589,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_compute_management_client.py index 828c213de527..799093bdf37e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -56,7 +58,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -70,7 +90,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -90,7 +110,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_configuration.py index d58206858a0a..c582a6b0af52 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_metadata.json index 26aeff47ccec..b89d55b65ff9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_compute_management_client.py index 613778ad9a15..fdda43b2e34d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -56,7 +58,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -70,7 +90,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -90,7 +112,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_configuration.py index 22ed223e3412..e21bd1d29906 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/operations/_operations.py index 366844b98ed4..0c73dd8c51a6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -81,7 +81,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -106,7 +106,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -114,16 +114,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -143,10 +142,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -170,14 +165,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -188,7 +175,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -202,18 +189,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -221,7 +200,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -232,19 +211,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_11_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_11_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -275,7 +243,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -285,20 +253,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -323,7 +289,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -331,16 +297,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -360,10 +325,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -387,14 +348,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -405,7 +358,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -419,18 +372,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -438,7 +383,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -449,19 +394,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_11_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_11_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -492,7 +426,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -502,17 +436,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -524,7 +456,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -543,21 +474,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -569,13 +499,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -594,21 +520,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -618,11 +543,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -634,14 +555,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -668,7 +581,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -677,17 +590,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -695,7 +604,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -717,16 +625,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -737,14 +644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -754,11 +661,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -770,15 +677,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -800,15 +702,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -819,14 +720,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -836,11 +737,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -852,13 +753,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -884,7 +783,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -892,16 +791,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -915,13 +813,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -947,14 +841,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -966,7 +852,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -981,18 +867,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -1004,7 +882,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1016,19 +894,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -1060,7 +927,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1072,17 +939,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1101,21 +966,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1125,11 +989,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1143,14 +1003,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1177,7 +1029,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1188,17 +1040,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1222,7 +1070,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1247,7 +1095,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1255,16 +1103,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1284,10 +1131,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1311,14 +1154,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1330,7 +1165,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1344,18 +1179,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1364,7 +1191,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1375,19 +1202,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1419,7 +1235,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1429,20 +1245,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1467,7 +1285,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1475,16 +1293,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1504,10 +1321,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1532,14 +1345,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1551,7 +1356,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1566,18 +1371,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1586,7 +1383,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1597,19 +1398,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1641,7 +1431,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1651,17 +1441,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1673,7 +1461,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1692,21 +1479,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1718,13 +1504,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1743,21 +1525,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1767,11 +1548,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1783,14 +1560,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1817,7 +1586,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1826,17 +1595,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1844,7 +1609,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1867,16 +1631,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1887,14 +1650,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1904,11 +1667,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1920,15 +1683,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.Snapshot] @@ -1951,15 +1709,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1970,14 +1727,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1987,11 +1744,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2003,13 +1760,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2035,7 +1790,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2043,16 +1798,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2066,13 +1820,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2098,14 +1848,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -2117,7 +1859,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2132,18 +1874,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -2155,7 +1889,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2167,19 +1901,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] @@ -2211,7 +1934,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2223,17 +1946,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2252,21 +1973,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2276,11 +1996,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2294,14 +2010,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2328,7 +2036,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2339,17 +2047,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2376,7 +2080,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2402,7 +2106,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2410,16 +2114,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2439,10 +2142,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2467,14 +2166,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2487,7 +2178,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2502,18 +2193,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2526,7 +2209,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2538,19 +2221,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2583,7 +2256,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2593,23 +2266,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2635,7 +2306,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2643,16 +2314,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2672,10 +2342,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2700,14 +2366,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2720,7 +2378,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2735,18 +2393,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2759,7 +2409,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2771,19 +2421,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2816,7 +2457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2826,17 +2467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2850,7 +2489,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2869,21 +2507,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2895,13 +2532,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2920,21 +2553,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2944,11 +2576,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2962,14 +2590,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2996,7 +2616,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3005,17 +2625,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3025,7 +2641,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3048,16 +2663,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3068,14 +2682,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3085,11 +2699,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3101,15 +2715,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3132,15 +2741,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3151,14 +2759,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3168,11 +2776,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3183,5 +2791,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_compute_management_client_enums.py index 2827040025d9..50608ea19486 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_compute_management_client_enums.py @@ -27,21 +27,18 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -76,10 +73,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_models_py3.py index cf9f66cc8164..d79ec959dd08 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/models/_models_py3.py @@ -131,11 +131,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2019_11_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -191,8 +190,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2019_11_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -230,7 +228,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -279,7 +277,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -499,7 +497,7 @@ class DiskEncryptionSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -575,7 +573,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -645,7 +643,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2019_11_01.models.Disk] @@ -917,7 +915,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1009,7 +1007,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2019_11_01.models.AccessLevel @@ -1042,7 +1040,7 @@ def __init__(self, *, access: Union[str, "_models.AccessLevel"], duration_in_sec class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1109,7 +1107,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_11_01.models.SourceVault @@ -1142,7 +1140,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2019_11_01.models.SourceVault @@ -1200,7 +1198,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1350,7 +1348,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2019_11_01.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/operations/_operations.py index 89dd46f287d9..2af0913a718a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_11_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -718,7 +718,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -743,7 +743,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -751,16 +751,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -780,10 +779,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -807,14 +802,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -825,7 +812,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -839,18 +826,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -858,7 +837,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -869,19 +848,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_11_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_11_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -912,7 +880,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -922,20 +890,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -960,7 +926,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -968,16 +934,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -997,10 +962,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1024,14 +985,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1042,7 +995,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1056,18 +1009,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1075,7 +1020,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1086,19 +1031,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2019_11_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2019_11_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1129,7 +1063,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1139,17 +1073,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1161,7 +1093,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1180,21 +1111,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1206,13 +1136,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1231,21 +1157,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1255,11 +1180,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1271,14 +1192,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1305,7 +1218,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1314,17 +1227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1332,7 +1241,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1354,16 +1262,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1374,14 +1281,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1391,11 +1298,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1407,15 +1314,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1437,15 +1339,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1456,14 +1357,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1473,11 +1374,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1489,13 +1390,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -1521,7 +1420,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1529,16 +1428,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,13 +1450,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -1584,14 +1478,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1602,7 +1488,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1617,18 +1503,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1639,7 +1517,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1651,19 +1529,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1694,7 +1561,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1706,17 +1573,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1735,21 +1600,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1759,11 +1623,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1775,14 +1635,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1809,7 +1661,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -1820,17 +1672,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1854,7 +1702,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1879,7 +1727,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1887,16 +1735,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1916,10 +1763,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -1943,14 +1786,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1961,7 +1796,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1975,18 +1810,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -1994,7 +1821,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -2005,19 +1832,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2048,7 +1864,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2058,20 +1874,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2096,7 +1914,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2104,16 +1922,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2133,10 +1950,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -2161,14 +1974,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2179,7 +1984,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2194,18 +1999,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2213,7 +2010,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2224,19 +2025,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2019_11_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2267,7 +2057,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2277,17 +2067,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2299,7 +2087,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2318,21 +2105,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2344,13 +2130,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2369,21 +2151,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2393,11 +2174,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2409,14 +2186,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2443,7 +2212,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2452,17 +2221,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -2470,7 +2235,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2492,16 +2256,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2512,14 +2275,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2529,11 +2292,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2545,15 +2308,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2575,15 +2333,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2594,14 +2351,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2611,11 +2368,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2627,13 +2384,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2659,7 +2414,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2667,16 +2422,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2690,13 +2444,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2722,14 +2472,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2482,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2497,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2511,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2789,19 +2523,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2019_11_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2555,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2844,17 +2567,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2873,21 +2594,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2897,11 +2617,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2913,14 +2629,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2947,7 +2655,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2958,17 +2666,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2995,7 +2699,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3021,7 +2725,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3029,16 +2733,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3058,10 +2761,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3086,14 +2785,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3105,7 +2796,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3120,18 +2811,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3143,7 +2826,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3155,19 +2838,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3199,7 +2872,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3209,23 +2882,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3251,7 +2922,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3259,16 +2930,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3288,10 +2958,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -3316,14 +2982,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3335,7 +2993,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3350,18 +3008,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3373,7 +3023,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3385,19 +3035,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] @@ -3429,7 +3070,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3439,17 +3080,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -3461,7 +3100,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3480,21 +3118,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3506,13 +3143,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3531,21 +3164,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3555,11 +3187,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3571,14 +3199,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3605,7 +3225,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3614,17 +3234,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -3632,7 +3248,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3654,16 +3269,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3674,14 +3288,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3691,11 +3305,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3707,15 +3321,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_11_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3737,15 +3346,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3756,14 +3364,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3773,11 +3381,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3788,5 +3396,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_compute_management_client.py index 1d5ce3427a4e..1b3cc07219b4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -144,7 +146,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -220,7 +240,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -240,7 +260,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_configuration.py index 803fb789babf..ba9c2d0736b8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_metadata.json index 2e2d85f01830..238505ba4c65 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_compute_management_client.py index de97881bd544..aa2cb9dba2e6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -146,7 +148,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -222,7 +242,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2019-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -242,7 +264,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_configuration.py index e8e70626c4d6..be4ddefc1a5c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2019-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/operations/_operations.py index 945d9941a39a..254cdd7b282c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -191,10 +190,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -223,7 +218,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -247,14 +241,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -265,14 +258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -282,11 +275,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -298,8 +291,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -342,7 +333,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -353,7 +343,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -365,11 +355,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -380,7 +369,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -390,12 +379,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -423,7 +408,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -431,16 +416,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -452,13 +436,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -481,7 +461,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -492,7 +471,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -504,11 +483,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -519,7 +497,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -529,12 +507,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -562,7 +536,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -570,16 +544,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -591,13 +564,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -609,7 +578,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -628,21 +596,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -652,11 +619,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -666,7 +629,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -685,21 +647,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -711,13 +672,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -728,7 +685,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet] @@ -751,16 +707,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -771,14 +726,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -788,11 +743,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -804,17 +759,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet] @@ -837,16 +787,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -857,14 +806,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -874,11 +823,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -890,10 +839,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -905,7 +850,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] @@ -928,17 +872,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -949,14 +892,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -966,11 +909,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,10 +925,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1029,7 +968,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1040,7 +978,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1053,11 +991,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1068,7 +1005,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1078,12 +1015,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1111,7 +1044,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1119,16 +1052,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1148,10 +1080,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1174,7 +1102,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1185,7 +1112,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1198,11 +1125,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1213,7 +1139,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1223,12 +1149,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1256,7 +1179,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1264,16 +1187,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1285,13 +1207,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1303,7 +1221,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1322,21 +1239,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1346,11 +1262,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1370,7 +1282,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1389,22 +1300,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1416,19 +1326,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1452,15 +1357,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1471,14 +1375,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1488,11 +1392,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1504,10 +1408,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1516,7 +1416,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1540,16 +1439,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1560,14 +1458,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1577,11 +1475,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1593,10 +1491,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1641,7 +1535,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1652,7 +1545,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1666,11 +1559,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1681,7 +1573,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1693,12 +1585,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1726,7 +1614,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1734,16 +1622,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1763,10 +1650,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1788,7 +1671,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1799,7 +1681,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1811,11 +1693,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1826,7 +1707,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1836,12 +1717,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1869,7 +1746,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1877,16 +1754,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1898,13 +1774,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1916,7 +1788,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1935,21 +1806,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1959,11 +1829,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -1973,7 +1839,6 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,21 +1857,20 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2018,13 +1882,9 @@ async def get(self, resource_group_name: str, host_group_name: str, **kwargs: An deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2035,7 +1895,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup] @@ -2058,16 +1917,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2078,14 +1936,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2095,11 +1953,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2111,16 +1969,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup] @@ -2143,15 +1996,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2162,14 +2014,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2179,11 +2031,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2195,8 +2047,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2223,7 +2073,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2249,7 +2099,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2258,16 +2108,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2287,10 +2136,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2315,14 +2160,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2336,7 +2173,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2350,18 +2187,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2375,7 +2204,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2387,19 +2216,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2433,7 +2251,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2443,24 +2261,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2486,7 +2302,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2495,16 +2311,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2516,13 +2331,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2548,14 +2359,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2569,7 +2372,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2583,18 +2386,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2608,7 +2403,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2620,19 +2415,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2666,7 +2450,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2676,17 +2460,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2705,22 +2487,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2730,11 +2511,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2748,14 +2525,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2783,7 +2552,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2792,17 +2561,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2825,7 +2590,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2844,23 +2608,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2872,13 +2635,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2891,7 +2650,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -2914,17 +2672,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2935,14 +2692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -2952,11 +2709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2968,10 +2725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -2998,7 +2751,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3022,15 +2774,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3041,14 +2792,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3058,11 +2809,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3074,8 +2825,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -3085,7 +2834,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3109,16 +2857,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3129,14 +2876,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3146,11 +2893,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3162,10 +2909,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -3187,7 +2930,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3198,7 +2940,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3210,11 +2952,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,7 +2966,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -3235,12 +2976,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3268,7 +3005,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3276,16 +3013,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3305,10 +3041,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -3330,7 +3062,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3341,7 +3072,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3353,11 +3084,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3368,7 +3098,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -3378,12 +3108,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3411,7 +3138,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3419,16 +3146,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3440,13 +3166,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -3458,7 +3180,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3477,21 +3198,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3501,11 +3221,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -3517,7 +3233,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3536,21 +3251,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3562,13 +3276,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -3582,7 +3292,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3601,21 +3310,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3627,13 +3335,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -3670,7 +3374,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3689,23 +3392,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3717,13 +3419,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3735,7 +3433,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3754,21 +3451,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3780,13 +3476,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3814,7 +3506,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3833,7 +3524,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3842,16 +3533,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3863,13 +3553,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3897,7 +3583,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3923,7 +3609,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3932,16 +3618,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3961,10 +3646,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -3991,14 +3672,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4012,7 +3685,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4028,18 +3701,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4053,7 +3718,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4066,20 +3731,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4113,7 +3767,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4123,24 +3777,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4166,7 +3818,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4175,16 +3827,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4196,13 +3847,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -4231,14 +3878,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4252,7 +3891,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4268,18 +3907,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4293,7 +3924,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4306,20 +3937,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4353,7 +3973,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4363,17 +3983,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4392,22 +4010,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4417,11 +4034,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4436,14 +4049,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4471,7 +4076,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4480,17 +4085,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4512,7 +4113,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4531,23 +4131,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4559,13 +4158,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4579,7 +4174,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4598,22 +4192,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4625,13 +4218,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4670,7 +4259,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4689,7 +4277,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4697,16 +4285,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4718,13 +4305,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4756,7 +4339,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4775,7 +4357,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4785,16 +4367,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4806,13 +4387,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4824,7 +4401,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4843,21 +4419,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4869,13 +4444,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4883,7 +4454,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4902,20 +4472,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4927,13 +4496,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4947,7 +4512,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4966,22 +4530,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4993,13 +4556,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5029,7 +4588,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5051,16 +4609,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5071,14 +4628,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5088,11 +4645,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5104,8 +4661,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -5134,7 +4689,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -5157,16 +4711,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5177,14 +4730,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5194,11 +4747,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5210,15 +4763,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -5244,7 +4793,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5252,16 +4801,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5275,13 +4823,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -5305,14 +4849,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5325,7 +4861,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5338,18 +4874,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5362,7 +4890,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5373,19 +4901,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5418,7 +4936,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5430,20 +4948,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5468,7 +4988,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5476,16 +4996,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5505,10 +5024,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -5531,14 +5046,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5551,7 +5058,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5564,18 +5071,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5585,7 +5084,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -5595,19 +5098,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5640,7 +5132,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5650,20 +5142,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5688,7 +5182,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5696,16 +5190,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5725,10 +5218,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -5750,14 +5239,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5770,7 +5251,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5782,18 +5263,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5803,7 +5276,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -5812,19 +5289,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5857,7 +5323,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5867,17 +5333,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -5896,21 +5360,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5920,11 +5383,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5934,14 +5393,6 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5968,7 +5419,7 @@ async def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5977,17 +5428,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6002,7 +5449,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6021,22 +5467,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6048,13 +5493,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -6066,7 +5507,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6085,21 +5525,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6111,13 +5550,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6136,21 +5571,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6160,11 +5594,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -6181,14 +5611,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6215,7 +5637,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6224,17 +5646,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6253,21 +5671,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6277,11 +5694,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6292,14 +5705,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6326,7 +5731,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6335,17 +5740,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -6362,7 +5763,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6381,21 +5781,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6405,11 +5804,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -6418,7 +5813,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -6441,16 +5835,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6461,14 +5854,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6478,11 +5871,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6494,10 +5887,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -6506,7 +5895,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -6529,16 +5917,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6549,14 +5936,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6566,11 +5953,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6582,8 +5969,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6595,7 +5980,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] @@ -6618,17 +6002,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6639,14 +6022,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6656,11 +6039,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6672,10 +6055,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -6693,22 +6072,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6718,11 +6096,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -6739,14 +6113,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6774,7 +6140,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6783,17 +6149,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6812,21 +6174,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6836,11 +6197,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6850,14 +6207,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6884,7 +6233,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6893,17 +6242,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6922,21 +6267,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6946,11 +6290,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6960,14 +6300,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6994,7 +6326,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7003,17 +6335,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7032,21 +6360,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7056,11 +6383,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7070,14 +6393,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7104,7 +6419,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7113,17 +6428,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7142,21 +6453,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7166,11 +6476,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7180,14 +6486,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7214,7 +6512,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7223,23 +6521,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7268,7 +6562,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7276,16 +6570,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7295,11 +6588,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -7323,14 +6612,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7341,7 +6622,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7354,18 +6635,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7376,7 +6649,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -7386,19 +6659,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7428,7 +6691,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7437,17 +6700,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7466,21 +6725,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7490,11 +6748,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -7507,14 +6761,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7541,7 +6787,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7550,17 +6796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -7573,7 +6815,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7592,21 +6833,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7616,14 +6856,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -7648,7 +6888,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7656,16 +6896,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7679,13 +6918,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -7708,14 +6943,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7728,7 +6955,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7740,18 +6967,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7761,7 +6980,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -7770,19 +6993,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7815,7 +7027,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7827,17 +7039,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -7867,7 +7077,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] @@ -7890,16 +7099,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7910,14 +7118,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7927,11 +7135,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7943,8 +7151,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -7967,7 +7173,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -7992,7 +7198,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8000,16 +7206,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8029,10 +7234,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -8054,14 +7255,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8072,7 +7265,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8084,18 +7277,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8103,7 +7288,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -8112,19 +7297,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8155,7 +7329,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8165,20 +7339,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8203,7 +7379,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8211,16 +7387,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8240,10 +7415,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -8265,14 +7436,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8283,7 +7446,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8295,18 +7458,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8314,7 +7469,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -8323,19 +7482,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8366,7 +7514,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8376,17 +7524,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -8405,21 +7551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8429,11 +7574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -8443,14 +7584,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8477,7 +7610,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8486,17 +7619,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8510,7 +7639,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -8529,22 +7657,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8556,13 +7683,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -8570,7 +7693,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8592,16 +7714,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8612,14 +7733,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8629,11 +7750,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8645,16 +7766,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8676,15 +7792,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8695,14 +7810,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8712,11 +7827,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8728,8 +7843,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -8755,7 +7868,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8781,7 +7894,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8789,16 +7902,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8818,10 +7930,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -8843,14 +7951,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8863,7 +7963,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8875,18 +7975,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8899,7 +7991,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8908,20 +8000,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8954,7 +8035,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8964,23 +8045,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9006,7 +8085,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9014,16 +8093,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9035,13 +8113,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9064,14 +8138,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9084,7 +8150,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9096,18 +8162,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9120,7 +8178,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9129,20 +8187,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9175,7 +8223,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9185,17 +8233,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9214,21 +8260,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9238,11 +8283,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9254,14 +8295,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9288,7 +8321,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9297,17 +8330,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9319,7 +8348,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9338,21 +8366,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9364,19 +8391,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9405,7 +8428,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9413,16 +8436,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9432,11 +8454,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -9463,14 +8481,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9481,7 +8491,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9496,18 +8506,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9518,7 +8520,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9530,20 +8532,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9573,7 +8564,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9582,23 +8573,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9624,7 +8611,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9632,16 +8619,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9651,11 +8637,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -9680,14 +8662,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9698,7 +8672,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9711,18 +8685,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9733,7 +8699,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -9743,20 +8709,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9786,7 +8741,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9795,17 +8750,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -9817,7 +8768,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9836,21 +8786,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9862,13 +8811,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -9876,7 +8821,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9900,16 +8844,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9920,14 +8863,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -9937,11 +8880,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9953,17 +8896,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9987,15 +8925,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10006,14 +8943,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10023,11 +8960,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10039,8 +8976,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10052,7 +8987,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10076,17 +9010,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10097,14 +9030,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10114,11 +9047,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10130,10 +9063,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10144,7 +9073,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10168,17 +9096,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10189,14 +9116,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10206,11 +9133,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10222,15 +9149,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10261,7 +9184,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10270,16 +9193,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10289,11 +9211,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -10325,14 +9243,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10343,7 +9253,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -10359,7 +9269,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -10367,14 +9277,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10385,7 +9287,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10399,24 +9301,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10447,7 +9338,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10456,23 +9347,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10501,7 +9388,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10509,16 +9396,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10528,11 +9414,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -10557,14 +9439,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10575,7 +9449,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10588,18 +9462,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10610,7 +9476,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -10620,20 +9486,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10663,7 +9518,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10672,23 +9527,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10717,7 +9568,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10725,16 +9576,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10744,11 +9594,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10773,14 +9619,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10791,7 +9629,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10804,18 +9642,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10826,7 +9656,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -10836,20 +9666,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10879,7 +9698,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10888,23 +9707,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10933,7 +9748,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10941,16 +9756,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10960,11 +9774,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -10990,14 +9800,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11008,7 +9810,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11022,18 +9824,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11044,7 +9838,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11055,20 +9849,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11098,7 +9881,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11107,23 +9890,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11152,7 +9931,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11160,16 +9939,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11179,11 +9957,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -11211,14 +9985,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11229,7 +9995,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11245,18 +10011,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11267,7 +10025,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11280,20 +10038,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11323,7 +10070,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11332,23 +10079,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11374,7 +10117,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11382,16 +10125,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11401,11 +10143,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -11430,14 +10168,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11448,7 +10178,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11461,18 +10191,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11483,7 +10205,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11493,20 +10215,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11536,7 +10247,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11545,23 +10256,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11590,7 +10297,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11598,16 +10305,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11617,11 +10323,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11647,14 +10349,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11665,7 +10359,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11679,18 +10373,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11701,7 +10387,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11713,20 +10399,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11756,7 +10431,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11765,23 +10440,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11810,7 +10481,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11818,16 +10489,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11837,11 +10507,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -11867,14 +10533,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11885,7 +10543,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11899,18 +10557,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11921,7 +10571,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -11932,20 +10582,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11975,7 +10614,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11984,20 +10623,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12010,7 +10645,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12029,22 +10663,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12056,13 +10689,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -12087,7 +10716,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12098,7 +10726,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12111,11 +10739,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12126,7 +10753,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -12137,13 +10764,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12171,7 +10795,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12179,16 +10803,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12198,17 +10821,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12234,7 +10853,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12242,16 +10861,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12261,11 +10879,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -12289,14 +10903,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12307,7 +10913,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12320,18 +10926,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12342,7 +10940,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -12353,19 +10951,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12395,7 +10983,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12404,20 +10992,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12442,7 +11026,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12468,7 +11052,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12477,16 +11061,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12506,10 +11089,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -12537,14 +11116,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12558,7 +11129,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12574,18 +11145,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12599,7 +11162,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -12612,20 +11175,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12659,7 +11211,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12669,24 +11221,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12712,7 +11262,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12721,16 +11271,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12750,10 +11299,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -12781,14 +11326,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12802,7 +11339,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12818,18 +11355,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12843,7 +11372,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -12856,20 +11385,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12903,7 +11422,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12913,17 +11432,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -12942,22 +11459,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12967,11 +11483,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -12986,14 +11498,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13021,7 +11525,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13030,17 +11534,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13062,7 +11562,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -13081,23 +11580,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13109,13 +11607,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13127,7 +11621,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13151,17 +11644,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13172,14 +11664,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -13189,11 +11681,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13205,12 +11697,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13247,21 +11735,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13271,11 +11758,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -13287,14 +11770,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13321,7 +11796,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13330,17 +11805,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13359,21 +11830,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13383,11 +11853,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -13401,14 +11867,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13435,7 +11893,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13444,17 +11902,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13473,21 +11927,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13497,11 +11950,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -13515,14 +11964,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13549,7 +11990,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13558,17 +11999,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -13580,7 +12017,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -13599,21 +12035,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13625,16 +12060,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13660,7 +12091,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -13686,7 +12117,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13696,16 +12127,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13725,10 +12155,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -13757,14 +12183,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13779,7 +12197,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13796,18 +12214,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13822,7 +12232,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the VMSS VM extension. @@ -13836,20 +12246,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13884,7 +12283,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13894,17 +12293,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -13912,7 +12309,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -13938,7 +12335,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13948,16 +12345,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13969,13 +12365,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -14006,14 +12398,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14028,7 +12412,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14045,18 +12429,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14071,7 +12447,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the VMSS VM extension. @@ -14085,20 +12461,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14133,7 +12498,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14143,17 +12508,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -14172,23 +12535,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14198,11 +12560,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14218,14 +12576,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14254,7 +12604,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14263,17 +12613,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14298,7 +12644,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14317,7 +12662,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14325,16 +12670,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14346,13 +12690,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -14374,7 +12714,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14393,23 +12732,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14421,13 +12759,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -14455,7 +12789,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -14484,7 +12820,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14493,16 +12829,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14512,11 +12847,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -14544,14 +12875,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14563,7 +12886,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14578,18 +12901,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14601,7 +12916,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -14613,20 +12930,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14657,7 +12964,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14666,17 +12973,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14695,22 +12998,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14720,11 +13022,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -14739,14 +13037,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14774,7 +13064,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14783,17 +13073,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14812,22 +13098,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14837,11 +13122,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -14857,14 +13138,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14892,7 +13165,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14901,24 +13174,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -14944,7 +13213,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14953,16 +13222,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14982,10 +13250,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -15012,14 +13276,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15033,7 +13289,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15049,18 +13305,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15074,7 +13322,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -15087,19 +13335,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15133,7 +13370,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15143,17 +13380,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15172,22 +13407,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15197,11 +13431,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15215,14 +13445,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15250,7 +13472,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15259,17 +13481,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15292,7 +13510,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -15311,23 +13528,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15339,13 +13555,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -15359,7 +13571,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15378,22 +13589,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15405,13 +13615,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15440,7 +13646,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15464,7 +13669,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -15472,12 +13677,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15488,14 +13692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -15505,11 +13709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15521,10 +13725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -15548,23 +13748,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15574,11 +13773,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -15604,14 +13799,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15640,7 +13827,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15649,17 +13836,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15678,22 +13861,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15703,11 +13885,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -15721,14 +13899,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15756,7 +13926,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15765,17 +13935,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15794,22 +13960,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15819,11 +13984,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -15837,14 +13998,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15872,7 +14025,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15881,17 +14034,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15910,22 +14059,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15935,11 +14083,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -15954,14 +14098,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15989,7 +14125,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15998,17 +14134,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -16027,22 +14159,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16052,11 +14183,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -16071,14 +14198,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16106,7 +14225,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16115,17 +14234,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -16140,7 +14255,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16159,22 +14273,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16184,18 +14297,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -16221,7 +14330,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -16230,16 +14339,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16253,13 +14361,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -16285,14 +14389,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16306,7 +14402,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16320,18 +14416,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16345,7 +14433,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -16357,19 +14445,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16403,7 +14480,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16415,17 +14492,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -16449,7 +14524,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16474,23 +14549,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16504,13 +14578,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -16532,14 +14602,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16549,7 +14611,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16558,18 +14620,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16579,7 +14633,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16587,19 +14641,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16631,7 +14675,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16644,20 +14688,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16682,23 +14724,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16712,13 +14753,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -16739,14 +14776,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16756,7 +14785,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -16764,18 +14793,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16785,7 +14806,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -16793,19 +14814,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16837,7 +14847,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16850,17 +14860,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -16889,7 +14897,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -16913,16 +14920,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16933,14 +14939,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -16950,11 +14956,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16966,10 +14972,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -16978,7 +14980,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -16997,21 +14998,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17023,13 +15023,9 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -17053,7 +15049,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -17078,7 +15074,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -17086,16 +15082,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17118,10 +15113,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -17146,14 +15137,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17165,7 +15148,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17180,18 +15163,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17200,7 +15175,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -17211,19 +15186,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_12_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_12_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17255,7 +15219,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17265,20 +15229,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -17303,7 +15269,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -17311,16 +15277,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17332,13 +15297,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -17363,14 +15324,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17382,7 +15335,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17396,18 +15349,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17416,7 +15361,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -17427,19 +15376,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_12_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_12_01.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -17471,7 +15409,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17481,17 +15419,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -17501,7 +15437,6 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -17520,21 +15455,20 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17546,13 +15480,9 @@ async def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -17571,21 +15501,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17595,11 +15524,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17609,14 +15534,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17643,7 +15560,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17652,17 +15569,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -17670,7 +15583,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17692,16 +15604,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17712,14 +15623,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -17729,11 +15640,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17745,15 +15656,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -17775,15 +15681,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17794,14 +15699,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -17811,11 +15716,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17827,8 +15732,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -17855,7 +15758,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -17881,7 +15784,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -17890,16 +15793,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17922,10 +15824,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -17954,14 +15852,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -17974,7 +15864,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17992,18 +15882,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18016,7 +15898,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -18031,19 +15913,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18076,7 +15947,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18086,24 +15957,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -18129,7 +15998,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -18138,16 +16007,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18159,13 +16027,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -18194,14 +16058,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18214,7 +16070,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18231,18 +16087,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18255,7 +16103,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery Image Definition. @@ -18270,19 +16118,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18315,7 +16152,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18325,17 +16162,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -18350,7 +16185,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18369,22 +16203,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18396,13 +16229,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -18421,22 +16250,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18446,11 +16274,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18465,14 +16289,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18500,7 +16316,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18509,17 +16325,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -18532,7 +16344,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -18555,17 +16366,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18576,14 +16386,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -18593,11 +16403,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18609,10 +16419,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -18640,7 +16446,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -18666,7 +16472,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -18676,16 +16482,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18708,10 +16513,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -18745,14 +16546,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18767,7 +16560,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18789,18 +16582,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18815,7 +16600,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -18834,19 +16619,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -18881,7 +16656,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18891,17 +16666,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -18909,7 +16682,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -18935,7 +16708,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -18945,16 +16718,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18966,13 +16738,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19007,14 +16775,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -19029,7 +16789,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19051,18 +16811,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -19077,7 +16829,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery Image Version. @@ -19096,20 +16848,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -19144,7 +16885,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19154,17 +16895,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -19193,7 +16932,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_12_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -19212,7 +16950,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -19220,16 +16958,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19241,13 +16978,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -19271,23 +17004,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19297,11 +17029,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -19325,14 +17053,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19361,7 +17081,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19370,17 +17090,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -19396,7 +17112,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion] @@ -19419,18 +17134,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19441,14 +17155,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -19458,11 +17172,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19474,10 +17188,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -19504,7 +17214,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -19530,7 +17240,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -19539,16 +17249,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19571,10 +17280,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -19603,14 +17308,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19624,7 +17321,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19642,18 +17339,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19667,7 +17356,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -19682,19 +17371,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19728,7 +17407,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19738,24 +17417,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -19781,7 +17458,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -19790,16 +17467,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19811,13 +17487,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19847,14 +17519,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19868,7 +17532,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19886,18 +17550,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19911,7 +17567,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -19926,20 +17582,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -19973,7 +17618,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19983,17 +17628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -20009,7 +17652,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -20028,22 +17670,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20055,13 +17696,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -20080,22 +17717,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20105,11 +17741,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20125,14 +17757,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20160,7 +17784,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20169,17 +17793,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -20192,7 +17812,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryApplication] @@ -20215,17 +17834,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20236,14 +17854,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -20253,11 +17871,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20269,10 +17887,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -20300,7 +17914,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -20326,7 +17940,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -20336,16 +17950,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20368,10 +17981,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -20406,14 +18015,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20428,7 +18029,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20450,18 +18051,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20476,7 +18069,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -20495,21 +18088,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20544,7 +18126,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20554,17 +18136,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -20572,7 +18152,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -20598,7 +18178,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -20608,16 +18188,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20629,13 +18208,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20671,14 +18246,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20693,7 +18260,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20715,18 +18282,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20741,7 +18300,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -20760,20 +18319,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -20808,7 +18357,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20818,17 +18367,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -20857,7 +18404,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_12_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -20876,7 +18422,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -20884,16 +18430,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20905,13 +18450,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20935,23 +18476,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20961,11 +18501,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20989,14 +18525,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21025,7 +18553,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21034,17 +18562,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -21060,7 +18584,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -21084,18 +18607,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21106,14 +18628,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -21123,11 +18645,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21138,7 +18660,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/models/_models_py3.py index b7af51dc29c1..b280d92544bc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -230,7 +226,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -255,7 +251,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -270,7 +266,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -336,7 +332,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -394,7 +390,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet] @@ -805,7 +801,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -996,7 +992,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -1035,7 +1031,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1212,7 +1208,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1284,7 +1280,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup] @@ -1419,7 +1415,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -1665,17 +1661,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -1800,7 +1785,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1874,7 +1859,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1975,7 +1960,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.GalleryApplication] @@ -2148,7 +2133,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2220,7 +2205,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion] @@ -2331,12 +2316,12 @@ def __init__( self.storage_account_type = storage_account_type -class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): +class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): # pylint: disable=name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -2503,7 +2488,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2019_12_01.models.ManagedArtifact @@ -2597,7 +2582,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -2676,7 +2661,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2835,7 +2820,7 @@ def __init__( class GalleryImageIdentifier(_serialization.Model): """This is the gallery Image Definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery Image Definition publisher. Required. :vartype publisher: str @@ -2875,7 +2860,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -3063,7 +3048,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3141,7 +3126,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Image Versions. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion] @@ -3200,58 +3185,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) :vartype storage_account_type: str or ~azure.mgmt.compute.v2019_12_01.models.StorageAccountType """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2019_12_01.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery Image Version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2019_12_01.models.StorageAccountType - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -3366,7 +3299,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.Gallery] @@ -3411,32 +3344,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2019_12_01.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2019_12_01.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2019_12_01.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryUpdate(UpdateResourceDefinition): """Specifies information about the Shared Image Gallery that you want to update. @@ -3507,7 +3414,7 @@ class HardwareProfile(_serialization.Model): :ivar vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -3564,7 +3471,7 @@ def __init__( """ :keyword vm_size: Specifies the size of the virtual machine. For more information about virtual machine sizes, see `Sizes for virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` The available VM sizes depend on region and availability set. For a list of available sizes use these APIs: :code:`
    `:code:`
    ` `List all available virtual machine sizes in an availability set @@ -3621,7 +3528,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3786,7 +3693,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2019_12_01.models.SubResource @@ -3892,7 +3799,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.Image] @@ -3926,7 +3833,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2019_12_01.models.SubResource @@ -4162,12 +4069,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_12_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4192,12 +4099,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_12_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4352,7 +4259,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -4385,7 +4292,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4471,7 +4378,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.Usage] @@ -4505,7 +4412,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4694,7 +4601,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -4827,7 +4734,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2019_12_01.models.OrchestrationServiceNames @@ -4901,7 +4808,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -5044,7 +4951,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5077,18 +4984,6 @@ class OSDiskImageEncryption(DiskImageEncryption): :vartype disk_encryption_set_id: str """ - _attribute_map = { - "disk_encryption_set_id": {"key": "diskEncryptionSetId", "type": "str"}, - } - - def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword disk_encryption_set_id: A relative URI containing the resource ID of the disk - encryption set. - :paramtype disk_encryption_set_id: str - """ - super().__init__(disk_encryption_set_id=disk_encryption_set_id, **kwargs) - class OSProfile(_serialization.Model): """Specifies the operating system settings for the virtual machine. Some of the settings cannot be @@ -5099,7 +4994,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. :code:`
    `:code:`
    ` @@ -5111,10 +5006,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5127,10 +5022,10 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5141,7 +5036,7 @@ class OSProfile(_serialization.Model): `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5149,10 +5044,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2019_12_01.models.VaultSecretGroup] @@ -5197,7 +5092,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. @@ -5210,10 +5105,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5226,10 +5121,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5240,7 +5135,7 @@ def __init__( `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5248,10 +5143,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -5333,7 +5228,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5427,7 +5322,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup] @@ -5465,22 +5360,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5653,7 +5537,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5958,7 +5842,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6021,7 +5905,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6084,7 +5968,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6158,7 +6042,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6205,7 +6089,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6238,7 +6122,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.RunCommandDocumentBase] @@ -6274,7 +6158,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6488,7 +6372,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -6506,7 +6390,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -6517,7 +6401,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -6527,7 +6411,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -6556,7 +6440,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -6571,7 +6455,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6627,7 +6511,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource] @@ -6704,12 +6588,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_12_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.DataDisk] """ @@ -6736,12 +6620,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_12_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.DataDisk] """ super().__init__(**kwargs) @@ -6809,7 +6693,7 @@ def __init__( class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -6904,7 +6788,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6921,57 +6805,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_resource_name: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -7007,7 +6840,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7150,7 +6983,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -7223,7 +7056,7 @@ def __init__(self, *, value: Optional[str] = None, localized_value: Optional[str class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -7270,7 +7103,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. @@ -7443,7 +7276,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7486,10 +7319,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -7534,7 +7367,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -7641,10 +7474,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -7686,7 +7519,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -7758,7 +7591,7 @@ def __init__( class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -7848,7 +7681,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7966,7 +7799,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8010,7 +7843,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8291,7 +8124,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_12_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_12_01.models.UserAssignedIdentitiesValue] """ @@ -8323,7 +8156,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2019_12_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_12_01.models.UserAssignedIdentitiesValue] """ @@ -8337,7 +8170,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8393,7 +8226,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8614,7 +8447,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -8675,7 +8508,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8882,7 +8715,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -9107,10 +8940,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension] @@ -9296,7 +9129,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2019_12_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -9335,7 +9168,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -9346,7 +9179,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -9417,7 +9250,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -9444,7 +9277,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9599,10 +9432,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -9644,7 +9477,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet] @@ -9680,7 +9513,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetSku] @@ -9716,7 +9549,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet] @@ -9749,7 +9582,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. Managed OS @@ -9793,10 +9626,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9883,7 +9716,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9908,7 +9741,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_12_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -9933,7 +9766,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_12_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -9947,7 +9780,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10087,10 +9920,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -10103,16 +9936,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -10120,10 +9953,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2019_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -10166,10 +9999,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -10182,16 +10015,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -10199,10 +10032,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2019_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -10218,10 +10051,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10293,10 +10126,12 @@ def __init__( self.public_ip_address_version = public_ip_address_version -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -10323,7 +10158,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -10331,18 +10166,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -10461,12 +10284,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -10494,12 +10317,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -10651,7 +10474,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -10769,7 +10592,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -10854,12 +10677,12 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2019_12_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -10886,7 +10709,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2019_12_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11016,7 +10839,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -11060,7 +10883,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -11192,7 +11015,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11249,10 +11072,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2019_12_01.models.SubResource @@ -11264,7 +11087,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar model_definition_applied: Specifies whether the model applied to the virtual machine is @@ -11375,10 +11198,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2019_12_01.models.SubResource @@ -11388,7 +11211,7 @@ def __init__( # pylint: disable=too-many-locals :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword protection_policy: Specifies the protection policy of the virtual machine. @@ -11418,7 +11241,7 @@ def __init__( # pylint: disable=too-many-locals self.protection_policy = protection_policy -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11471,10 +11294,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -11610,7 +11433,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM] @@ -11643,7 +11466,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -11699,7 +11522,7 @@ class VirtualMachineScaleSetVMProfile(_serialization.Model): :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar priority: Specifies the priority for the virtual machines in the scale set. @@ -11773,7 +11596,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str :keyword priority: Specifies the priority for the virtual machines in the scale set. @@ -11993,10 +11816,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -12041,7 +11864,7 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :vartype license_type: str :ivar vm_id: Specifies the VM unique ID which is a 128-bits identifier that is encoded and @@ -12135,10 +11958,10 @@ def __init__( machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -12180,7 +12003,7 @@ def __init__( :code:`
    `:code:`
    ` Windows_Server :code:`
    `:code:`
    ` If this element is included in a request for an update, the value must match the initial value. This value cannot be updated. :code:`
    `:code:`
    ` For more information, see `Azure Hybrid Use Benefit for Windows Server - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Minimum api-version: 2015-06-15. :paramtype license_type: str """ @@ -12207,7 +12030,7 @@ def __init__( self.vm_id = None -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/operations/_operations.py index d97b35a931c2..ea8d0d98ebd5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2019_12_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -415,7 +410,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -441,7 +438,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -471,7 +468,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -505,7 +502,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -539,7 +536,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -595,7 +592,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -625,7 +622,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -649,7 +648,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -785,7 +784,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -816,7 +815,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -840,7 +841,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -994,7 +995,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1025,7 +1026,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1058,7 +1059,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1089,7 +1090,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -1135,7 +1136,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1170,7 +1171,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1205,7 +1206,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1231,7 +1232,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1271,7 +1272,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1338,7 +1339,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1386,7 +1387,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1417,7 +1418,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1446,7 +1447,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1505,7 +1506,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1567,7 +1570,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1698,7 +1701,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1729,7 +1732,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1754,7 +1757,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1779,7 +1782,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1860,7 +1863,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2055,7 +2058,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2080,7 +2083,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2105,7 +2108,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2292,7 +2295,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2346,7 +2349,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2380,7 +2383,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2414,7 +2417,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2439,7 +2442,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2470,7 +2473,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2501,7 +2504,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2532,7 +2535,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2563,7 +2566,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2593,7 +2596,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2619,7 +2624,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2650,7 +2655,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2681,7 +2686,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2719,7 +2724,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2750,7 +2755,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2781,7 +2786,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2812,7 +2817,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2843,7 +2848,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2874,7 +2879,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2905,7 +2910,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2936,7 +2941,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2973,7 +2978,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3004,7 +3009,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3035,7 +3040,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3070,7 +3075,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3105,7 +3110,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3131,7 +3136,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -3171,7 +3176,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3202,7 +3207,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3227,7 +3232,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3252,7 +3257,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3277,7 +3282,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3308,7 +3313,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3349,7 +3354,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3390,7 +3395,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3428,7 +3433,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3470,7 +3475,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3510,7 +3515,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3542,7 +3547,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3568,7 +3573,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3594,7 +3599,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3629,7 +3634,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3655,7 +3660,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3695,7 +3700,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3727,7 +3732,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -3773,7 +3778,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3807,7 +3812,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3833,7 +3838,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3859,7 +3864,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3885,7 +3890,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3911,7 +3916,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3937,7 +3942,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3972,7 +3977,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4005,7 +4010,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4038,7 +4043,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4065,7 +4072,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4226,7 +4233,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4280,7 +4287,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4414,7 +4421,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4445,7 +4452,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4486,7 +4493,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4569,7 +4576,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -4607,7 +4614,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4639,7 +4646,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4674,7 +4681,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4741,7 +4748,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4773,7 +4780,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4804,7 +4811,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4847,7 +4854,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4890,7 +4897,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4934,7 +4941,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -4974,7 +4981,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5030,7 +5037,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5054,14 +5060,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5072,14 +5077,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5089,11 +5094,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5105,8 +5110,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -5149,7 +5152,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5160,7 +5162,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5172,11 +5174,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5187,7 +5188,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -5197,12 +5198,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5230,7 +5227,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5238,16 +5235,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5259,13 +5255,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -5288,7 +5280,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5299,7 +5290,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5311,11 +5302,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5326,7 +5316,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -5336,12 +5326,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5369,7 +5355,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5377,16 +5363,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5398,13 +5383,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5416,7 +5397,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5435,21 +5415,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5459,11 +5438,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -5473,7 +5448,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5492,21 +5466,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5518,13 +5491,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -5535,7 +5504,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5557,16 +5525,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5577,14 +5544,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5594,11 +5561,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5610,17 +5577,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5642,16 +5604,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5662,14 +5623,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5679,11 +5640,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5695,10 +5656,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -5710,7 +5667,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -5732,17 +5688,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5753,14 +5708,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5770,11 +5725,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5786,10 +5741,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -5833,7 +5784,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5844,7 +5794,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5857,11 +5807,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5872,7 +5821,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -5882,12 +5831,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5915,7 +5860,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5923,16 +5868,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5952,10 +5896,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -5978,7 +5918,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5989,7 +5928,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6002,11 +5941,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6017,7 +5955,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -6027,12 +5965,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6060,7 +5995,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6068,16 +6003,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6089,13 +6023,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6107,7 +6037,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6126,21 +6055,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6150,11 +6078,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -6174,7 +6098,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6193,22 +6116,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6220,19 +6142,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6256,15 +6173,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6275,14 +6191,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6292,11 +6208,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6308,10 +6224,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -6320,7 +6232,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6344,16 +6255,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6364,14 +6274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6381,11 +6291,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6397,10 +6307,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -6445,7 +6351,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6456,7 +6361,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6470,11 +6375,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6485,7 +6389,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -6497,12 +6401,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6530,7 +6430,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6538,16 +6438,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6567,10 +6466,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -6592,7 +6487,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6603,7 +6497,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6615,11 +6509,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6630,7 +6523,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -6640,12 +6533,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6673,7 +6562,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6681,16 +6570,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6702,13 +6590,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6720,7 +6604,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6739,21 +6622,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6763,11 +6645,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> _models.DedicatedHostGroup: @@ -6777,7 +6655,6 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6796,21 +6673,20 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6822,13 +6698,9 @@ def get(self, resource_group_name: str, host_group_name: str, **kwargs: Any) -> deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -6837,7 +6709,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6859,16 +6730,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6879,14 +6749,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6896,11 +6766,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6912,16 +6782,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6943,15 +6808,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6962,14 +6826,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6979,11 +6843,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6995,8 +6859,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -7023,7 +6885,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7049,7 +6911,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7058,16 +6920,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7087,10 +6948,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -7115,14 +6972,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7135,7 +6984,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7149,18 +6998,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7173,7 +7014,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -7185,19 +7026,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7230,7 +7060,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7240,24 +7070,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7283,7 +7111,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7292,16 +7120,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7313,13 +7140,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7345,14 +7168,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7365,7 +7180,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7379,18 +7194,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7403,7 +7210,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -7415,19 +7222,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] @@ -7460,7 +7256,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7470,17 +7266,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -7499,22 +7293,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7524,11 +7317,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -7542,14 +7331,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7577,7 +7358,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7586,17 +7367,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7619,7 +7396,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -7638,23 +7414,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7666,13 +7441,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -7685,7 +7456,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -7707,17 +7477,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7728,14 +7497,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -7745,11 +7514,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7761,10 +7530,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -7791,7 +7556,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -7815,15 +7579,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7834,14 +7597,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -7851,11 +7614,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7867,8 +7630,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -7878,7 +7639,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -7902,16 +7662,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7922,14 +7681,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -7939,11 +7698,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7955,10 +7714,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -7980,7 +7735,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7991,7 +7745,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8003,11 +7757,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8018,7 +7771,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -8028,12 +7781,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8061,7 +7810,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -8069,16 +7818,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8098,10 +7846,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -8123,7 +7867,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8134,7 +7877,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8146,11 +7889,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8161,7 +7903,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -8171,12 +7913,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8204,7 +7943,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -8212,16 +7951,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8233,13 +7971,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -8251,7 +7985,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8270,21 +8003,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8294,11 +8026,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -8308,7 +8036,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8327,21 +8054,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8353,13 +8079,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -8373,7 +8095,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8392,21 +8113,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8418,13 +8138,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -8461,7 +8177,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8480,23 +8195,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8508,13 +8222,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -8526,7 +8236,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8545,21 +8254,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8571,13 +8279,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -8605,7 +8309,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8624,7 +8327,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -8633,16 +8336,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8654,13 +8356,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -8688,7 +8386,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8714,7 +8412,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8723,16 +8421,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8752,10 +8449,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -8782,14 +8475,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8803,7 +8488,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8819,18 +8504,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8844,7 +8521,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8857,20 +8534,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8904,7 +8570,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8914,24 +8580,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8957,7 +8621,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8966,16 +8630,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8987,13 +8650,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9022,14 +8681,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9043,7 +8694,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9059,18 +8710,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9084,7 +8727,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -9097,20 +8740,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9144,7 +8776,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9154,17 +8786,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -9183,22 +8813,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9208,11 +8837,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -9227,14 +8852,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9262,7 +8879,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9271,17 +8888,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9303,7 +8916,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9322,23 +8934,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9350,13 +8961,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9370,7 +8977,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9389,22 +8995,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9416,13 +9021,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -9461,7 +9062,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9480,7 +9080,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9488,16 +9088,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9509,13 +9108,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9547,7 +9142,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9566,7 +9160,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9576,16 +9170,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9597,13 +9190,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -9615,7 +9204,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9634,21 +9222,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9660,13 +9247,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -9674,7 +9257,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9693,20 +9275,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9718,13 +9299,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -9738,7 +9315,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9757,22 +9333,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9784,13 +9359,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -9820,7 +9391,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9842,16 +9412,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9862,14 +9431,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -9879,11 +9448,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9895,8 +9464,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -9925,7 +9492,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -9947,16 +9513,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9967,14 +9532,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9984,11 +9549,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10000,15 +9565,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -10034,7 +9595,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10042,16 +9603,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10065,13 +9625,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -10095,14 +9651,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10115,7 +9663,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10128,18 +9676,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10152,7 +9692,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -10163,19 +9703,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10208,7 +9738,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10220,20 +9750,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -10258,7 +9790,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10266,16 +9798,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10295,10 +9826,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -10321,14 +9848,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10340,7 +9859,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10353,18 +9872,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10373,7 +9884,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -10383,19 +9898,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10427,7 +9931,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10437,20 +9941,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -10475,7 +9981,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10483,16 +9989,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10512,10 +10017,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -10537,14 +10038,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10556,7 +10049,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10568,18 +10061,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10588,7 +10073,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -10597,19 +10086,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] @@ -10641,7 +10119,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10651,17 +10129,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10680,21 +10156,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10704,11 +10179,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10718,14 +10189,6 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10752,7 +10215,7 @@ def begin_delete(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10761,17 +10224,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10786,7 +10245,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10805,22 +10263,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10832,13 +10289,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -10850,7 +10303,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10869,21 +10321,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10895,13 +10346,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10920,21 +10367,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10944,11 +10390,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10963,14 +10405,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10997,7 +10431,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11006,17 +10440,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11035,21 +10465,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11059,11 +10488,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11074,14 +10499,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11108,7 +10525,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11117,17 +10534,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -11144,7 +10557,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11163,21 +10575,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11187,11 +10598,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -11200,7 +10607,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -11222,16 +10628,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11242,14 +10647,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11259,11 +10664,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11275,10 +10680,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -11287,7 +10688,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -11309,16 +10709,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11329,14 +10728,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11346,11 +10745,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11362,8 +10761,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11375,7 +10772,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -11397,17 +10793,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11418,14 +10813,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -11435,11 +10830,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11451,10 +10846,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -11472,22 +10863,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11497,11 +10887,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -11518,14 +10904,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11553,7 +10931,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11562,17 +10940,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11591,21 +10965,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11615,11 +10988,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11629,14 +10998,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11663,7 +11024,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11672,17 +11033,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11701,21 +11058,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11725,11 +11081,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11739,14 +11091,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11773,7 +11117,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11782,17 +11126,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11811,21 +11151,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11835,11 +11174,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11849,14 +11184,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11883,7 +11210,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11892,17 +11219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11921,21 +11244,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11945,11 +11267,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11959,14 +11277,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11993,7 +11303,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12002,23 +11312,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12047,7 +11353,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12055,16 +11361,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12074,11 +11379,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12102,14 +11403,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12120,7 +11413,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12133,18 +11426,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12155,7 +11440,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -12165,19 +11450,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12207,7 +11482,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12216,17 +11491,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12245,21 +11516,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12269,11 +11539,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -12284,14 +11550,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12318,7 +11576,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12327,17 +11585,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -12350,7 +11604,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12369,21 +11622,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12393,14 +11645,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12425,7 +11677,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12433,16 +11685,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12456,13 +11707,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -12485,14 +11732,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -12504,7 +11743,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12516,18 +11755,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -12536,7 +11767,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12545,19 +11780,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -12589,7 +11813,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12601,17 +11825,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -12641,7 +11863,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -12663,16 +11884,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12683,14 +11903,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12700,11 +11920,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12716,8 +11936,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12740,7 +11958,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12765,7 +11983,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12773,16 +11991,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12802,10 +12019,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -12827,14 +12040,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12845,7 +12050,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12857,18 +12062,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12876,7 +12073,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -12885,19 +12082,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12928,7 +12114,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12938,20 +12124,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12976,7 +12164,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12984,16 +12172,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13013,10 +12200,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -13038,14 +12221,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13056,7 +12231,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13068,18 +12243,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13087,7 +12254,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -13096,19 +12267,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13139,7 +12299,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13149,17 +12309,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -13178,21 +12336,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13202,11 +12359,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13216,14 +12369,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13250,7 +12395,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13259,17 +12404,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13283,7 +12424,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -13302,22 +12442,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13329,13 +12468,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -13343,7 +12478,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13365,16 +12499,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13385,14 +12518,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13402,11 +12535,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13418,16 +12551,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13449,15 +12577,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13468,14 +12595,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13485,11 +12612,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13501,8 +12628,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -13528,7 +12653,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13554,7 +12679,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13562,16 +12687,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13591,10 +12715,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -13616,14 +12736,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13636,7 +12748,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13648,18 +12760,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13672,7 +12776,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -13681,20 +12785,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13727,7 +12820,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13737,23 +12830,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13779,7 +12870,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13787,16 +12878,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13808,13 +12898,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -13837,14 +12923,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13857,7 +12935,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13869,18 +12947,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13893,7 +12963,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -13902,20 +12972,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13948,7 +13008,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13958,17 +13018,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13987,21 +13045,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14011,11 +13068,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14025,14 +13078,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14059,7 +13104,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14068,17 +13113,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -14088,7 +13129,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -14107,21 +13147,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14133,19 +13172,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14174,7 +13209,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14182,16 +13217,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14201,11 +13235,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -14232,14 +13262,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14250,7 +13272,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14265,18 +13287,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14287,7 +13301,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -14299,20 +13313,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14342,7 +13345,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14351,23 +13354,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -14393,7 +13392,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14401,16 +13400,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14420,11 +13418,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -14449,14 +13443,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14467,7 +13453,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14480,18 +13466,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14502,7 +13480,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -14512,20 +13490,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14555,7 +13522,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14564,17 +13531,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -14586,7 +13549,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14605,21 +13567,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14631,13 +13592,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -14645,7 +13602,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14669,16 +13625,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14689,14 +13644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -14706,11 +13661,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14722,17 +13677,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14756,15 +13706,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14775,14 +13724,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -14792,11 +13741,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14808,8 +13757,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14821,7 +13768,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -14845,17 +13791,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14866,14 +13811,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -14883,11 +13828,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14899,10 +13844,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14913,7 +13854,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -14937,17 +13877,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14958,14 +13897,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -14975,11 +13914,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14991,15 +13930,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -15030,7 +13965,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15039,16 +13974,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15058,11 +13992,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -15094,14 +14024,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15112,7 +14034,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -15128,7 +14050,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -15136,14 +14058,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15154,7 +14068,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -15168,24 +14082,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15216,7 +14119,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15225,23 +14128,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15270,7 +14169,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15278,16 +14177,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15297,11 +14195,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -15326,14 +14220,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15344,7 +14230,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15357,18 +14243,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15379,7 +14257,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -15389,20 +14267,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15432,7 +14299,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15441,23 +14308,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15486,7 +14349,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15494,16 +14357,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15513,11 +14375,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -15542,14 +14400,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15560,7 +14410,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15573,18 +14423,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15595,7 +14437,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -15605,20 +14447,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15648,7 +14479,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15657,23 +14488,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15702,7 +14529,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15710,16 +14537,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15729,11 +14555,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -15759,14 +14581,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15777,7 +14591,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15791,18 +14605,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15813,7 +14619,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -15824,20 +14630,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15867,7 +14662,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15876,23 +14671,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15921,7 +14712,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15929,16 +14720,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15948,11 +14738,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -15980,14 +14766,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15998,7 +14776,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16014,18 +14792,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16036,7 +14806,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -16049,20 +14819,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16092,7 +14851,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16101,23 +14860,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -16143,7 +14898,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16151,16 +14906,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16170,11 +14924,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -16199,14 +14949,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16217,7 +14959,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16230,18 +14972,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16252,7 +14986,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -16262,20 +14996,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16305,7 +15028,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16314,23 +15037,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16359,7 +15078,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16367,16 +15086,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16386,11 +15104,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -16416,14 +15130,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16434,7 +15140,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16448,18 +15154,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16470,7 +15168,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -16482,20 +15180,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16525,7 +15212,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16534,23 +15221,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16579,7 +15262,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16587,16 +15270,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16606,11 +15288,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -16636,14 +15314,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16654,7 +15324,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16668,18 +15338,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16690,7 +15352,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -16701,20 +15363,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16744,7 +15395,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16753,20 +15404,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -16779,7 +15426,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -16798,22 +15444,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16825,13 +15470,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -16856,7 +15497,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16867,7 +15507,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16880,11 +15520,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16895,7 +15534,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to true for a existing virtual machine scale set. @@ -16906,13 +15545,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2019_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2019_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16940,7 +15576,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16948,16 +15584,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16967,17 +15602,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -17003,7 +15634,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17011,16 +15642,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17030,11 +15660,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -17058,14 +15684,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17076,7 +15694,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17089,18 +15707,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17111,7 +15721,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -17122,19 +15732,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17164,7 +15764,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17173,20 +15773,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17211,7 +15807,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -17237,7 +15833,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -17246,16 +15842,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17275,10 +15870,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -17306,14 +15897,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17327,7 +15910,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17343,18 +15926,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17368,7 +15943,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -17381,20 +15956,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17428,7 +15992,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17438,24 +16002,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -17481,7 +16043,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -17490,16 +16052,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17519,10 +16080,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -17550,14 +16107,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17571,7 +16120,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17587,18 +16136,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17612,7 +16153,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -17625,20 +16166,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17672,7 +16203,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17682,17 +16213,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -17711,22 +16240,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17736,11 +16264,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -17755,14 +16279,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17790,7 +16306,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17799,17 +16315,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17831,7 +16343,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -17850,23 +16361,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17878,13 +16388,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -17896,7 +16402,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17920,17 +16425,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17941,14 +16445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -17958,11 +16462,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17974,12 +16478,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18016,21 +16516,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18040,11 +16539,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18054,14 +16549,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18088,7 +16575,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18097,17 +16584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -18126,21 +16609,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18150,11 +16632,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -18168,14 +16646,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18202,7 +16672,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18211,17 +16681,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -18240,21 +16706,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18264,11 +16729,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -18282,14 +16743,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18316,7 +16769,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18325,17 +16778,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -18347,7 +16796,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -18366,21 +16814,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18392,16 +16839,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18427,7 +16870,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -18453,7 +16896,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18463,16 +16906,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18492,10 +16934,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -18524,14 +16962,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18546,7 +16976,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18563,18 +16993,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18589,7 +17011,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the VMSS VM extension. @@ -18603,20 +17025,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18651,7 +17062,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18661,17 +17072,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -18679,7 +17088,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -18705,7 +17114,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18715,16 +17124,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18736,13 +17144,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -18773,14 +17177,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18795,7 +17191,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18812,18 +17208,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18838,7 +17226,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the VMSS VM extension. @@ -18852,20 +17240,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -18900,7 +17277,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18910,17 +17287,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -18939,23 +17314,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18965,11 +17339,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18985,14 +17355,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19021,7 +17383,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19030,17 +17392,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19065,7 +17423,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -19084,7 +17441,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19092,16 +17449,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19113,13 +17469,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19141,7 +17493,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -19160,23 +17511,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19188,13 +17538,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -19222,7 +17568,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -19251,7 +17599,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19260,16 +17608,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19279,11 +17626,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -19311,14 +17654,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19330,7 +17665,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -19345,18 +17680,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19368,7 +17695,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -19380,20 +17709,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -19424,7 +17743,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19433,17 +17752,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19462,22 +17777,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19487,11 +17801,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -19506,14 +17816,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19541,7 +17843,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19550,17 +17852,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19579,22 +17877,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19604,11 +17901,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -19624,14 +17917,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19659,7 +17944,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19668,24 +17953,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -19711,7 +17992,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19720,16 +18001,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19749,10 +18029,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -19779,14 +18055,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19800,7 +18068,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19816,18 +18084,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19841,7 +18101,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -19854,19 +18114,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19900,7 +18149,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19910,17 +18159,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19939,22 +18186,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19964,11 +18210,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -19982,14 +18224,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20017,7 +18251,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20026,17 +18260,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -20059,7 +18289,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -20078,23 +18307,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20106,13 +18334,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -20126,7 +18350,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -20145,22 +18368,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20172,13 +18394,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -20207,7 +18425,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -20231,7 +18448,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -20239,12 +18456,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20255,14 +18471,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -20272,11 +18488,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20288,10 +18504,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -20315,23 +18527,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20341,11 +18552,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -20371,14 +18578,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20407,7 +18606,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20416,17 +18615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20445,22 +18640,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20470,11 +18664,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -20488,14 +18678,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20523,7 +18705,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20532,17 +18714,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20561,22 +18739,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20586,11 +18763,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -20604,14 +18777,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20639,7 +18804,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20648,17 +18813,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20677,22 +18838,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20702,11 +18862,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -20721,14 +18877,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20756,7 +18904,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20765,17 +18913,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20794,22 +18938,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20819,11 +18962,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -20838,14 +18977,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20873,7 +19004,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20882,17 +19013,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -20907,7 +19034,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20926,22 +19052,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20951,18 +19076,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -20988,7 +19109,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20997,16 +19118,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21020,13 +19140,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -21052,14 +19168,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -21072,7 +19180,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21086,18 +19194,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -21110,7 +19210,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -21122,19 +19222,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.RunCommandResult] @@ -21167,7 +19256,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21179,17 +19268,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -21213,7 +19300,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -21238,23 +19325,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21268,13 +19354,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -21296,14 +19378,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21313,7 +19387,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21322,18 +19396,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21343,7 +19409,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21351,19 +19417,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21395,7 +19451,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21407,20 +19463,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -21445,23 +19499,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21475,13 +19528,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -21502,14 +19551,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21519,7 +19560,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -21527,18 +19568,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21548,7 +19581,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -21556,19 +19589,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2019_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21600,7 +19622,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21612,17 +19634,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -21651,7 +19671,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -21675,16 +19694,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21695,14 +19713,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -21712,11 +19730,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21728,10 +19746,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -21740,7 +19754,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -21759,21 +19772,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21785,13 +19797,9 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore class GalleriesOperations: @@ -21815,7 +19823,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -21840,7 +19848,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -21848,16 +19856,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21880,10 +19887,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -21908,14 +19911,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -21926,7 +19921,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21941,18 +19936,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -21960,7 +19947,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -21971,19 +19958,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_12_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_12_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22014,7 +19990,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22024,20 +20000,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -22062,7 +20040,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -22070,16 +20048,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22091,13 +20068,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22122,14 +20095,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22140,7 +20105,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22154,18 +20119,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22173,7 +20130,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -22184,19 +20145,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2019_12_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2019_12_01.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22227,7 +20177,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22237,17 +20187,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _models.Gallery: @@ -22257,7 +20205,6 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -22276,21 +20223,20 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22302,13 +20248,9 @@ def get(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> _mo deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -22327,21 +20269,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22351,11 +20292,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22365,14 +20302,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22399,7 +20328,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22408,17 +20337,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -22426,7 +20351,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22448,16 +20372,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22468,14 +20391,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -22485,11 +20408,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22501,15 +20424,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -22531,15 +20449,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22550,14 +20467,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -22567,11 +20484,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22583,8 +20500,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -22611,7 +20526,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -22637,7 +20552,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -22646,16 +20561,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22678,10 +20592,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -22710,14 +20620,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -22730,7 +20632,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22748,18 +20650,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -22772,7 +20666,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery Image Definition. @@ -22787,19 +20681,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -22832,7 +20715,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22842,24 +20725,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -22885,7 +20766,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -22894,16 +20775,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22915,13 +20795,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22950,14 +20826,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -22970,7 +20838,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22987,18 +20855,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -23011,7 +20871,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery Image Definition. @@ -23026,19 +20886,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] @@ -23071,7 +20920,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23081,17 +20930,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -23106,7 +20953,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -23125,22 +20971,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23152,13 +20997,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -23177,22 +21018,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23202,11 +21042,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23221,14 +21057,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery Image Definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23256,7 +21084,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23265,17 +21093,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -23288,7 +21112,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -23310,17 +21133,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23331,14 +21153,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -23348,11 +21170,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23364,10 +21186,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -23395,7 +21213,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -23421,7 +21239,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -23431,16 +21249,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23463,10 +21280,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -23500,14 +21313,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23522,7 +21327,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23544,18 +21349,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23570,7 +21367,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery Image Version. @@ -23589,19 +21386,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery Image Version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23636,7 +21423,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23646,17 +21433,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -23664,7 +21449,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -23690,7 +21475,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -23700,16 +21485,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23721,13 +21505,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -23762,14 +21542,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23784,7 +21556,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23806,18 +21578,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23832,7 +21596,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery Image Version. @@ -23851,20 +21615,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery Image Version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -23899,7 +21652,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23909,17 +21662,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -23948,7 +21699,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_12_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -23967,7 +21717,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -23975,16 +21725,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23996,13 +21745,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -24026,23 +21771,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24052,11 +21796,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24080,14 +21820,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery Image Version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24116,7 +21848,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24125,17 +21857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -24151,7 +21879,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryImageVersion] @@ -24174,18 +21901,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24196,14 +21922,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -24213,11 +21939,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24229,10 +21955,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -24259,7 +21981,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -24285,7 +22007,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -24294,16 +22016,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24326,10 +22047,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -24358,14 +22075,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24379,7 +22088,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24397,18 +22106,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24422,7 +22123,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -24437,19 +22138,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24483,7 +22174,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24493,24 +22184,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -24536,7 +22225,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -24545,16 +22234,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24566,13 +22254,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -24602,14 +22286,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24623,7 +22299,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24641,18 +22317,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24666,7 +22334,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -24681,20 +22349,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24728,7 +22385,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24738,17 +22395,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -24764,7 +22419,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -24783,22 +22437,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24810,13 +22463,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -24835,22 +22484,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24860,11 +22508,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24880,14 +22524,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24915,7 +22551,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24924,17 +22560,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -24947,7 +22579,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2019_12_01.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -24969,17 +22600,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24990,14 +22620,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -25007,11 +22637,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25023,10 +22653,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -25054,7 +22680,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -25080,7 +22706,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25090,16 +22716,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25122,10 +22747,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -25160,14 +22781,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25182,7 +22795,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25204,18 +22817,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25230,7 +22835,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -25249,21 +22854,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25298,7 +22892,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25308,17 +22902,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -25326,7 +22918,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -25352,7 +22944,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25362,16 +22954,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25383,13 +22974,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25425,14 +23012,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25447,7 +23026,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25469,18 +23048,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25495,7 +23066,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -25514,20 +23085,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25562,7 +23123,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25572,17 +23133,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -25611,7 +23170,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2019_12_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2019_12_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -25630,7 +23188,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25638,16 +23196,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25659,13 +23216,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25689,23 +23242,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2019-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25715,11 +23267,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25743,14 +23291,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25779,7 +23319,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25788,17 +23328,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -25814,7 +23350,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25838,18 +23373,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25860,14 +23394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -25877,11 +23411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25892,7 +23426,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_compute_management_client.py index 50bdf7d04da5..f67cfa755971 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -58,7 +60,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -75,7 +95,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-05-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -95,7 +115,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_configuration.py index c82e4940d2e0..05050d2d3b56 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-05-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_metadata.json index e6fe27f31672..71b18b4ae370 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_compute_management_client.py index 789b456a48e2..44e95732f292 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -58,7 +60,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -75,7 +95,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-05-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -95,7 +117,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_configuration.py index a643e1b452db..84c3a58290ab 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-05-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/operations/_operations.py index 289e0f3de62c..17bf47541bc7 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -88,7 +88,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -113,7 +113,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -121,16 +121,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -150,10 +149,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -177,14 +172,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -195,7 +182,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -209,18 +196,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -228,7 +207,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -239,19 +218,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_05_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_05_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -282,7 +250,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -292,20 +260,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -330,7 +296,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -338,16 +304,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -367,10 +332,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -394,14 +355,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -412,7 +365,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -426,18 +379,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -445,7 +390,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -456,19 +401,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_05_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_05_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -499,7 +433,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -509,17 +443,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -531,7 +463,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -550,21 +481,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -576,13 +506,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -601,21 +527,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -625,11 +550,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -641,14 +562,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -675,7 +588,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -684,17 +597,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -702,7 +611,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -724,16 +632,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -744,14 +651,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -761,11 +668,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -777,15 +684,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -807,15 +709,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -826,14 +727,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -843,11 +744,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -859,13 +760,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -891,7 +790,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -899,16 +798,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -922,13 +820,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -954,14 +848,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -973,7 +859,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -988,18 +874,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -1011,7 +889,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1023,19 +901,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -1067,7 +934,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1079,17 +946,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1108,21 +973,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1132,11 +996,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1150,14 +1010,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1184,7 +1036,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1195,17 +1047,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1229,7 +1077,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1254,7 +1102,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1262,16 +1110,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1291,10 +1138,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1318,14 +1161,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1337,7 +1172,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1351,18 +1186,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1371,7 +1198,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1382,19 +1209,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1426,7 +1242,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1436,20 +1252,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1474,7 +1292,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1482,16 +1300,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1511,10 +1328,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1539,14 +1352,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1558,7 +1363,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1573,18 +1378,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1593,7 +1390,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1604,19 +1405,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1648,7 +1438,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1658,17 +1448,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1680,7 +1468,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1699,21 +1486,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1725,13 +1511,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1750,21 +1532,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1774,11 +1555,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1790,14 +1567,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1824,7 +1593,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1833,17 +1602,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1851,7 +1616,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1874,16 +1638,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1894,14 +1657,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1911,11 +1674,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1927,15 +1690,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.Snapshot] @@ -1958,15 +1716,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1977,14 +1734,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1994,11 +1751,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2010,13 +1767,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2042,7 +1797,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2050,16 +1805,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2073,13 +1827,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2105,14 +1855,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -2124,7 +1866,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2139,18 +1881,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -2162,7 +1896,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2174,19 +1908,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] @@ -2218,7 +1941,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2230,17 +1953,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2259,21 +1980,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2283,11 +2003,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2301,14 +2017,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2335,7 +2043,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2346,17 +2054,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2383,7 +2087,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2409,7 +2113,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2417,16 +2121,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2446,10 +2149,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2474,14 +2173,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2494,7 +2185,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2509,18 +2200,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2533,7 +2216,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2545,19 +2228,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2590,7 +2263,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2600,23 +2273,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2642,7 +2313,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2650,16 +2321,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2679,10 +2349,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2707,14 +2373,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2727,7 +2385,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2742,18 +2400,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2766,7 +2416,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2778,19 +2428,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2823,7 +2464,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2833,17 +2474,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2857,7 +2496,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2876,21 +2514,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2902,13 +2539,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2927,21 +2560,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2951,11 +2583,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2969,14 +2597,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3003,7 +2623,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3012,17 +2632,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3032,7 +2648,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3055,16 +2670,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3075,14 +2689,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3092,11 +2706,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3108,15 +2722,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3139,15 +2748,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3158,14 +2766,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3175,11 +2783,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3191,8 +2799,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - class DiskAccessesOperations: """ @@ -3215,7 +2821,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3240,7 +2850,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3248,16 +2858,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3277,10 +2886,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3305,14 +2910,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3324,7 +2921,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3339,18 +2936,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3359,7 +2948,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3370,19 +2963,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3414,7 +2996,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3424,23 +3006,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3466,7 +3046,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3474,16 +3054,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3503,10 +3082,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3531,14 +3106,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3550,7 +3117,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3565,18 +3132,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3588,7 +3147,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3600,19 +3159,8 @@ async def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3644,7 +3192,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3654,17 +3202,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3676,7 +3222,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3695,21 +3240,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3721,13 +3265,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3746,21 +3286,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3770,11 +3309,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3788,14 +3323,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3822,7 +3349,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3831,17 +3358,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -3849,7 +3372,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3872,16 +3394,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3892,14 +3413,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3909,11 +3430,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3925,15 +3446,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -3956,15 +3472,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3975,14 +3490,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3992,11 +3507,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4008,8 +3523,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4022,7 +3535,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4041,21 +3553,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4067,10 +3578,6 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_compute_management_client_enums.py index 8873f7b2cd02..7d3369af677f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_compute_management_client_enums.py @@ -27,21 +27,18 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -76,10 +73,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -87,13 +84,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class HyperVGeneration(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_models_py3.py index 7018d1bbbc65..c8c7fc751b5d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/models/_models_py3.py @@ -131,11 +131,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2020_05_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -191,8 +190,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2020_05_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -230,7 +228,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -279,7 +277,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -518,7 +516,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -580,7 +578,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] @@ -636,7 +634,7 @@ class DiskEncryptionSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -723,7 +721,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -804,7 +802,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2020_05_01.models.Disk] @@ -1097,7 +1095,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1189,7 +1187,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2020_05_01.models.AccessLevel @@ -1222,7 +1220,7 @@ def __init__(self, *, access: Union[str, "_models.AccessLevel"], duration_in_sec class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1289,7 +1287,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_05_01.models.SourceVault @@ -1322,7 +1320,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_05_01.models.SourceVault @@ -1585,7 +1583,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1754,7 +1752,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2020_05_01.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/operations/_operations.py index a855dee699db..46fbf4693d0a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_05_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -697,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -827,7 +827,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -881,7 +881,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -933,7 +933,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -958,7 +958,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -966,16 +966,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -995,10 +994,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1022,14 +1017,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1040,7 +1027,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1054,18 +1041,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1073,7 +1052,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1084,19 +1063,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_05_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_05_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1127,7 +1095,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1137,20 +1105,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1175,7 +1141,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1183,16 +1149,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1212,10 +1177,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1239,14 +1200,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1257,7 +1210,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1271,18 +1224,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1290,7 +1235,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1301,19 +1246,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_05_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_05_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1344,7 +1278,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1354,17 +1288,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1376,7 +1308,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1395,21 +1326,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1421,13 +1351,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1446,21 +1372,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1470,11 +1395,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1486,14 +1407,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1520,7 +1433,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1529,17 +1442,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1547,7 +1456,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1569,16 +1477,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1589,14 +1496,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1606,11 +1513,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1622,15 +1529,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1652,15 +1554,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1671,14 +1572,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1688,11 +1589,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1704,13 +1605,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -1736,7 +1635,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1744,16 +1643,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1767,13 +1665,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -1799,14 +1693,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1817,7 +1703,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1832,18 +1718,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1854,7 +1732,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1866,19 +1744,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,7 +1776,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1921,17 +1788,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1950,21 +1815,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1974,11 +1838,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1990,14 +1850,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2024,7 +1876,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2035,17 +1887,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -2069,7 +1917,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2094,7 +1942,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2102,16 +1950,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2131,10 +1978,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -2158,14 +2001,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2176,7 +2011,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2190,18 +2025,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2209,7 +2036,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -2220,19 +2047,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2263,7 +2079,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2273,20 +2089,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2311,7 +2129,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2319,16 +2137,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2348,10 +2165,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -2376,14 +2189,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2394,7 +2199,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2409,18 +2214,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2428,7 +2225,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2439,19 +2240,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_05_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2482,7 +2272,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2492,17 +2282,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2514,7 +2302,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2533,21 +2320,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2559,13 +2345,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2584,21 +2366,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2608,11 +2389,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2624,14 +2401,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2658,7 +2427,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2667,17 +2436,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -2685,7 +2450,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2707,16 +2471,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2727,14 +2490,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2744,11 +2507,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2760,15 +2523,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2790,15 +2548,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2809,14 +2566,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2826,11 +2583,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2842,13 +2599,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2874,7 +2629,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2882,16 +2637,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2905,13 +2659,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2937,14 +2687,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2955,7 +2697,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2970,18 +2712,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2992,7 +2726,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -3004,19 +2738,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_05_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3047,7 +2770,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3059,17 +2782,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -3088,21 +2809,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3112,11 +2832,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3128,14 +2844,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3162,7 +2870,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -3173,17 +2881,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -3210,7 +2914,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3236,7 +2940,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3244,16 +2948,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3273,10 +2976,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3301,14 +3000,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3320,7 +3011,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3335,18 +3026,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3358,7 +3041,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3370,19 +3053,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3414,7 +3087,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3424,23 +3097,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3466,7 +3137,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3474,16 +3145,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3503,10 +3173,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -3531,14 +3197,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3550,7 +3208,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3565,18 +3223,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3588,7 +3238,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3600,19 +3250,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] @@ -3644,7 +3285,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3654,17 +3295,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -3676,7 +3315,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3695,21 +3333,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3721,13 +3358,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3746,21 +3379,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3770,11 +3402,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3786,14 +3414,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3820,7 +3440,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3829,17 +3449,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -3847,7 +3463,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3869,16 +3484,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3889,14 +3503,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3906,11 +3520,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3922,15 +3536,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3952,15 +3561,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3971,14 +3579,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3988,11 +3596,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4004,8 +3612,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - class DiskAccessesOperations: """ @@ -4028,7 +3634,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -4053,7 +3663,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4061,16 +3671,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4090,10 +3699,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -4118,14 +3723,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4136,7 +3733,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4151,18 +3748,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4170,7 +3759,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -4181,19 +3774,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4224,7 +3806,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4234,23 +3816,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -4276,7 +3856,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4284,16 +3864,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4313,10 +3892,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -4341,14 +3916,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4359,7 +3926,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4374,18 +3941,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4396,7 +3955,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -4408,19 +3967,8 @@ def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_05_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4451,7 +3999,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4461,17 +4009,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -4483,7 +4029,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -4502,21 +4047,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4528,13 +4072,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4553,21 +4093,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4577,11 +4116,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4593,14 +4128,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4627,7 +4154,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4636,17 +4163,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -4654,7 +4177,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4676,16 +4198,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4696,14 +4217,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4713,11 +4234,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4729,15 +4250,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_05_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4759,15 +4275,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4778,14 +4293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4795,11 +4310,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4811,8 +4326,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4825,7 +4338,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_05_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4844,21 +4356,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-05-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4870,10 +4381,6 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_compute_management_client.py index ad86c10eb988..e4c757cbb239 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -131,7 +133,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -195,7 +215,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-06-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -215,7 +235,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_configuration.py index 5b88bb3e7406..a273a41d70a6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-06-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_metadata.json index eb8d24efa732..11dfa9d9e9be 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_compute_management_client.py index 4c770038532a..3564626bd6f1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -133,7 +135,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -197,7 +217,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-06-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -217,7 +239,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_configuration.py index 1fe205e0e152..0b961a9bdc74 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-06-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/operations/_operations.py index fd5447df4763..1c78afac2315 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -178,10 +177,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -210,7 +205,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -234,14 +228,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -252,14 +245,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -269,11 +262,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -285,8 +278,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -329,7 +320,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -340,7 +330,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -352,11 +342,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -367,7 +356,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -377,12 +366,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -410,7 +395,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -418,16 +403,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -439,13 +423,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -468,7 +448,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -479,7 +458,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -491,11 +470,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -506,7 +484,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -516,12 +494,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -549,7 +523,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -557,16 +531,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -578,13 +551,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -596,7 +565,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -615,21 +583,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -639,11 +606,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -653,7 +616,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -672,21 +634,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -698,13 +659,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -715,7 +672,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet] @@ -738,16 +694,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -758,14 +713,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -775,11 +730,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -791,17 +746,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet] @@ -824,16 +774,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -844,14 +793,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -861,11 +810,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -877,10 +826,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -892,7 +837,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] @@ -915,17 +859,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -936,14 +879,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -953,11 +896,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -969,10 +912,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1016,7 +955,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1027,7 +965,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1040,11 +978,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1055,7 +992,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1065,12 +1002,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1098,7 +1031,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1106,16 +1039,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1135,10 +1067,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1161,7 +1089,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1172,7 +1099,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1185,11 +1112,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1200,7 +1126,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1210,12 +1136,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1243,7 +1166,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1251,16 +1174,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1272,13 +1194,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1290,7 +1208,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1309,21 +1226,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1333,11 +1249,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1357,7 +1269,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1376,22 +1287,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1403,19 +1313,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1439,15 +1344,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1458,14 +1362,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1475,11 +1379,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1491,10 +1395,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1503,7 +1403,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1527,16 +1426,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1547,14 +1445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1564,11 +1462,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1580,10 +1478,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1628,7 +1522,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1639,7 +1532,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1653,11 +1546,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1668,7 +1560,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1680,12 +1572,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1713,7 +1601,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1721,16 +1609,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1750,10 +1637,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1775,7 +1658,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1786,7 +1668,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1798,11 +1680,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1813,7 +1694,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1823,12 +1704,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1856,7 +1733,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1864,16 +1741,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1885,13 +1761,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1903,7 +1775,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1922,21 +1793,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1946,11 +1816,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1971,7 +1837,6 @@ async def get( of instance view of the dedicated hosts under the dedicated host group. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1990,22 +1855,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2017,13 +1881,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2034,7 +1894,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup] @@ -2057,16 +1916,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2077,14 +1935,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2094,11 +1952,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2110,16 +1968,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup] @@ -2142,15 +1995,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2161,14 +2013,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2178,11 +2030,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2194,8 +2046,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2222,7 +2072,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2248,7 +2098,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2257,16 +2107,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2286,10 +2135,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2314,14 +2159,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2335,7 +2172,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2349,18 +2186,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2374,7 +2203,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2386,19 +2215,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2432,7 +2250,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2442,24 +2260,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2485,7 +2301,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2494,16 +2310,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2515,13 +2330,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2547,14 +2358,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2568,7 +2371,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2582,18 +2385,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2607,7 +2402,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2619,19 +2414,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2665,7 +2449,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2675,17 +2459,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2704,22 +2486,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2729,11 +2510,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2747,14 +2524,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2782,7 +2551,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2791,17 +2560,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2824,7 +2589,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2843,23 +2607,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2871,13 +2634,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2890,7 +2649,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -2913,17 +2671,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2934,14 +2691,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -2951,11 +2708,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2967,10 +2724,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -2997,7 +2750,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3021,15 +2773,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3040,14 +2791,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3057,11 +2808,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3073,8 +2824,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -3084,7 +2833,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3108,16 +2856,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3128,14 +2875,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3145,11 +2892,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3161,10 +2908,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -3186,7 +2929,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3197,7 +2939,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3209,11 +2951,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3224,7 +2965,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -3234,12 +2975,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3267,7 +3004,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3275,16 +3012,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3304,10 +3040,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -3329,7 +3061,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3340,7 +3071,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3352,11 +3083,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3367,7 +3097,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -3377,12 +3107,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3410,7 +3137,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3418,16 +3145,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3439,13 +3165,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -3457,7 +3179,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3476,21 +3197,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3500,11 +3220,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -3516,7 +3232,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3535,21 +3250,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3561,13 +3275,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -3581,7 +3291,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3600,21 +3309,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3626,13 +3334,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -3669,7 +3373,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3688,23 +3391,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3716,13 +3418,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3734,7 +3432,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3753,21 +3450,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3779,13 +3475,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3813,7 +3505,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3832,7 +3523,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3841,16 +3532,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3862,13 +3552,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3896,7 +3582,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3922,7 +3608,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3931,16 +3617,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3960,10 +3645,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -3990,14 +3671,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4011,7 +3684,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4027,18 +3700,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4052,7 +3717,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4065,20 +3730,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4112,7 +3766,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4122,24 +3776,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4165,7 +3817,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4174,16 +3826,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4195,13 +3846,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -4230,14 +3877,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4251,7 +3890,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4267,18 +3906,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4292,7 +3923,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4305,20 +3936,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4352,7 +3972,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4362,17 +3982,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4391,22 +4009,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4416,11 +4033,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4435,14 +4048,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4470,7 +4075,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4479,17 +4084,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4511,7 +4112,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4530,23 +4130,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4558,13 +4157,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4578,7 +4173,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4597,22 +4191,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4624,13 +4217,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4669,7 +4258,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4688,7 +4276,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4696,16 +4284,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4717,13 +4304,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4755,7 +4338,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4774,7 +4356,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4784,16 +4366,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4805,13 +4386,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4823,7 +4400,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4842,21 +4418,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4868,13 +4443,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4882,7 +4453,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4901,20 +4471,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4926,13 +4495,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -4946,7 +4511,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4965,22 +4529,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4992,13 +4555,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5028,7 +4587,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5050,16 +4608,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5070,14 +4627,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5087,11 +4644,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5103,8 +4660,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -5133,7 +4688,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -5156,16 +4710,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5176,14 +4729,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5193,11 +4746,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5209,15 +4762,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -5243,7 +4792,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5251,16 +4800,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5274,13 +4822,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -5304,14 +4848,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5324,7 +4860,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5337,18 +4873,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5361,7 +4889,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5372,19 +4900,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5417,7 +4935,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5429,20 +4947,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5467,7 +4987,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5475,16 +4995,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5504,10 +5023,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -5530,14 +5045,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5550,7 +5057,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5563,18 +5070,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5584,7 +5083,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -5594,19 +5097,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5639,7 +5131,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5649,20 +5141,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5687,7 +5181,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5695,16 +5189,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5724,10 +5217,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_update( self, @@ -5749,14 +5238,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5769,7 +5250,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5781,18 +5262,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5802,7 +5275,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -5811,19 +5288,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -5856,7 +5322,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5866,17 +5332,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -5895,22 +5359,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5920,11 +5383,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -5939,14 +5398,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5974,7 +5425,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5983,17 +5434,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6008,7 +5455,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6027,22 +5473,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6054,13 +5499,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -6072,7 +5513,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6091,21 +5531,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6117,13 +5556,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6142,21 +5577,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6166,11 +5600,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -6183,14 +5613,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6217,7 +5639,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6226,17 +5648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6255,21 +5673,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6279,11 +5696,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6294,14 +5707,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6328,7 +5733,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6337,17 +5742,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -6364,7 +5765,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6383,21 +5783,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6407,11 +5806,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -6420,7 +5815,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -6443,16 +5837,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6463,14 +5856,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6480,11 +5873,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6496,10 +5889,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -6508,7 +5897,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -6531,16 +5919,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6551,14 +5938,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6568,11 +5955,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6584,8 +5971,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6597,7 +5982,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] @@ -6620,17 +6004,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6641,14 +6024,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6658,11 +6041,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6674,10 +6057,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -6695,22 +6074,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6720,11 +6098,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -6741,14 +6115,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6776,7 +6142,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6785,17 +6151,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6814,21 +6176,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6838,11 +6199,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6852,14 +6209,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6886,7 +6235,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6895,17 +6244,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6924,21 +6269,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6948,11 +6292,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6962,14 +6302,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6996,7 +6328,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7005,17 +6337,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7034,21 +6362,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7058,11 +6385,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7072,14 +6395,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7106,7 +6421,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7115,17 +6430,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7144,21 +6455,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7168,11 +6478,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7182,14 +6488,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7216,7 +6514,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7225,23 +6523,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7270,7 +6564,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7278,16 +6572,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7297,11 +6590,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -7325,14 +6614,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7343,7 +6624,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7356,18 +6637,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7378,7 +6651,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -7388,19 +6661,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7430,7 +6693,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7439,17 +6702,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7471,7 +6730,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7490,22 +6748,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7517,13 +6774,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7542,21 +6795,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7566,11 +6818,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -7582,14 +6830,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7616,7 +6856,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7625,17 +6865,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -7647,7 +6883,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7666,21 +6901,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7690,11 +6924,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7713,21 +6943,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7741,13 +6970,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -7759,14 +6984,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -7796,7 +7013,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7808,20 +7025,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -7846,7 +7065,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7854,16 +7073,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7877,13 +7095,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -7906,14 +7120,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7926,7 +7132,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7938,18 +7144,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -7959,7 +7157,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -7968,19 +7170,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8013,7 +7204,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8025,17 +7216,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -8065,7 +7254,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] @@ -8088,16 +7276,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8108,14 +7295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -8125,11 +7312,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8141,8 +7328,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -8165,7 +7350,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8190,7 +7375,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8198,16 +7383,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8227,10 +7411,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -8252,14 +7432,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8270,7 +7442,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8282,18 +7454,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8301,7 +7465,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -8310,19 +7474,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8353,7 +7506,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8363,20 +7516,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -8401,7 +7556,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -8409,16 +7564,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8438,10 +7592,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -8463,14 +7613,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8481,7 +7623,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8493,18 +7635,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8512,7 +7646,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -8521,19 +7659,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8564,7 +7691,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8574,17 +7701,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -8603,21 +7728,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8627,11 +7751,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -8641,14 +7761,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8675,7 +7787,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8684,17 +7796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8708,7 +7816,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -8727,22 +7834,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8754,13 +7860,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -8768,7 +7870,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8790,16 +7891,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8810,14 +7910,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8827,11 +7927,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8843,16 +7943,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -8874,15 +7969,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8893,14 +7987,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -8910,11 +8004,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8926,8 +8020,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -8953,7 +8045,7 @@ async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8979,7 +8071,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8987,16 +8079,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9016,10 +8107,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -9041,14 +8128,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9061,7 +8140,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9073,18 +8152,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9097,7 +8168,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -9106,20 +8177,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9152,7 +8212,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9162,23 +8222,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9204,7 +8262,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9212,16 +8270,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9233,13 +8290,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9262,14 +8315,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9282,7 +8327,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9294,18 +8339,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9318,7 +8355,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9327,20 +8364,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9373,7 +8400,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9383,17 +8410,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -9412,21 +8437,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9436,11 +8460,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9452,14 +8472,6 @@ async def begin_delete( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9486,7 +8498,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9495,17 +8507,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9517,7 +8525,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9536,21 +8543,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9562,19 +8568,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9603,7 +8605,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9611,16 +8613,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9630,11 +8631,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -9661,14 +8658,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9679,7 +8668,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9694,18 +8683,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9716,7 +8697,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9728,20 +8709,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9771,7 +8741,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9780,23 +8750,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9822,7 +8788,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9830,16 +8796,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9849,11 +8814,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -9878,14 +8839,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9896,7 +8849,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9909,18 +8862,10 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9931,7 +8876,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -9941,20 +8886,9 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9984,7 +8918,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9993,17 +8927,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -10015,7 +8945,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10034,21 +8963,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10060,13 +8988,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -10074,7 +8998,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10098,16 +9021,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10118,14 +9040,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10135,11 +9057,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10151,17 +9073,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10185,15 +9102,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10204,14 +9120,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10221,11 +9137,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10237,8 +9153,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10250,7 +9164,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10274,17 +9187,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10295,14 +9207,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10312,11 +9224,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10328,10 +9240,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10342,7 +9250,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10366,17 +9273,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10387,14 +9293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10404,11 +9310,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10420,15 +9326,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10459,7 +9361,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10468,16 +9370,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10487,11 +9388,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -10523,14 +9420,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10541,7 +9430,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -10557,7 +9446,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -10565,14 +9454,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10583,7 +9464,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10597,24 +9478,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10645,7 +9515,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10654,23 +9524,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10699,7 +9565,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10707,16 +9573,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10726,11 +9591,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -10755,14 +9616,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10773,7 +9626,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10786,18 +9639,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10808,7 +9653,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -10818,20 +9663,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10861,7 +9695,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10870,23 +9704,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10915,7 +9745,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10923,16 +9753,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10942,11 +9771,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10971,14 +9796,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10989,7 +9806,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11002,18 +9819,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11024,7 +9833,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -11034,20 +9843,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11077,7 +9875,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11086,23 +9884,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11131,7 +9925,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11139,16 +9933,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11158,11 +9951,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -11188,14 +9977,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11206,7 +9987,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11220,18 +10001,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11242,7 +10015,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11253,20 +10026,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11296,7 +10058,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11305,23 +10067,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11350,7 +10108,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11358,16 +10116,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11377,11 +10134,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -11409,14 +10162,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11427,7 +10172,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11443,18 +10188,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11465,7 +10202,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11478,20 +10215,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11521,7 +10247,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11530,23 +10256,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11572,7 +10294,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11580,16 +10302,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11599,11 +10320,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -11628,14 +10345,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11646,7 +10355,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11659,18 +10368,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11681,7 +10382,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11691,20 +10392,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11734,7 +10424,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11743,23 +10433,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11788,7 +10474,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11796,16 +10482,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11815,11 +10500,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11845,14 +10526,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11863,7 +10536,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11877,18 +10550,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11899,7 +10564,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11911,20 +10576,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11954,7 +10608,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11963,23 +10617,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12008,7 +10658,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12016,16 +10666,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12035,11 +10684,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -12065,14 +10710,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12083,7 +10720,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12097,18 +10734,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12119,7 +10748,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12130,20 +10759,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12173,7 +10791,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12182,20 +10800,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12208,7 +10822,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12227,22 +10840,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12254,13 +10866,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -12285,7 +10893,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12296,7 +10903,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12309,11 +10916,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12324,7 +10930,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -12335,13 +10941,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2020_06_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2020_06_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12369,7 +10972,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12377,16 +10980,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12396,17 +10998,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12432,7 +11030,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12440,16 +11038,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12459,11 +11056,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -12487,14 +11080,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12505,7 +11090,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12518,18 +11103,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12540,7 +11117,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -12551,19 +11128,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12593,7 +11160,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12602,20 +11169,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12640,7 +11203,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12666,7 +11229,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12675,16 +11238,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12704,10 +11266,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -12735,14 +11293,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12756,7 +11306,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12772,18 +11322,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12797,7 +11339,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -12810,20 +11352,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -12857,7 +11388,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12867,24 +11398,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -12910,7 +11439,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -12919,16 +11448,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12948,10 +11476,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -12979,14 +11503,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13000,7 +11516,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13016,18 +11532,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13041,7 +11549,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -13054,20 +11562,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13101,7 +11599,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13111,17 +11609,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -13140,22 +11636,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13165,11 +11660,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13184,14 +11675,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13219,7 +11702,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13228,17 +11711,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13260,7 +11739,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -13279,23 +11757,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13307,13 +11784,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13325,7 +11798,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -13349,17 +11821,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13370,14 +11841,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -13387,11 +11858,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13403,12 +11874,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13445,21 +11912,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13469,11 +11935,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -13485,14 +11947,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13519,7 +11973,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13528,17 +11982,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13557,21 +12007,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13581,11 +12030,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -13599,14 +12044,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13633,7 +12070,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13642,17 +12079,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13671,21 +12104,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13695,11 +12127,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -13713,14 +12141,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13747,7 +12167,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13756,17 +12176,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -13778,7 +12194,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -13797,21 +12212,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13823,16 +12237,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -13858,7 +12268,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -13884,7 +12294,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13894,16 +12304,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13923,10 +12332,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -13956,14 +12361,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -13978,7 +12375,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13995,18 +12392,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -14021,7 +12410,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -14035,20 +12424,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -14083,7 +12461,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14093,17 +12471,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -14111,7 +12487,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -14137,7 +12513,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14147,16 +12523,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14168,13 +12543,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -14205,14 +12576,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -14227,7 +12590,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14244,18 +12607,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -14270,7 +12625,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -14284,20 +12639,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -14332,7 +12677,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14342,17 +12687,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -14371,23 +12714,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14397,11 +12739,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14417,14 +12755,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14453,7 +12783,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14462,17 +12792,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14497,7 +12823,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14516,7 +12841,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14524,16 +12849,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14545,13 +12869,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -14573,7 +12893,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14592,23 +12911,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14620,13 +12938,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -14654,7 +12968,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -14683,7 +12999,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14692,16 +13008,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14711,11 +13026,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -14743,14 +13054,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14762,7 +13065,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14777,18 +13080,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14800,7 +13095,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -14812,20 +13109,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14856,7 +13143,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14865,17 +13152,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14894,22 +13177,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14919,11 +13201,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -14938,14 +13216,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14973,7 +13243,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14982,17 +13252,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15011,22 +13277,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15036,11 +13301,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -15056,14 +13317,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15091,7 +13344,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15100,24 +13353,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -15143,7 +13392,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15152,16 +13401,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15181,10 +13429,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -15211,14 +13455,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15232,7 +13468,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15248,18 +13484,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15273,7 +13501,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -15286,19 +13514,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15332,7 +13549,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15342,17 +13559,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15371,22 +13586,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15396,11 +13610,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15414,14 +13624,6 @@ async def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15449,7 +13651,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15458,17 +13660,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15491,7 +13689,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -15510,23 +13707,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15538,13 +13734,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -15558,7 +13750,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15577,22 +13768,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15604,13 +13794,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15639,7 +13825,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -15663,7 +13848,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -15671,12 +13856,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15687,14 +13871,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -15704,11 +13888,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15720,10 +13904,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -15747,23 +13927,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15773,11 +13952,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -15803,14 +13978,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15839,7 +14006,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15848,17 +14015,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15877,22 +14040,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15902,11 +14064,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -15920,14 +14078,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15955,7 +14105,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15964,17 +14114,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -15993,22 +14139,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16018,11 +14163,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -16036,14 +14177,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16071,7 +14204,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16080,17 +14213,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -16109,22 +14238,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16134,11 +14262,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -16153,14 +14277,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16188,7 +14304,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16197,17 +14313,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -16233,7 +14345,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -16252,23 +14363,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16280,13 +14390,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -16305,22 +14411,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16330,11 +14435,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -16348,14 +14449,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16383,7 +14476,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16392,17 +14485,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -16416,7 +14505,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16435,22 +14523,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16460,18 +14547,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -16497,7 +14580,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -16506,16 +14589,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16529,13 +14611,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -16561,14 +14639,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16582,7 +14652,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16596,18 +14666,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16621,7 +14683,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -16633,19 +14695,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -16679,7 +14730,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16691,17 +14742,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -16725,7 +14774,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16750,23 +14799,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16780,13 +14828,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -16808,14 +14852,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16825,7 +14861,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16834,18 +14870,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16855,7 +14883,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -16863,19 +14891,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -16907,7 +14925,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16920,20 +14938,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -16958,23 +14974,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16988,13 +15003,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -17015,14 +15026,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -17032,7 +15035,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -17040,18 +15043,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -17061,7 +15056,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -17069,19 +15064,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -17113,7 +15097,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17126,17 +15110,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -17165,7 +15147,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -17189,16 +15170,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17209,14 +15189,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -17226,11 +15206,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17242,10 +15222,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -17254,7 +15230,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -17273,21 +15248,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17299,20 +15273,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -17338,7 +15308,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -17347,16 +15317,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17376,10 +15345,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -17406,14 +15371,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17427,7 +15384,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17443,18 +15400,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17468,7 +15417,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -17481,19 +15430,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17527,7 +15465,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17537,24 +15475,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -17580,7 +15516,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -17589,16 +15525,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17610,13 +15545,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -17644,14 +15575,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17665,7 +15588,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17681,18 +15604,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17706,7 +15621,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -17719,19 +15634,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -17765,7 +15670,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17775,17 +15680,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -17804,22 +15707,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17829,11 +15731,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17848,14 +15746,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17883,7 +15773,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17892,17 +15782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -17924,7 +15810,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -17943,23 +15828,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17971,13 +15855,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -17991,7 +15871,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18015,18 +15894,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18037,14 +15915,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -18054,11 +15932,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18070,12 +15948,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18101,7 +15975,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -18127,7 +16001,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18137,16 +16011,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18166,10 +16039,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -18198,14 +16067,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18220,7 +16081,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18237,18 +16098,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18263,7 +16116,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -18277,19 +16130,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18324,7 +16166,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18334,17 +16176,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -18352,7 +16192,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -18378,7 +16218,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18388,16 +16228,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18409,13 +16248,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -18445,14 +16280,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18467,7 +16294,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18484,18 +16311,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18510,7 +16329,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -18524,19 +16343,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18571,7 +16380,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18581,17 +16390,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -18610,23 +16417,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18636,11 +16442,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18656,14 +16458,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18692,7 +16486,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18701,17 +16495,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18736,7 +16526,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -18755,7 +16544,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18763,16 +16552,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18784,13 +16572,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18812,7 +16596,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -18836,19 +16619,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18859,14 +16641,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -18876,11 +16658,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18891,7 +16673,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/models/_models_py3.py index ccd4ce09720f..399349bca8f5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -230,7 +226,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -255,7 +251,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -270,7 +266,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -336,7 +332,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -394,7 +390,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet] @@ -877,7 +873,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1048,7 +1044,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1225,7 +1221,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1341,7 +1337,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup] @@ -1545,7 +1541,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -1793,17 +1789,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2016,7 +2001,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2181,7 +2166,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2020_06_01.models.SubResource @@ -2287,7 +2272,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.Image] @@ -2321,7 +2306,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2020_06_01.models.SubResource @@ -2518,12 +2503,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_06_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -2548,12 +2533,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_06_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -2708,7 +2693,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -2741,7 +2726,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -2924,7 +2909,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.Usage] @@ -2958,7 +2943,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -3270,7 +3255,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2020_06_01.models.OrchestrationServiceNames @@ -3344,7 +3329,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -3487,7 +3472,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -3521,7 +3506,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. :code:`
    `:code:`
    ` @@ -3533,10 +3518,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3549,10 +3534,10 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -3563,7 +3548,7 @@ class OSProfile(_serialization.Model): `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -3571,10 +3556,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2020_06_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2020_06_01.models.VaultSecretGroup] @@ -3620,7 +3605,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. @@ -3633,10 +3618,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -3649,10 +3634,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -3663,7 +3648,7 @@ def __init__( `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -3671,10 +3656,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2020_06_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -3795,7 +3780,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3889,7 +3874,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup] @@ -3927,22 +3912,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -4013,7 +3987,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4328,7 +4302,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4391,7 +4365,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -4454,7 +4428,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -4528,7 +4502,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -4575,7 +4549,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -4608,7 +4582,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.RunCommandDocumentBase] @@ -4644,7 +4618,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -4886,7 +4860,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -4904,7 +4878,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -4915,7 +4889,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -4925,7 +4899,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -4954,7 +4928,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -4969,7 +4943,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5025,7 +4999,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource] @@ -5102,12 +5076,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_06_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.DataDisk] """ @@ -5134,12 +5108,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_06_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.DataDisk] """ super().__init__(**kwargs) @@ -5241,7 +5215,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5262,67 +5236,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UpgradeOperationHistoricalStatusInfo(_serialization.Model): """Virtual Machine Scale Set OS Upgrade History operation response. @@ -5358,7 +5271,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -5501,7 +5414,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -5711,7 +5624,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5757,10 +5670,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -5932,10 +5845,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -6130,7 +6043,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -6220,7 +6133,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6347,7 +6260,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -6391,7 +6304,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6681,7 +6594,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2020_06_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_06_01.models.UserAssignedIdentitiesValue] """ @@ -6713,7 +6626,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2020_06_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_06_01.models.UserAssignedIdentitiesValue] """ @@ -6727,7 +6640,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6783,7 +6696,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7038,7 +6951,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -7141,7 +7054,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7388,7 +7301,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand] @@ -7537,7 +7450,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7753,7 +7666,7 @@ def __init__( class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -7987,10 +7900,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension] @@ -8201,7 +8114,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2020_06_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -8240,7 +8153,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -8251,7 +8164,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -8322,7 +8235,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8349,7 +8262,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8504,10 +8417,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -8549,7 +8462,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet] @@ -8585,7 +8498,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetSku] @@ -8621,7 +8534,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet] @@ -8654,7 +8567,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -8696,10 +8609,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8793,7 +8706,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -8818,7 +8731,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2020_06_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -8843,7 +8756,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2020_06_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -8857,7 +8770,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -8997,10 +8910,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -9013,16 +8926,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -9030,10 +8943,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2020_06_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -9076,10 +8989,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -9092,16 +9005,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -9109,10 +9022,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2020_06_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -9128,10 +9041,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -9203,10 +9116,12 @@ def __init__( self.public_ip_address_version = public_ip_address_version -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -9233,7 +9148,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -9241,18 +9156,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -9371,12 +9274,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetDataDisk] """ @@ -9404,12 +9307,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetDataDisk] """ @@ -9561,7 +9464,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -9679,7 +9582,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -9771,12 +9674,12 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2020_06_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -9803,7 +9706,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2020_06_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -9933,7 +9836,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -9977,7 +9880,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -10116,7 +10019,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10176,10 +10079,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2020_06_01.models.SubResource @@ -10309,10 +10212,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2020_06_01.models.SubResource @@ -10474,7 +10377,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -10497,7 +10400,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -10655,10 +10558,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -10802,7 +10705,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM] @@ -10835,7 +10738,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -11273,10 +11176,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -11435,10 +11338,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -11521,7 +11424,7 @@ def __init__( # pylint: disable=too-many-locals self.extensions_time_budget = extensions_time_budget -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/operations/_operations.py index 551dfff1216a..d1a387772748 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -63,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -187,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -243,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -274,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -310,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -346,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -373,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -415,7 +410,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -441,7 +438,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -471,7 +468,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -505,7 +502,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -539,7 +536,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -602,7 +599,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -632,7 +629,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -656,7 +655,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -792,7 +791,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +822,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -847,7 +848,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1001,7 +1002,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1032,7 +1033,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1065,7 +1066,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1096,7 +1097,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -1142,7 +1143,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1177,7 +1178,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1212,7 +1213,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1238,7 +1239,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1278,7 +1279,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1345,7 +1346,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1393,7 +1394,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1424,7 +1425,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1453,7 +1454,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1512,7 +1513,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1574,7 +1577,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1712,7 +1715,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1743,7 +1746,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1768,7 +1771,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1793,7 +1796,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1874,7 +1877,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2069,7 +2072,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2109,7 +2112,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2134,7 +2137,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2159,7 +2162,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2190,7 +2193,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2377,7 +2380,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2431,7 +2434,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2465,7 +2468,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2499,7 +2502,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2524,7 +2527,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2555,7 +2558,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2586,7 +2589,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2617,7 +2620,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2648,7 +2651,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2678,7 +2681,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2704,7 +2709,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2735,7 +2740,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2766,7 +2771,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2804,7 +2809,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2835,7 +2840,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2866,7 +2871,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2897,7 +2902,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2928,7 +2933,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2959,7 +2964,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2990,7 +2995,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3021,7 +3026,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3058,7 +3063,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3089,7 +3094,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3120,7 +3125,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3155,7 +3160,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3190,7 +3195,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3216,7 +3221,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -3256,7 +3261,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3287,7 +3292,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3312,7 +3317,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3337,7 +3342,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3362,7 +3367,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3393,7 +3398,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3434,7 +3439,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3475,7 +3480,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3513,7 +3518,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3555,7 +3560,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3595,7 +3600,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3627,7 +3632,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3653,7 +3658,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3679,7 +3684,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3714,7 +3719,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3740,7 +3745,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3780,7 +3785,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3812,7 +3817,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -3858,7 +3863,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -3892,7 +3897,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3918,7 +3923,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3944,7 +3949,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3970,7 +3975,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4012,7 +4017,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4038,7 +4043,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4064,7 +4069,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4099,7 +4104,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4132,7 +4137,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4165,7 +4170,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4192,7 +4199,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4223,7 +4230,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4258,7 +4265,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4293,7 +4300,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4325,7 +4332,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -4365,7 +4372,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4398,7 +4405,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4439,7 +4446,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4480,7 +4487,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4518,7 +4525,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4560,7 +4567,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4624,7 +4631,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -4648,14 +4654,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4671,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -4683,11 +4688,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,8 +4704,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -4743,7 +4746,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4754,7 +4756,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4766,11 +4768,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4781,7 +4782,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -4791,12 +4792,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4824,7 +4821,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4832,16 +4829,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4853,13 +4849,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -4882,7 +4874,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4893,7 +4884,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4905,11 +4896,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4920,7 +4910,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -4930,12 +4920,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4963,7 +4949,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4971,16 +4957,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4992,13 +4977,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5010,7 +4991,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5029,21 +5009,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5053,11 +5032,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -5067,7 +5042,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5086,21 +5060,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5112,13 +5085,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -5129,7 +5098,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5151,16 +5119,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5171,14 +5138,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5188,11 +5155,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5204,17 +5171,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5236,16 +5198,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5256,14 +5217,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -5273,11 +5234,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5289,10 +5250,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -5304,7 +5261,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -5326,17 +5282,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5347,14 +5302,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -5364,11 +5319,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5380,10 +5335,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -5427,7 +5378,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5438,7 +5388,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5451,11 +5401,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5466,7 +5415,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -5476,12 +5425,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5509,7 +5454,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5517,16 +5462,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5546,10 +5490,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -5572,7 +5512,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5583,7 +5522,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5596,11 +5535,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5611,7 +5549,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -5621,12 +5559,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5654,7 +5589,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -5662,16 +5597,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5683,13 +5617,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -5701,7 +5631,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5720,21 +5649,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5744,11 +5672,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -5768,7 +5692,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5787,22 +5710,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5814,19 +5736,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5850,15 +5767,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5869,14 +5785,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5886,11 +5802,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5902,10 +5818,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -5914,7 +5826,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5938,16 +5849,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5958,14 +5868,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5975,11 +5885,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5991,10 +5901,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -6039,7 +5945,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6050,7 +5955,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6064,11 +5969,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6079,7 +5983,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -6091,12 +5995,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6124,7 +6024,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6132,16 +6032,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6161,10 +6060,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -6186,7 +6081,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6197,7 +6091,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6209,11 +6103,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6224,7 +6117,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -6234,12 +6127,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6267,7 +6156,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -6275,16 +6164,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6296,13 +6184,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6314,7 +6198,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6333,21 +6216,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6357,11 +6239,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -6382,7 +6260,6 @@ def get( of instance view of the dedicated hosts under the dedicated host group. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6401,22 +6278,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6428,13 +6304,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -6443,7 +6315,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6465,16 +6336,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6485,14 +6355,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6502,11 +6372,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6518,16 +6388,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -6549,15 +6414,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6568,14 +6432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -6585,11 +6449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6601,8 +6465,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -6629,7 +6491,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -6655,7 +6517,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -6664,16 +6526,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6693,10 +6554,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -6721,14 +6578,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -6741,7 +6590,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6755,18 +6604,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -6779,7 +6620,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -6791,19 +6632,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -6836,7 +6666,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6846,24 +6676,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -6889,7 +6717,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -6898,16 +6726,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6919,13 +6746,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6951,14 +6774,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -6971,7 +6786,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6985,18 +6800,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -7009,7 +6816,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -7021,19 +6828,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] @@ -7066,7 +6862,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7076,17 +6872,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -7105,22 +6899,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7130,11 +6923,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -7148,14 +6937,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7183,7 +6964,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7192,17 +6973,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7225,7 +7002,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -7244,23 +7020,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7272,13 +7047,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -7291,7 +7062,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -7313,17 +7083,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7334,14 +7103,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -7351,11 +7120,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7367,10 +7136,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -7397,7 +7162,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -7421,15 +7185,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7440,14 +7203,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -7457,11 +7220,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7473,8 +7236,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -7484,7 +7245,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -7508,16 +7268,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7528,14 +7287,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -7545,11 +7304,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7561,10 +7320,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -7586,7 +7341,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7597,7 +7351,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7609,11 +7363,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7624,7 +7377,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -7634,12 +7387,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7667,7 +7416,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -7675,16 +7424,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7704,10 +7452,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -7729,7 +7473,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7740,7 +7483,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7752,11 +7495,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7767,7 +7509,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -7777,12 +7519,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7810,7 +7549,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -7818,16 +7557,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7839,13 +7577,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7857,7 +7591,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7876,21 +7609,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7900,11 +7632,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -7914,7 +7642,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7933,21 +7660,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7959,13 +7685,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -7979,7 +7701,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7998,21 +7719,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8024,13 +7744,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -8067,7 +7783,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8086,23 +7801,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8114,13 +7828,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -8132,7 +7842,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8151,21 +7860,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8177,13 +7885,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -8211,7 +7915,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8230,7 +7933,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -8239,16 +7942,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8260,13 +7962,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -8294,7 +7992,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8320,7 +8018,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8329,16 +8027,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8358,10 +8055,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -8388,14 +8081,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8409,7 +8094,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8425,18 +8110,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8450,7 +8127,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8463,20 +8140,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8510,7 +8176,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8520,24 +8186,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8563,7 +8227,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8572,16 +8236,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8593,13 +8256,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8628,14 +8287,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8649,7 +8300,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8665,18 +8316,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8690,7 +8333,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8703,20 +8346,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8750,7 +8382,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8760,17 +8392,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8789,22 +8419,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8814,11 +8443,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -8833,14 +8458,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8868,7 +8485,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8877,17 +8494,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -8909,7 +8522,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -8928,23 +8540,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8956,13 +8567,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -8976,7 +8583,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8995,22 +8601,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9022,13 +8627,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -9067,7 +8668,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9086,7 +8686,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9094,16 +8694,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9115,13 +8714,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -9153,7 +8748,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9172,7 +8766,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9182,16 +8776,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9203,13 +8796,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -9221,7 +8810,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9240,21 +8828,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9266,13 +8853,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -9280,7 +8863,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9299,20 +8881,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9324,13 +8905,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -9344,7 +8921,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9363,22 +8939,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9390,13 +8965,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -9426,7 +8997,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9448,16 +9018,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9468,14 +9037,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -9485,11 +9054,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9501,8 +9070,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -9531,7 +9098,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -9553,16 +9119,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9573,14 +9138,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9590,11 +9155,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9606,15 +9171,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9640,7 +9201,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9648,16 +9209,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9671,13 +9231,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -9701,14 +9257,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9721,7 +9269,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9734,18 +9282,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9758,7 +9298,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9769,19 +9309,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9814,7 +9344,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9826,20 +9356,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9864,7 +9396,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9872,16 +9404,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9901,10 +9432,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -9927,14 +9454,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -9946,7 +9465,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9959,18 +9478,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -9979,7 +9490,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9989,19 +9504,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -10033,7 +9537,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10043,20 +9547,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -10081,7 +9587,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10089,16 +9595,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10118,10 +9623,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_update( self, @@ -10143,14 +9644,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -10162,7 +9655,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10174,18 +9667,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -10194,7 +9679,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -10203,19 +9692,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] @@ -10247,7 +9725,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10257,17 +9735,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -10286,22 +9762,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10311,11 +9786,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10330,14 +9801,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10365,7 +9828,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10374,17 +9837,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10399,7 +9858,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10418,22 +9876,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10445,13 +9902,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -10463,7 +9916,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10482,21 +9934,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10508,13 +9959,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10533,21 +9980,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10557,11 +10003,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10572,14 +10014,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10606,7 +10040,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10615,17 +10049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10644,21 +10074,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10668,11 +10097,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10683,14 +10108,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10717,7 +10134,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10726,17 +10143,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -10753,7 +10166,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10772,21 +10184,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10796,11 +10207,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -10809,7 +10216,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -10831,16 +10237,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10851,14 +10256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10868,11 +10273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10884,10 +10289,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -10896,7 +10297,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -10918,16 +10318,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10938,14 +10337,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10955,11 +10354,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10971,8 +10370,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10984,7 +10381,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -11006,17 +10402,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11027,14 +10422,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -11044,11 +10439,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11060,10 +10455,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -11081,22 +10472,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11106,11 +10496,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -11127,14 +10513,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11162,7 +10540,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11171,17 +10549,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11200,21 +10574,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11224,11 +10597,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11238,14 +10607,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11272,7 +10633,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11281,17 +10642,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11310,21 +10667,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11334,11 +10690,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11348,14 +10700,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11382,7 +10726,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11391,17 +10735,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11420,21 +10760,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11444,11 +10783,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11458,14 +10793,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11492,7 +10819,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11501,17 +10828,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11530,21 +10853,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11554,11 +10876,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11568,14 +10886,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11602,7 +10912,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11611,23 +10921,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11656,7 +10962,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11664,16 +10970,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11683,11 +10988,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -11711,14 +11012,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11729,7 +11022,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11742,18 +11035,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11764,7 +11049,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11774,19 +11059,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11816,7 +11091,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11825,17 +11100,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -11857,7 +11128,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11876,22 +11146,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11903,13 +11172,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11928,21 +11193,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11952,11 +11216,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11966,14 +11226,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12000,7 +11252,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12009,17 +11261,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -12031,7 +11279,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12050,21 +11297,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12074,11 +11320,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12097,21 +11339,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12125,13 +11366,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -12143,14 +11380,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -12180,7 +11409,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12192,20 +11421,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12230,7 +11461,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12238,16 +11469,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12261,13 +11491,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -12290,14 +11516,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -12309,7 +11527,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12321,18 +11539,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -12341,7 +11551,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12350,19 +11564,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -12394,7 +11597,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12406,17 +11609,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineSizesOperations: @@ -12446,7 +11647,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -12468,16 +11668,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12488,14 +11687,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12505,11 +11704,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12521,8 +11720,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12545,7 +11742,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12570,7 +11767,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12578,16 +11775,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12607,10 +11803,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -12632,14 +11824,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12650,7 +11834,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12662,18 +11846,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12681,7 +11857,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -12690,19 +11866,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12733,7 +11898,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12743,20 +11908,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12781,7 +11948,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12789,16 +11956,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12818,10 +11984,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -12843,14 +12005,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12861,7 +12015,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12873,18 +12027,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12892,7 +12038,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -12901,19 +12051,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12944,7 +12083,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12954,17 +12093,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -12983,21 +12120,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13007,11 +12143,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13021,14 +12153,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13055,7 +12179,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13064,17 +12188,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13088,7 +12208,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -13107,22 +12226,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13134,13 +12252,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -13148,7 +12262,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13170,16 +12283,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13190,14 +12302,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13207,11 +12319,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13223,16 +12335,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13254,15 +12361,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13273,14 +12379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13290,11 +12396,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13306,8 +12412,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -13333,7 +12437,7 @@ def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13359,7 +12463,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13367,16 +12471,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13396,10 +12499,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -13421,14 +12520,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13441,7 +12532,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13453,18 +12544,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13477,7 +12560,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -13486,20 +12569,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13532,7 +12604,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13542,23 +12614,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -13584,7 +12654,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13592,16 +12662,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13613,13 +12682,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -13642,14 +12707,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13662,7 +12719,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13674,18 +12731,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13698,7 +12747,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -13707,20 +12756,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13753,7 +12792,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13763,17 +12802,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13792,21 +12829,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13816,11 +12852,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13830,14 +12862,6 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13864,7 +12888,7 @@ def begin_delete(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13873,17 +12897,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -13893,7 +12913,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -13912,21 +12931,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13938,19 +12956,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13979,7 +12993,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13987,16 +13001,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14006,11 +13019,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -14037,14 +13046,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14055,7 +13056,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14070,18 +13071,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14092,7 +13085,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -14104,20 +13097,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14147,7 +13129,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14156,23 +13138,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -14198,7 +13176,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14206,16 +13184,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14225,11 +13202,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -14254,14 +13227,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14272,7 +13237,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14285,18 +13250,10 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14307,7 +13264,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Deletes virtual machines in a VM scale set. @@ -14317,20 +13274,9 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14360,7 +13306,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14369,17 +13315,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -14391,7 +13333,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -14410,21 +13351,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14436,13 +13376,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -14450,7 +13386,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14474,16 +13409,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14494,14 +13428,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -14511,11 +13445,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14527,17 +13461,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14561,15 +13490,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14580,14 +13508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -14597,11 +13525,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14613,8 +13541,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14626,7 +13552,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -14650,17 +13575,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14671,14 +13595,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -14688,11 +13612,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14704,10 +13628,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -14718,7 +13638,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -14742,17 +13661,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14763,14 +13681,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -14780,11 +13698,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14796,15 +13714,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14835,7 +13749,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14844,16 +13758,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14863,11 +13776,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -14899,14 +13808,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14917,7 +13818,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -14933,7 +13834,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -14941,14 +13842,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14959,7 +13852,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14973,24 +13866,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15021,7 +13903,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15030,23 +13912,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15075,7 +13953,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15083,16 +13961,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15102,11 +13979,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -15131,14 +14004,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15149,7 +14014,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15162,18 +14027,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15184,7 +14041,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -15194,20 +14051,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15237,7 +14083,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15246,23 +14092,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15291,7 +14133,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15299,16 +14141,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15318,11 +14159,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -15347,14 +14184,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15365,7 +14194,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15378,18 +14207,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15400,7 +14221,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -15410,20 +14231,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15453,7 +14263,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15462,23 +14272,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15507,7 +14313,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15515,16 +14321,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15534,11 +14339,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -15564,14 +14365,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15582,7 +14375,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15596,18 +14389,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15618,7 +14403,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -15629,20 +14414,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15672,7 +14446,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15681,23 +14455,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15726,7 +14496,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15734,16 +14504,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15753,11 +14522,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -15785,14 +14550,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15803,7 +14560,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15819,18 +14576,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15841,7 +14590,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -15854,20 +14603,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15897,7 +14635,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15906,23 +14644,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -15948,7 +14682,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15956,16 +14690,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15975,11 +14708,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -16004,14 +14733,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16022,7 +14743,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16035,18 +14756,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16057,7 +14770,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -16067,20 +14780,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16110,7 +14812,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16119,23 +14821,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16164,7 +14862,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16172,16 +14870,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16191,11 +14888,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -16221,14 +14914,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16239,7 +14924,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16253,18 +14938,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16275,7 +14952,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -16287,20 +14964,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16330,7 +14996,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16339,23 +15005,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16384,7 +15046,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16392,16 +15054,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16411,11 +15072,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -16441,14 +15098,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16459,7 +15108,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16473,18 +15122,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16495,7 +15136,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -16506,20 +15147,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16549,7 +15179,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16558,20 +15188,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -16584,7 +15210,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -16603,22 +15228,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16630,13 +15254,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -16661,7 +15281,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16672,7 +15291,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16685,11 +15304,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16700,7 +15318,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -16711,13 +15329,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2020_06_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2020_06_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16745,7 +15360,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16753,16 +15368,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16772,17 +15386,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -16808,7 +15418,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16816,16 +15426,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16835,11 +15444,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -16863,14 +15468,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16881,7 +15478,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16894,18 +15491,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16916,7 +15505,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -16927,19 +15516,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16969,7 +15548,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16978,20 +15557,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17016,7 +15591,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -17042,7 +15617,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -17051,16 +15626,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17080,10 +15654,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -17111,14 +15681,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17132,7 +15694,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17148,18 +15710,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17173,7 +15727,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -17186,20 +15740,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17233,7 +15776,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17243,24 +15786,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -17286,7 +15827,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -17295,16 +15836,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17324,10 +15864,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -17355,14 +15891,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17376,7 +15904,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17392,18 +15920,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17417,7 +15937,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -17430,20 +15950,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17477,7 +15987,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17487,17 +15997,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -17516,22 +16024,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17541,11 +16048,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -17560,14 +16063,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17595,7 +16090,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17604,17 +16099,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17636,7 +16127,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -17655,23 +16145,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17683,13 +16172,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -17701,7 +16186,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -17725,17 +16209,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17746,14 +16229,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -17763,11 +16246,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17779,12 +16262,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17821,21 +16300,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17845,11 +16323,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17859,14 +16333,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17893,7 +16359,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17902,17 +16368,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17931,21 +16393,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17955,11 +16416,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -17973,14 +16430,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18007,7 +16456,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18016,17 +16465,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -18045,21 +16490,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18069,11 +16513,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -18087,14 +16527,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18121,7 +16553,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18130,17 +16562,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -18152,7 +16580,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -18171,21 +16598,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18197,16 +16623,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18232,7 +16654,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -18258,7 +16680,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18268,16 +16690,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18297,10 +16718,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -18330,14 +16747,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18352,7 +16761,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18369,18 +16778,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18395,7 +16796,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -18409,20 +16810,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18457,7 +16847,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18467,17 +16857,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -18485,7 +16873,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -18511,7 +16899,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18521,16 +16909,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18542,13 +16929,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -18579,14 +16962,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18601,7 +16976,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18618,18 +16993,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18644,7 +17011,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -18658,20 +17025,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -18706,7 +17063,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18716,17 +17073,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -18745,23 +17100,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18771,11 +17125,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18791,14 +17141,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18827,7 +17169,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18836,17 +17178,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -18871,7 +17209,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -18890,7 +17227,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18898,16 +17235,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18919,13 +17255,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18947,7 +17279,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18966,23 +17297,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18994,13 +17324,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -19028,7 +17354,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -19057,7 +17385,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19066,16 +17394,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19085,11 +17412,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -19117,14 +17440,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19136,7 +17451,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -19151,18 +17466,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19174,7 +17481,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -19186,20 +17495,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -19230,7 +17529,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19239,17 +17538,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19268,22 +17563,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19293,11 +17587,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -19312,14 +17602,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19347,7 +17629,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19356,17 +17638,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19385,22 +17663,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19410,11 +17687,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -19430,14 +17703,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19465,7 +17730,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19474,24 +17739,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -19517,7 +17778,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19526,16 +17787,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19555,10 +17815,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -19585,14 +17841,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19606,7 +17854,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19622,18 +17870,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19647,7 +17887,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -19660,19 +17900,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19706,7 +17935,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19716,17 +17945,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19745,22 +17972,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19770,11 +17996,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -19788,14 +18010,6 @@ def begin_delete( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19823,7 +18037,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19832,17 +18046,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19865,7 +18075,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -19884,23 +18093,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19912,13 +18120,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -19932,7 +18136,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -19951,22 +18154,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19978,13 +18180,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -20013,7 +18211,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -20037,7 +18234,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -20045,12 +18242,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20061,14 +18257,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -20078,11 +18274,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20094,10 +18290,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -20121,23 +18313,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20147,11 +18338,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -20177,14 +18364,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20213,7 +18392,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20222,17 +18401,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20251,22 +18426,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20276,11 +18450,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -20294,14 +18464,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20329,7 +18491,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20338,17 +18500,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20367,22 +18525,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20392,11 +18549,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -20410,14 +18563,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20445,7 +18590,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20454,17 +18599,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20483,22 +18624,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20508,11 +18648,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -20527,14 +18663,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20562,7 +18690,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20571,17 +18699,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -20607,7 +18731,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -20626,23 +18749,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20654,13 +18776,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20679,22 +18797,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20704,11 +18821,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -20722,14 +18835,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20757,7 +18862,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20766,17 +18871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -20790,7 +18891,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20809,22 +18909,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20834,18 +18933,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -20871,7 +18966,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20880,16 +18975,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20903,13 +18997,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -20935,14 +19025,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -20955,7 +19037,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20969,18 +19051,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -20993,7 +19067,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -21005,19 +19079,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_01.models.RunCommandResult] @@ -21050,7 +19113,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21062,17 +19125,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -21096,7 +19157,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -21121,23 +19182,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21151,13 +19211,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -21179,14 +19235,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21196,7 +19244,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21205,18 +19253,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21226,7 +19266,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21234,19 +19274,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21278,7 +19308,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21290,20 +19320,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -21328,23 +19356,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21358,13 +19385,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -21385,14 +19408,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21402,7 +19417,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -21410,18 +19425,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21431,7 +19438,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -21439,19 +19446,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_06_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21483,7 +19479,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21495,17 +19491,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -21534,7 +19528,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -21558,16 +19551,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21578,14 +19570,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -21595,11 +19587,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21611,10 +19603,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -21623,7 +19611,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -21642,21 +19629,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21668,20 +19654,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21707,7 +19689,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21716,16 +19698,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21745,10 +19726,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -21775,14 +19752,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21796,7 +19765,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21812,18 +19781,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21837,7 +19798,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21850,19 +19811,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21896,7 +19846,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21906,24 +19856,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21949,7 +19897,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21958,16 +19906,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21979,13 +19926,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22013,14 +19956,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22034,7 +19969,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22050,18 +19985,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22075,7 +20002,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -22088,19 +20015,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22134,7 +20051,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22144,17 +20061,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -22173,22 +20088,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22198,11 +20112,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -22217,14 +20127,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22252,7 +20154,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22261,17 +20163,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -22293,7 +20191,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22312,23 +20209,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22340,13 +20236,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -22360,7 +20252,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22384,18 +20275,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22406,14 +20296,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22423,11 +20313,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22439,12 +20329,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -22470,7 +20356,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22496,7 +20382,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22506,16 +20392,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22535,10 +20420,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -22567,14 +20448,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22589,7 +20462,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22606,18 +20479,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22632,7 +20497,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -22646,19 +20511,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22693,7 +20547,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22703,17 +20557,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -22721,7 +20573,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22747,7 +20599,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22757,16 +20609,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22778,13 +20629,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22814,14 +20661,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22836,7 +20675,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22853,18 +20692,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22879,7 +20710,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22893,19 +20724,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22940,7 +20761,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22950,17 +20771,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22979,23 +20798,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23005,11 +20823,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23025,14 +20839,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23061,7 +20867,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23070,17 +20876,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23105,7 +20907,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -23124,7 +20925,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23132,16 +20933,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23153,13 +20953,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -23181,7 +20977,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23205,19 +21000,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23228,14 +21022,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -23245,11 +21039,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23260,7 +21054,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_compute_management_client.py index fe805233e397..1b1a1510620c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -58,7 +60,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -75,7 +95,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-06-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -95,7 +115,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_configuration.py index 8d9726a091af..351261f5d3a5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-06-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_metadata.json index cbe2d4ea9840..c6cec30768f9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_compute_management_client.py index 4d84b1335bce..64f5c120162d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -58,7 +60,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -75,7 +95,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-06-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -95,7 +117,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_configuration.py index 2bee7d403a83..d7763f4c48cb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-06-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/operations/_operations.py index 1112ec81a4c9..39d1e0627a09 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -89,7 +89,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -114,7 +114,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -122,16 +122,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -151,10 +150,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -178,14 +173,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -196,7 +183,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -210,18 +197,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -229,7 +208,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -240,19 +219,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_06_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_06_30.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -283,7 +251,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -293,20 +261,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -331,7 +297,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -339,16 +305,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -368,10 +333,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -395,14 +356,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -413,7 +366,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -427,18 +380,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -446,7 +391,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -457,19 +402,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_06_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_06_30.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -500,7 +434,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -510,17 +444,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -532,7 +464,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -551,21 +482,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -577,13 +507,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -602,21 +528,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -626,11 +551,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -642,14 +563,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -676,7 +589,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -685,17 +598,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -703,7 +612,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -725,16 +633,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -745,14 +652,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -762,11 +669,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -778,15 +685,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -808,15 +710,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -827,14 +728,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -844,11 +745,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -860,13 +761,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -892,7 +791,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -900,16 +799,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -923,13 +821,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -955,14 +849,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -974,7 +860,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -989,18 +875,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -1012,7 +890,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1024,19 +902,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -1068,7 +935,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1080,17 +947,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1109,21 +974,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1133,11 +997,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1151,14 +1011,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1185,7 +1037,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1196,17 +1048,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1230,7 +1078,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1255,7 +1103,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1263,16 +1111,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1292,10 +1139,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1319,14 +1162,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1338,7 +1173,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1352,18 +1187,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1372,7 +1199,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1383,19 +1210,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1427,7 +1243,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1437,20 +1253,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1475,7 +1293,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1483,16 +1301,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1512,10 +1329,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1540,14 +1353,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1559,7 +1364,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1574,18 +1379,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1594,7 +1391,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1605,19 +1406,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1649,7 +1439,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1659,17 +1449,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1681,7 +1469,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1700,21 +1487,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1726,13 +1512,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1751,21 +1533,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1775,11 +1556,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1791,14 +1568,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1825,7 +1594,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1834,17 +1603,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1852,7 +1617,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1875,16 +1639,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1895,14 +1658,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1912,11 +1675,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1928,15 +1691,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.Snapshot] @@ -1959,15 +1717,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1978,14 +1735,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1995,11 +1752,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2011,13 +1768,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2043,7 +1798,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2051,16 +1806,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2074,13 +1828,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2106,14 +1856,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -2125,7 +1867,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2140,18 +1882,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -2163,7 +1897,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2175,19 +1909,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] @@ -2219,7 +1942,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2231,17 +1954,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2260,21 +1981,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2284,11 +2004,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2302,14 +2018,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2336,7 +2044,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2347,17 +2055,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2384,7 +2088,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2410,7 +2114,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2418,16 +2122,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2447,10 +2150,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2475,14 +2174,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2495,7 +2186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2510,18 +2201,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2534,7 +2217,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2546,19 +2229,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2591,7 +2264,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2601,23 +2274,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2643,7 +2314,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2651,16 +2322,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2680,10 +2350,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2708,14 +2374,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2728,7 +2386,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2743,18 +2401,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2767,7 +2417,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2779,19 +2429,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2824,7 +2465,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2834,17 +2475,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2858,7 +2497,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2877,21 +2515,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2903,13 +2540,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2928,21 +2561,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2952,11 +2584,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2970,14 +2598,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3004,7 +2624,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3013,17 +2633,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3033,7 +2649,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3056,16 +2671,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3076,14 +2690,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3093,11 +2707,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3109,15 +2723,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3140,15 +2749,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2767,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3176,11 +2784,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2800,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3206,7 +2812,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3228,17 +2833,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3249,14 +2853,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3266,11 +2870,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3282,10 +2886,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -3308,7 +2908,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3333,7 +2937,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3341,16 +2945,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3370,10 +2973,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3398,14 +2997,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3417,7 +3008,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3432,18 +3023,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3452,7 +3035,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3463,19 +3050,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3507,7 +3083,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3517,23 +3093,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3559,7 +3133,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3567,16 +3141,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3596,10 +3169,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3624,14 +3193,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3643,7 +3204,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3658,18 +3219,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3681,7 +3234,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3693,19 +3246,8 @@ async def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3737,7 +3279,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3747,17 +3289,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3769,7 +3309,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3788,21 +3327,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3352,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3839,21 +3373,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3863,11 +3396,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3881,14 +3410,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3915,7 +3436,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3924,17 +3445,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -3942,7 +3459,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -3965,16 +3481,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3985,14 +3500,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4002,11 +3517,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4018,15 +3533,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -4049,15 +3559,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4068,14 +3577,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4085,11 +3594,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4101,8 +3610,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4115,7 +3622,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4134,21 +3640,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4160,10 +3665,6 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_compute_management_client_enums.py index 1e68aed798af..7d940d785701 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_compute_management_client_enums.py @@ -27,21 +27,18 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -55,10 +52,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -87,10 +84,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -98,13 +95,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class HyperVGeneration(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_models_py3.py index a1e58ae52ba3..bb5975f4f51f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/models/_models_py3.py @@ -131,11 +131,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2020_06_30.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -196,8 +195,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2020_06_30.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -239,7 +237,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -288,7 +286,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -538,7 +536,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -600,7 +598,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] @@ -656,7 +654,7 @@ class DiskEncryptionSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -741,7 +739,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -820,7 +818,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_30.models.Disk] @@ -1124,7 +1122,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1216,7 +1214,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2020_06_30.models.AccessLevel @@ -1249,7 +1247,7 @@ def __init__(self, *, access: Union[str, "_models.AccessLevel"], duration_in_sec class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1316,7 +1314,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_06_30.models.SourceVault @@ -1349,7 +1347,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_06_30.models.SourceVault @@ -1587,7 +1585,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -1648,7 +1646,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1823,7 +1821,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2020_06_30.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/operations/_operations.py index 7a413abf85e0..10042ac5692d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_06_30/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -697,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -728,7 +728,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -858,7 +858,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -964,7 +964,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -989,7 +989,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -997,16 +997,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1026,10 +1025,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1053,14 +1048,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1071,7 +1058,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1085,18 +1072,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1104,7 +1083,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1115,19 +1094,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_06_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_06_30.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1158,7 +1126,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1168,20 +1136,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1206,7 +1172,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1214,16 +1180,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1243,10 +1208,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1270,14 +1231,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1288,7 +1241,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1302,18 +1255,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1321,7 +1266,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1332,19 +1277,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_06_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_06_30.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1375,7 +1309,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1385,17 +1319,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1407,7 +1339,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1426,21 +1357,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1452,13 +1382,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1477,21 +1403,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1501,11 +1426,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1517,14 +1438,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1551,7 +1464,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1560,17 +1473,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1578,7 +1487,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1600,16 +1508,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1620,14 +1527,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1637,11 +1544,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1653,15 +1560,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1683,15 +1585,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1702,14 +1603,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1719,11 +1620,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1735,13 +1636,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -1767,7 +1666,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1775,16 +1674,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1798,13 +1696,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -1830,14 +1724,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1848,7 +1734,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1863,18 +1749,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1885,7 +1763,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1897,19 +1775,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -1940,7 +1807,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1952,17 +1819,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1981,21 +1846,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2005,11 +1869,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2021,14 +1881,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2055,7 +1907,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2066,17 +1918,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -2100,7 +1948,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2125,7 +1973,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2133,16 +1981,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2162,10 +2009,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -2189,14 +2032,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2207,7 +2042,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2221,18 +2056,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2240,7 +2067,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -2251,19 +2078,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2294,7 +2110,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2304,20 +2120,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2342,7 +2160,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2350,16 +2168,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2379,10 +2196,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -2407,14 +2220,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2425,7 +2230,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2440,18 +2245,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2459,7 +2256,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2470,19 +2271,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_06_30.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2513,7 +2303,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2523,17 +2313,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2545,7 +2333,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2564,21 +2351,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2590,13 +2376,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2615,21 +2397,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2639,11 +2420,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2655,14 +2432,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2689,7 +2458,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2698,17 +2467,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -2716,7 +2481,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2738,16 +2502,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2758,14 +2521,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2775,11 +2538,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2791,15 +2554,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2821,15 +2579,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2840,14 +2597,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2857,11 +2614,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2873,13 +2630,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2905,7 +2660,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2913,16 +2668,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2936,13 +2690,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2968,14 +2718,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2986,7 +2728,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3001,18 +2743,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3023,7 +2757,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -3035,19 +2769,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_06_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3078,7 +2801,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3090,17 +2813,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -3119,21 +2840,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3143,11 +2863,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3159,14 +2875,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3193,7 +2901,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -3204,17 +2912,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -3241,7 +2945,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3267,7 +2971,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3275,16 +2979,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3304,10 +3007,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3332,14 +3031,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3351,7 +3042,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3366,18 +3057,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3389,7 +3072,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3401,19 +3084,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3445,7 +3118,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3455,23 +3128,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3497,7 +3168,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3505,16 +3176,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3534,10 +3204,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -3562,14 +3228,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3581,7 +3239,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3596,18 +3254,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3619,7 +3269,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3631,19 +3281,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] @@ -3675,7 +3316,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3685,17 +3326,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -3707,7 +3346,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3726,21 +3364,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3752,13 +3389,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3777,21 +3410,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3801,11 +3433,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3817,14 +3445,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3851,7 +3471,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3860,17 +3480,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -3878,7 +3494,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3900,16 +3515,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3920,14 +3534,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3937,11 +3551,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3953,15 +3567,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -3983,15 +3592,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4002,14 +3610,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4019,11 +3627,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4035,8 +3643,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4049,7 +3655,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4071,17 +3676,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4092,14 +3696,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4109,11 +3713,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4125,10 +3729,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -4151,7 +3751,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -4176,7 +3780,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4184,16 +3788,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4213,10 +3816,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -4241,14 +3840,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4259,7 +3850,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4274,18 +3865,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4293,7 +3876,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -4304,19 +3891,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4347,7 +3923,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4357,23 +3933,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -4399,7 +3973,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4407,16 +3981,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4436,10 +4009,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -4464,14 +4033,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4482,7 +4043,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4497,18 +4058,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4519,7 +4072,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -4531,19 +4084,8 @@ def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_06_30.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4574,7 +4116,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4584,17 +4126,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -4606,7 +4146,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -4625,21 +4164,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4651,13 +4189,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4676,21 +4210,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4700,11 +4233,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4716,14 +4245,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4750,7 +4271,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4759,17 +4280,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -4777,7 +4294,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4799,16 +4315,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4819,14 +4334,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4836,11 +4351,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4852,15 +4367,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_06_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4882,15 +4392,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4901,14 +4410,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4918,11 +4427,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4934,8 +4443,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4948,7 +4455,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_06_30.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4967,21 +4473,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-06-30")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4993,10 +4498,6 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_compute_management_client.py index dfd2d076a49e..aa90e8f9b8e6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -100,7 +102,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -147,7 +167,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-09-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -167,7 +187,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_configuration.py index b24e56d43664..772bcbe13f84 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-09-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_metadata.json index eda11b338739..b17a6e2b3f73 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_compute_management_client.py index c4da54f917b5..e9a75c5da180 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -101,7 +103,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -148,7 +168,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-09-30" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -168,7 +190,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_configuration.py index 7dd5fffa73c0..55eb9f613725 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-09-30") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/operations/_operations.py index c10b4e8a599f..b663bfb296d1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -128,7 +128,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -153,7 +153,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -161,16 +161,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -190,10 +189,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -217,14 +212,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -235,7 +222,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -249,18 +236,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -268,7 +247,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -279,19 +258,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_09_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_09_30.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -322,7 +290,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -332,20 +300,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -370,7 +336,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -378,16 +344,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -407,10 +372,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -434,14 +395,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -452,7 +405,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -466,18 +419,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -485,7 +430,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -496,19 +441,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_09_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_09_30.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -539,7 +473,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -549,17 +483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -571,7 +503,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -590,21 +521,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -616,13 +546,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -641,21 +567,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -665,11 +590,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -681,14 +602,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -715,7 +628,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -724,17 +637,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -742,7 +651,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -764,16 +672,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -784,14 +691,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -801,11 +708,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -817,15 +724,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -847,15 +749,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -866,14 +767,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -883,11 +784,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -899,13 +800,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -931,7 +830,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -939,16 +838,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -962,13 +860,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -994,14 +888,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -1013,7 +899,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1028,18 +914,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -1051,7 +929,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1063,19 +941,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -1107,7 +974,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1119,17 +986,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1148,21 +1013,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1172,11 +1036,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1190,14 +1050,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1224,7 +1076,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1235,17 +1087,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1269,7 +1117,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1294,7 +1142,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1302,16 +1150,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1331,10 +1178,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1358,14 +1201,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1377,7 +1212,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1391,18 +1226,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1411,7 +1238,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1422,19 +1249,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1466,7 +1282,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1476,20 +1292,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1514,7 +1332,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1522,16 +1340,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1551,10 +1368,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1579,14 +1392,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1598,7 +1403,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1613,18 +1418,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1633,7 +1430,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1644,19 +1445,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1688,7 +1478,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1698,17 +1488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1720,7 +1508,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1739,21 +1526,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1765,13 +1551,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1790,21 +1572,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1814,11 +1595,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1830,14 +1607,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1864,7 +1633,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1873,17 +1642,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1891,7 +1656,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1914,16 +1678,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1934,14 +1697,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1951,11 +1714,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1967,15 +1730,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -1998,15 +1756,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2017,14 +1774,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2034,11 +1791,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2050,13 +1807,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2082,7 +1837,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2090,16 +1845,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2113,13 +1867,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2145,14 +1895,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -2164,7 +1906,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2179,18 +1921,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -2202,7 +1936,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2214,19 +1948,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] @@ -2258,7 +1981,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2270,17 +1993,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2299,21 +2020,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2323,11 +2043,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2341,14 +2057,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2375,7 +2083,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2386,17 +2094,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2423,7 +2127,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2449,7 +2153,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2457,16 +2161,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2486,10 +2189,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2514,14 +2213,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2534,7 +2225,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2549,18 +2240,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2573,7 +2256,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2585,19 +2268,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2630,7 +2303,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2640,23 +2313,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2682,7 +2353,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2690,16 +2361,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2719,10 +2389,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2747,14 +2413,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2767,7 +2425,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2782,18 +2440,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2806,7 +2456,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2818,19 +2468,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2863,7 +2504,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2873,17 +2514,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2897,7 +2536,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2916,21 +2554,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2942,13 +2579,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2967,21 +2600,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2991,11 +2623,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3009,14 +2637,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3043,7 +2663,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3052,17 +2672,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3072,7 +2688,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -3095,16 +2710,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3115,14 +2729,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3132,11 +2746,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3148,15 +2762,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -3179,15 +2788,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3198,14 +2806,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3215,11 +2823,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3231,8 +2839,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3245,7 +2851,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3267,17 +2872,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3288,14 +2892,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3305,11 +2909,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3321,10 +2925,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -3347,7 +2947,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3372,7 +2976,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3380,16 +2984,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3409,10 +3012,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3437,14 +3036,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3456,7 +3047,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3471,18 +3062,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3491,7 +3074,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3502,19 +3089,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3546,7 +3122,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3556,23 +3132,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3598,7 +3172,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3606,16 +3180,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3635,10 +3208,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3663,14 +3232,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3682,7 +3243,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3697,18 +3258,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3720,7 +3273,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3732,19 +3285,8 @@ async def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -3776,7 +3318,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3786,17 +3328,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3808,7 +3348,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3827,21 +3366,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3853,13 +3391,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3878,21 +3412,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3902,11 +3435,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3920,14 +3449,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3954,7 +3475,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3963,17 +3484,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -3981,7 +3498,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -4004,16 +3520,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4024,14 +3539,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4041,11 +3556,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4057,15 +3572,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -4088,15 +3598,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4107,14 +3616,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4124,11 +3633,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4140,8 +3649,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4154,7 +3661,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4173,21 +3679,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4199,20 +3704,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -4238,7 +3739,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -4247,16 +3748,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4276,12 +3776,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -4309,14 +3805,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4325,12 +3813,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4348,18 +3836,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4368,12 +3848,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -4389,20 +3869,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4436,7 +3905,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4446,17 +3915,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -4472,7 +3939,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -4491,22 +3957,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4518,15 +3983,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -4543,22 +4004,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4568,14 +4028,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -4588,14 +4044,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4623,7 +4071,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4632,17 +4080,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -4656,7 +4100,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4680,17 +4123,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4701,14 +4143,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -4718,11 +4160,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4734,10 +4176,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -4784,7 +4222,6 @@ async def get( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4803,23 +4240,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4831,13 +4267,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4855,7 +4287,6 @@ def list_by_restore_point( belongs. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskRestorePoint] @@ -4878,18 +4309,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4900,14 +4330,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4917,11 +4347,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4933,10 +4363,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - class GalleriesOperations: """ @@ -4959,7 +4385,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -4984,7 +4410,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4992,16 +4418,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5024,10 +4449,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -5052,14 +4473,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5071,7 +4484,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5086,18 +4499,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5106,7 +4511,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -5117,19 +4522,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2020_09_30.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2020_09_30.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5161,7 +4555,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5171,20 +4565,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -5209,7 +4605,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5217,16 +4613,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5238,13 +4633,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -5269,14 +4660,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5288,7 +4671,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5302,18 +4685,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5322,7 +4697,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -5333,19 +4712,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2020_09_30.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2020_09_30.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -5377,7 +4745,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5387,17 +4755,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -5417,7 +4783,6 @@ async def get( :keyword select: The select expression to apply on the operation. "Permissions" Default value is None. :paramtype select: str or ~azure.mgmt.compute.v2020_09_30.models.SelectPermissions - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5436,22 +4801,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5463,13 +4827,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -5488,21 +4848,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5512,11 +4871,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -5526,14 +4881,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5560,7 +4907,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5569,17 +4916,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -5587,7 +4930,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5609,16 +4951,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5629,14 +4970,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -5646,11 +4987,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5662,15 +5003,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5692,15 +5028,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5711,14 +5046,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -5728,11 +5063,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5744,8 +5079,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -5772,7 +5105,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -5798,7 +5131,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -5807,16 +5140,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5839,10 +5171,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -5871,14 +5199,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -5891,7 +5211,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5909,18 +5229,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -5933,7 +5245,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -5948,19 +5260,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -5993,7 +5294,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6003,24 +5304,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -6046,7 +5345,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -6055,16 +5354,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6076,13 +5374,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6111,14 +5405,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -6131,7 +5417,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6148,18 +5434,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -6172,7 +5450,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -6187,19 +5465,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -6232,7 +5499,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6242,17 +5509,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -6267,7 +5532,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6286,22 +5550,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6313,13 +5576,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -6338,22 +5597,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6363,11 +5621,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6382,14 +5636,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6417,7 +5663,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6426,17 +5672,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -6449,7 +5691,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -6472,17 +5713,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6493,14 +5733,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -6510,11 +5750,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6526,10 +5766,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -6557,7 +5793,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -6583,7 +5819,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -6593,16 +5829,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6625,10 +5860,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -6662,14 +5893,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -6684,7 +5907,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6706,18 +5929,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -6732,7 +5947,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -6751,19 +5966,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -6798,7 +6003,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6808,17 +6013,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -6826,7 +6029,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -6852,7 +6055,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -6862,16 +6065,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6883,13 +6085,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6924,14 +6122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -6946,7 +6136,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6968,18 +6158,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -6994,7 +6176,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -7013,20 +6195,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -7061,7 +6232,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7071,17 +6242,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -7110,7 +6279,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2020_09_30.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -7129,7 +6297,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -7137,16 +6305,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7158,13 +6325,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7188,23 +6351,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7214,11 +6376,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7242,14 +6400,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7278,7 +6428,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7287,17 +6437,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -7313,7 +6459,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion] @@ -7336,18 +6481,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7358,14 +6502,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -7375,11 +6519,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7391,10 +6535,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -7421,7 +6561,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -7447,7 +6587,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -7456,16 +6596,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7488,10 +6627,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -7520,14 +6655,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7541,7 +6668,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7559,18 +6686,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7584,7 +6703,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -7599,19 +6718,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7645,7 +6754,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7655,24 +6764,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -7698,7 +6805,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -7707,16 +6814,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7728,13 +6834,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -7764,14 +6866,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7785,7 +6879,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7803,18 +6897,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7828,7 +6914,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -7843,20 +6929,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -7890,7 +6965,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7900,17 +6975,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -7926,7 +6999,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -7945,22 +7017,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7972,13 +7043,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -7997,22 +7064,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8022,11 +7088,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8042,14 +7104,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8077,7 +7131,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8086,17 +7140,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -8109,7 +7159,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryApplication] @@ -8132,17 +7181,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8153,14 +7201,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -8170,11 +7218,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8186,10 +7234,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -8217,7 +7261,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -8243,7 +7287,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -8253,16 +7297,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8285,10 +7328,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -8323,14 +7362,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8345,7 +7376,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8367,18 +7398,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8393,7 +7416,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -8412,21 +7435,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8461,7 +7473,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8471,17 +7483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -8489,7 +7499,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -8515,7 +7525,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -8525,16 +7535,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8546,13 +7555,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8588,14 +7593,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8610,7 +7607,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8632,18 +7629,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8658,7 +7647,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -8677,20 +7666,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -8725,7 +7704,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8735,17 +7714,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -8774,7 +7751,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2020_09_30.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -8793,7 +7769,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -8801,16 +7777,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8822,13 +7797,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -8852,23 +7823,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8878,11 +7848,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8906,14 +7872,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8942,7 +7900,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8951,17 +7909,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -8977,7 +7931,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -9001,18 +7954,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9023,14 +7975,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -9040,11 +7992,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9056,10 +8008,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -9085,7 +8033,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -9111,7 +8059,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -9119,16 +8067,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9148,10 +8095,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -9173,14 +8116,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -9193,7 +8128,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9205,18 +8140,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -9229,7 +8156,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -9239,19 +8166,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2020_09_30.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2020_09_30.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -9284,7 +8200,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9294,17 +8210,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -9338,7 +8252,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.SharedGallery] @@ -9361,17 +8274,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9382,14 +8294,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -9399,11 +8311,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9415,10 +8327,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -9427,7 +8335,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -9446,21 +8353,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9472,13 +8378,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -9519,7 +8421,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImage] @@ -9542,18 +8443,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9564,14 +8464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -9581,11 +8481,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9597,10 +8497,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -9614,7 +8510,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9633,22 +8528,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9660,13 +8554,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -9711,7 +8601,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -9735,19 +8624,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9758,14 +8646,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -9775,11 +8663,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9791,10 +8679,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -9818,7 +8702,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -9837,23 +8720,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9865,10 +8747,6 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_compute_management_client_enums.py index b0fc97859585..960c9b1212c1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_compute_management_client_enums.py @@ -36,21 +36,18 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -69,10 +66,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -101,10 +98,10 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -112,13 +109,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_models_py3.py index fa0ffe3d2fa8..75476c3ebf0c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/models/_models_py3.py @@ -132,11 +132,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2020_09_30.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -197,8 +196,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2020_09_30.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -260,7 +258,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -319,7 +317,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -368,7 +366,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -647,7 +645,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -709,7 +707,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] @@ -765,7 +763,7 @@ class DiskEncryptionSet(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -850,7 +848,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -929,7 +927,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.Disk] @@ -1080,7 +1078,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.DiskRestorePoint] @@ -1440,7 +1438,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1566,7 +1564,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1647,7 +1645,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1748,7 +1746,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.GalleryApplication] @@ -1921,7 +1919,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1993,7 +1991,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion] @@ -2104,12 +2102,12 @@ def __init__( self.storage_account_type = storage_account_type -class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): +class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): # pylint: disable=name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -2276,7 +2274,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2020_09_30.models.ManagedArtifact @@ -2380,7 +2378,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -2459,7 +2457,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2651,7 +2649,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -2691,7 +2689,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -2886,7 +2884,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2964,7 +2962,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion] @@ -3023,58 +3021,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) :vartype storage_account_type: str or ~azure.mgmt.compute.v2020_09_30.models.StorageAccountType """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2020_09_30.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2020_09_30.models.StorageAccountType - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -3189,7 +3135,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.Gallery] @@ -3234,32 +3180,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2020_09_30.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2020_09_30.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2020_09_30.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryUpdate(UpdateResourceDefinition): """Specifies information about the Shared Image Gallery that you want to update. @@ -3335,7 +3255,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2020_09_30.models.AccessLevel @@ -3368,7 +3288,7 @@ def __init__(self, *, access: Union[str, "_models.AccessLevel"], duration_in_sec class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -3473,7 +3393,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -3510,7 +3430,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_09_30.models.SourceVault @@ -3543,7 +3463,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_09_30.models.SourceVault @@ -3576,7 +3496,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3607,18 +3527,6 @@ class OSDiskImageEncryption(DiskImageEncryption): :vartype disk_encryption_set_id: str """ - _attribute_map = { - "disk_encryption_set_id": {"key": "diskEncryptionSetId", "type": "str"}, - } - - def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword disk_encryption_set_id: A relative URI containing the resource ID of the disk - encryption set. - :paramtype disk_encryption_set_id: str - """ - super().__init__(disk_encryption_set_id=disk_encryption_set_id, **kwargs) - class PirResource(_serialization.Model): """The Resource model definition. @@ -3923,7 +3831,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -4107,7 +4015,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -4153,24 +4061,6 @@ class SharedGallery(PirSharedGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this shared gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class SharedGalleryImage(PirSharedGalleryResource): # pylint: disable=too-many-instance-attributes """Specifies information about the gallery image definition that you want to create or update. @@ -4293,7 +4183,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImage] @@ -4384,7 +4274,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImageVersion] @@ -4420,7 +4310,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.SharedGallery] @@ -4551,7 +4441,7 @@ def __init__( class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. :code:`
    `:code:`
    ` Possible values are: :code:`
    `:code:`
    ` **Add** @@ -4599,7 +4489,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4796,7 +4686,7 @@ def __init__( class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2020_09_30.models.Snapshot] @@ -4984,7 +4874,7 @@ def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylin class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -5045,7 +4935,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -5092,7 +4982,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/operations/_operations.py index c80ce2297e08..45636b74579c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_09_30/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -697,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -728,7 +728,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -858,7 +858,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -943,7 +943,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -984,7 +984,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1022,7 +1022,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1060,7 +1060,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1131,7 +1131,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1306,7 +1306,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1360,7 +1360,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1494,7 +1494,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1525,7 +1525,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -1566,7 +1566,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -1649,7 +1649,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -1687,7 +1687,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1719,7 +1719,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1754,7 +1754,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1821,7 +1821,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1853,7 +1853,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1884,7 +1884,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -1927,7 +1927,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -1970,7 +1970,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -2014,7 +2014,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -2054,7 +2054,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2086,7 +2086,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2257,7 +2257,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -2297,7 +2297,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -2356,7 +2356,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -2381,7 +2381,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2389,16 +2389,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2418,10 +2417,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -2445,14 +2440,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2463,7 +2450,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2477,18 +2464,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2496,7 +2475,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -2507,19 +2486,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_09_30.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_09_30.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2550,7 +2518,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2560,20 +2528,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -2598,7 +2564,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2606,16 +2572,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2635,10 +2600,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -2662,14 +2623,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2680,7 +2633,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2694,18 +2647,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2713,7 +2658,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -2724,19 +2669,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_09_30.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_09_30.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2767,7 +2701,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2777,17 +2711,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -2799,7 +2731,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -2818,21 +2749,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2844,13 +2774,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2869,21 +2795,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2893,11 +2818,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2909,14 +2830,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2943,7 +2856,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2952,17 +2865,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -2970,7 +2879,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -2992,16 +2900,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3012,14 +2919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -3029,11 +2936,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3045,15 +2952,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -3075,15 +2977,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3094,14 +2995,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -3111,11 +3012,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3127,13 +3028,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3159,7 +3058,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -3167,16 +3066,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3190,13 +3088,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -3222,14 +3116,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3240,7 +3126,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3255,18 +3141,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3277,7 +3155,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -3289,19 +3167,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3332,7 +3199,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3344,17 +3211,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -3373,21 +3238,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3397,11 +3261,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3413,14 +3273,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3447,7 +3299,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -3458,17 +3310,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -3492,7 +3340,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -3517,7 +3365,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -3525,16 +3373,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3554,10 +3401,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -3581,14 +3424,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3599,7 +3434,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3613,18 +3448,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3632,7 +3459,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -3643,19 +3470,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3686,7 +3502,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3696,20 +3512,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -3734,7 +3552,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -3742,16 +3560,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3771,10 +3588,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -3799,14 +3612,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3817,7 +3622,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3832,18 +3637,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3851,7 +3648,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -3862,19 +3663,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_09_30.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3905,7 +3695,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3915,17 +3705,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -3937,7 +3725,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -3956,21 +3743,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3982,13 +3768,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4007,21 +3789,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4031,11 +3812,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4047,14 +3824,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4081,7 +3850,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4090,17 +3859,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -4108,7 +3873,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -4130,16 +3894,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4150,14 +3913,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4167,11 +3930,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4183,15 +3946,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -4213,15 +3971,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4232,14 +3989,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4249,11 +4006,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4265,13 +4022,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4297,7 +4052,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4305,16 +4060,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4328,13 +4082,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4360,14 +4110,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -4378,7 +4120,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4393,18 +4135,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -4415,7 +4149,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -4427,19 +4161,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_09_30.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -4470,7 +4193,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4482,17 +4205,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4511,21 +4232,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4535,11 +4255,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4551,14 +4267,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4585,7 +4293,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -4596,17 +4304,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -4633,7 +4337,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4659,7 +4363,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4667,16 +4371,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4696,10 +4399,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -4724,14 +4423,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -4743,7 +4434,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4758,18 +4449,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -4781,7 +4464,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -4793,19 +4476,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -4837,7 +4510,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4847,23 +4520,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4889,7 +4560,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4897,16 +4568,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4926,10 +4596,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4954,14 +4620,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -4973,7 +4631,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4988,18 +4646,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -5011,7 +4661,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -5023,19 +4673,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] @@ -5067,7 +4708,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5077,17 +4718,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -5099,7 +4738,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -5118,21 +4756,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5144,13 +4781,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -5169,21 +4802,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5193,11 +4825,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5209,14 +4837,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5243,7 +4863,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5252,17 +4872,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -5270,7 +4886,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5292,16 +4907,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5312,14 +4926,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -5329,11 +4943,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5345,15 +4959,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -5375,15 +4984,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5394,14 +5002,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -5411,11 +5019,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5427,8 +5035,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -5441,7 +5047,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -5463,17 +5068,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5484,14 +5088,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -5501,11 +5105,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5517,10 +5121,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -5543,7 +5143,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -5568,7 +5172,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -5576,16 +5180,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5605,10 +5208,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -5633,14 +5232,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5651,7 +5242,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5666,18 +5257,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5685,7 +5268,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -5696,19 +5283,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5739,7 +5315,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5749,23 +5325,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -5791,7 +5365,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -5799,16 +5373,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5828,10 +5401,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -5856,14 +5425,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5874,7 +5435,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5889,18 +5450,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5911,7 +5464,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -5923,19 +5476,8 @@ def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_09_30.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5966,7 +5508,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5976,17 +5518,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -5998,7 +5538,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -6017,21 +5556,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6043,13 +5581,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -6068,21 +5602,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6092,11 +5625,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6108,14 +5637,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6142,7 +5663,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6151,17 +5672,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -6169,7 +5686,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -6191,16 +5707,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6211,14 +5726,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -6228,11 +5743,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6244,15 +5759,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -6274,15 +5784,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6293,14 +5802,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -6310,11 +5819,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6326,8 +5835,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -6340,7 +5847,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6359,21 +5865,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6385,20 +5890,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -6424,7 +5925,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -6433,16 +5934,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6462,12 +5962,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -6495,14 +5991,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -6511,12 +5999,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6534,18 +6022,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -6554,12 +6034,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -6575,20 +6055,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -6622,7 +6091,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6632,17 +6101,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -6658,7 +6125,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -6677,22 +6143,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6704,15 +6169,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -6729,22 +6190,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6754,14 +6214,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -6774,14 +6230,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6809,7 +6257,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6818,17 +6266,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -6842,7 +6286,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -6866,17 +6309,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6887,14 +6329,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -6904,11 +6346,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6920,10 +6362,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -6970,7 +6408,6 @@ def get( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -6989,23 +6426,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7017,13 +6453,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -7041,7 +6473,6 @@ def list_by_restore_point( belongs. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -7063,18 +6494,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7085,14 +6515,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -7102,11 +6532,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7118,10 +6548,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - class GalleriesOperations: """ @@ -7144,7 +6570,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -7169,7 +6595,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -7177,16 +6603,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7209,10 +6634,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -7237,14 +6658,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7255,7 +6668,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7270,18 +6683,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7289,7 +6694,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -7300,19 +6705,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2020_09_30.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2020_09_30.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7343,7 +6737,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7353,20 +6747,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -7391,7 +6787,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -7399,16 +6795,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7420,13 +6815,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7451,14 +6842,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7469,7 +6852,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7483,18 +6866,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7502,7 +6877,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -7513,19 +6892,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2020_09_30.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2020_09_30.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7556,7 +6924,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7566,17 +6934,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -7596,7 +6962,6 @@ def get( :keyword select: The select expression to apply on the operation. "Permissions" Default value is None. :paramtype select: str or ~azure.mgmt.compute.v2020_09_30.models.SelectPermissions - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -7615,22 +6980,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7642,13 +7006,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -7667,21 +7027,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7691,11 +7050,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7705,14 +7060,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7739,7 +7086,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7748,17 +7095,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -7766,7 +7109,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7788,16 +7130,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7808,14 +7149,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -7825,11 +7166,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7841,15 +7182,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -7871,15 +7207,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7890,14 +7225,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -7907,11 +7242,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7923,8 +7258,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -7951,7 +7284,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -7977,7 +7310,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -7986,16 +7319,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8018,10 +7350,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -8050,14 +7378,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8070,7 +7390,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8088,18 +7408,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8112,7 +7424,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -8127,19 +7439,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8172,7 +7473,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8182,24 +7483,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -8225,7 +7524,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -8234,16 +7533,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8255,13 +7553,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8290,14 +7584,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8310,7 +7596,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8327,18 +7613,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8351,7 +7629,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -8366,19 +7644,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] @@ -8411,7 +7678,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8421,17 +7688,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -8446,7 +7711,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8465,22 +7729,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8492,13 +7755,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -8517,22 +7776,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8542,11 +7800,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -8561,14 +7815,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8596,7 +7842,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8605,17 +7851,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -8628,7 +7870,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -8650,17 +7891,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8671,14 +7911,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -8688,11 +7928,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8704,10 +7944,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -8735,7 +7971,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -8761,7 +7997,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -8771,16 +8007,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8803,10 +8038,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -8840,14 +8071,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -8862,7 +8085,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8884,18 +8107,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -8910,7 +8125,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -8929,19 +8144,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -8976,7 +8181,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8986,17 +8191,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -9004,7 +8207,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -9030,7 +8233,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -9040,16 +8243,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9061,13 +8263,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9102,14 +8300,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -9124,7 +8314,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9146,18 +8336,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -9172,7 +8354,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -9191,20 +8373,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -9239,7 +8410,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9249,17 +8420,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -9288,7 +8457,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2020_09_30.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -9307,7 +8475,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -9315,16 +8483,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9336,13 +8503,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -9366,23 +8529,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9392,11 +8554,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -9420,14 +8578,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9456,7 +8606,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9465,17 +8615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -9491,7 +8637,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryImageVersion] @@ -9514,18 +8659,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9536,14 +8680,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -9553,11 +8697,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9569,10 +8713,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -9599,7 +8739,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -9625,7 +8765,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -9634,16 +8774,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9666,10 +8805,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -9698,14 +8833,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -9719,7 +8846,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9737,18 +8864,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -9762,7 +8881,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -9777,19 +8896,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -9823,7 +8932,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9833,24 +8942,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -9876,7 +8983,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -9885,16 +8992,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9906,13 +9012,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9942,14 +9044,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -9963,7 +9057,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9981,18 +9075,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -10006,7 +9092,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -10021,20 +9107,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -10068,7 +9143,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10078,17 +9153,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -10104,7 +9177,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -10123,22 +9195,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10150,13 +9221,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -10175,22 +9242,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10200,11 +9266,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10220,14 +9282,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10255,7 +9309,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10264,17 +9318,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -10287,7 +9337,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -10309,17 +9358,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10330,14 +9378,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -10347,11 +9395,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10363,10 +9411,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -10394,7 +9438,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -10420,7 +9464,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -10430,16 +9474,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10462,10 +9505,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -10500,14 +9539,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10522,7 +9553,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10544,18 +9575,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10570,7 +9593,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -10589,21 +9612,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10638,7 +9650,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10648,17 +9660,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -10666,7 +9676,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -10692,7 +9702,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -10702,16 +9712,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10723,13 +9732,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -10765,14 +9770,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10787,7 +9784,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10809,18 +9806,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10835,7 +9824,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -10854,20 +9843,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -10902,7 +9881,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10912,17 +9891,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -10951,7 +9928,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2020_09_30.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -10970,7 +9946,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -10978,16 +9954,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10999,13 +9974,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -11029,23 +10000,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11055,11 +10025,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11083,14 +10049,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11119,7 +10077,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11128,17 +10086,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -11154,7 +10108,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -11178,18 +10131,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11200,14 +10152,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -11217,11 +10169,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11233,10 +10185,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -11262,7 +10210,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -11288,7 +10236,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -11296,16 +10244,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11325,10 +10272,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -11350,14 +10293,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.SharingUpdate] @@ -11369,7 +10304,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11381,18 +10316,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.SharingUpdate] @@ -11404,7 +10331,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -11414,19 +10341,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2020_09_30.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2020_09_30.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_09_30.models.SharingUpdate] @@ -11458,7 +10374,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11468,17 +10384,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -11512,7 +10426,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -11534,17 +10447,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11555,14 +10467,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -11572,11 +10484,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11588,10 +10500,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -11600,7 +10508,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -11619,21 +10526,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11645,13 +10551,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -11692,7 +10594,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -11714,18 +10615,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11736,14 +10636,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -11753,11 +10653,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11769,10 +10669,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -11786,7 +10682,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -11805,22 +10700,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11832,13 +10726,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -11883,7 +10773,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2020_09_30.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -11907,19 +10796,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11930,14 +10818,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -11947,11 +10835,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11963,10 +10851,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -11990,7 +10874,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_09_30.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -12009,23 +10892,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-09-30")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12037,10 +10919,6 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_compute_management_client.py index be2c5c051b55..6bc345842fa1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -66,7 +68,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -85,7 +105,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-10-01-preview" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -105,7 +125,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_configuration.py index 4d691de15da5..26fdb9f93ef6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-10-01-preview") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_metadata.json index ca09fb715d00..d90e8a5a7f1e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "cloud_service_role_instances": "CloudServiceRoleInstancesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_compute_management_client.py index 92dbfdd19295..ccd92230a4de 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -66,7 +68,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -85,7 +105,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-10-01-preview" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -105,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_configuration.py index ea8a282acc14..ad31eb96480f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-10-01-preview") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/operations/_operations.py index 2864849c2521..adf38d0c890d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,20 @@ # 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, TypeVar, Union, cast, overload +from typing import ( + Any, + AsyncIterable, + AsyncIterator, + Callable, + Dict, + IO, + Literal, + Optional, + TypeVar, + Union, + cast, + overload, +) import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -61,10 +73,6 @@ build_cloud_services_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -108,22 +116,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -133,11 +140,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -151,14 +154,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -188,7 +183,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -197,17 +192,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -230,7 +221,6 @@ async def get( :keyword expand: The expand expression to apply to the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -251,23 +241,22 @@ async def get( ) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -279,13 +268,9 @@ async def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -299,7 +284,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -320,22 +304,21 @@ async def get_instance_view( ) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -347,13 +330,9 @@ async def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -375,7 +354,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstance] @@ -400,18 +378,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -422,14 +399,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -439,11 +416,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -455,10 +432,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -478,22 +451,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -503,11 +475,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -522,14 +490,6 @@ async def begin_restart( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -559,7 +519,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -568,17 +528,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -599,22 +555,21 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -624,11 +579,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage( @@ -643,14 +594,6 @@ async def begin_reimage( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -680,7 +623,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -689,17 +632,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -720,22 +659,21 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -745,11 +683,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_rebuild( @@ -765,14 +699,6 @@ async def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -802,7 +728,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -811,17 +737,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_remote_desktop_file( @@ -835,8 +757,7 @@ async def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Async iterator of the response bytes or the result of cls(response) + :return: AsyncIterator[bytes] or the result of cls(response) :rtype: AsyncIterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -856,22 +777,21 @@ async def get_remote_desktop_file( ) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -887,10 +807,6 @@ async def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -924,7 +840,6 @@ async def get( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -945,22 +860,21 @@ async def get( ) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -972,13 +886,9 @@ async def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -991,7 +901,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceRole] @@ -1016,17 +925,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1037,14 +945,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1054,11 +962,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1070,10 +978,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -1099,7 +1003,7 @@ async def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1130,7 +1034,7 @@ async def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1138,16 +1042,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1167,10 +1070,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload async def begin_create_or_update( self, @@ -1193,14 +1092,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1213,7 +1104,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1226,18 +1117,10 @@ async def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1250,7 +1133,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -1260,20 +1143,9 @@ async def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1308,7 +1180,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1318,23 +1190,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1365,7 +1235,7 @@ async def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1373,16 +1243,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1394,13 +1263,9 @@ async def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1423,14 +1288,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1443,7 +1300,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1455,18 +1312,10 @@ async def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1479,7 +1328,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -1488,20 +1337,10 @@ async def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -1536,7 +1375,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1546,17 +1385,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1577,21 +1414,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1601,11 +1437,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1617,14 +1449,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1653,7 +1477,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1662,17 +1486,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -1682,7 +1502,6 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -1703,21 +1522,20 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: ) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1729,13 +1547,9 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -1747,7 +1561,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1768,21 +1581,20 @@ async def get_instance_view( ) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1794,13 +1606,9 @@ async def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: @@ -1808,7 +1616,6 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudService] @@ -1833,15 +1640,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1852,14 +1658,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1869,11 +1675,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1885,8 +1691,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -1895,7 +1699,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudService] @@ -1920,16 +1723,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1940,14 +1742,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1957,11 +1759,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1973,10 +1775,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1996,21 +1794,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2020,11 +1817,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -2036,14 +1829,6 @@ async def begin_start( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2072,7 +1857,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2081,17 +1866,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2112,21 +1893,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2136,11 +1916,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -2153,14 +1929,6 @@ async def begin_power_off( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2189,7 +1957,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2198,23 +1966,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2245,7 +2009,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2253,16 +2017,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,11 +2035,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2299,14 +2058,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2317,7 +2068,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2329,18 +2080,10 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2351,7 +2094,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -2361,19 +2104,8 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2405,7 +2137,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2414,23 +2146,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2461,7 +2189,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2469,16 +2197,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2488,11 +2215,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -2516,14 +2239,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2534,7 +2249,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2547,18 +2262,10 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2569,7 +2276,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -2580,19 +2287,8 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2624,7 +2320,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2633,23 +2329,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2680,7 +2372,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2380,15 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2707,11 +2398,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_rebuild( @@ -2736,14 +2423,6 @@ async def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2754,7 +2433,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2768,18 +2447,10 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2790,7 +2461,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -2802,19 +2473,8 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2846,7 +2506,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2855,23 +2515,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2902,7 +2558,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2910,16 +2566,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2929,11 +2584,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -2956,14 +2607,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2974,7 +2617,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2986,18 +2629,10 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2643,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -3018,19 +2653,8 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3062,7 +2686,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3071,17 +2695,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3109,7 +2729,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3140,7 +2760,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3149,16 +2769,15 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3168,11 +2787,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_walk_update_domain( @@ -3200,14 +2815,6 @@ async def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3219,7 +2826,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3235,18 +2842,10 @@ async def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3258,7 +2857,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -3271,20 +2870,9 @@ async def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3317,7 +2905,7 @@ async def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3326,17 +2914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_update_domain( @@ -3354,7 +2938,6 @@ async def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -3375,22 +2958,21 @@ async def get_update_domain( ) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3402,13 +2984,9 @@ async def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -3420,7 +2998,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain] @@ -3445,17 +3022,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3466,14 +3042,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -3483,11 +3059,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3498,7 +3074,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/models/_models_py3.py index 4ad547918bc1..d4e8a175277a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/models/_models_py3.py @@ -108,7 +108,7 @@ class CloudService(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -359,7 +359,7 @@ def __init__(self, *, role_instance: Optional["_models.InstanceViewStatusesSumma class CloudServiceListResult(_serialization.Model): """CloudServiceListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2020_10_01_preview.models.CloudService] @@ -646,7 +646,7 @@ def __init__( class CloudServiceRoleListResult(_serialization.Model): """CloudServiceRoleListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceRole] @@ -1112,7 +1112,7 @@ def __init__( self.properties = properties -class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): +class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): # pylint: disable=name-too-long """Describes a cloud service IP Configuration. :ivar public_ip_address: @@ -1261,7 +1261,7 @@ def __init__( class RoleInstanceListResult(_serialization.Model): """RoleInstanceListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstance] @@ -1351,7 +1351,7 @@ def __init__( class RoleInstances(_serialization.Model): """Specifies a list of role instances from the cloud service. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar role_instances: List of cloud service role instance names. Value of '*' will signify all role instances of the cloud service. Required. @@ -1496,7 +1496,7 @@ def __init__(self, **kwargs: Any) -> None: class UpdateDomainListResult(_serialization.Model): """UpdateDomainListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/operations/_operations.py index 2db815326ec7..a25af517d046 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_10_01_preview/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, Iterator, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -44,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_cloud_service_role_instances_delete_request( +def build_cloud_service_role_instances_delete_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -76,7 +71,7 @@ def build_cloud_service_role_instances_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_request( +def build_cloud_service_role_instances_get_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, @@ -116,7 +111,7 @@ def build_cloud_service_role_instances_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_instance_view_request( +def build_cloud_service_role_instances_get_instance_view_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -148,7 +143,7 @@ def build_cloud_service_role_instances_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_list_request( +def build_cloud_service_role_instances_list_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, @@ -186,7 +181,7 @@ def build_cloud_service_role_instances_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_restart_request( +def build_cloud_service_role_instances_restart_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -218,7 +213,7 @@ def build_cloud_service_role_instances_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_reimage_request( +def build_cloud_service_role_instances_reimage_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -250,7 +245,7 @@ def build_cloud_service_role_instances_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_rebuild_request( +def build_cloud_service_role_instances_rebuild_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -282,7 +277,7 @@ def build_cloud_service_role_instances_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_remote_desktop_file_request( +def build_cloud_service_role_instances_get_remote_desktop_file_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -377,7 +372,7 @@ def build_cloud_service_roles_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_create_or_update_request( +def build_cloud_services_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -507,7 +502,7 @@ def build_cloud_services_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_get_instance_view_request( +def build_cloud_services_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -754,7 +749,7 @@ def build_cloud_services_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_delete_instances_request( +def build_cloud_services_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -788,7 +783,7 @@ def build_cloud_services_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_walk_update_domain_request( +def build_cloud_services_update_domain_walk_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +818,7 @@ def build_cloud_services_update_domain_walk_update_domain_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_get_update_domain_request( +def build_cloud_services_update_domain_get_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -855,7 +850,7 @@ def build_cloud_services_update_domain_get_update_domain_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_list_update_domains_request( +def build_cloud_services_update_domain_list_update_domains_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -925,22 +920,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -950,11 +944,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -968,14 +958,6 @@ def begin_delete( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1005,7 +987,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1014,17 +996,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -1047,7 +1025,6 @@ def get( :keyword expand: The expand expression to apply to the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -1068,23 +1045,22 @@ def get( ) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1096,13 +1072,9 @@ def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -1116,7 +1088,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1137,22 +1108,21 @@ def get_instance_view( ) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1164,13 +1134,9 @@ def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1192,7 +1158,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstance] @@ -1217,18 +1182,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1239,14 +1203,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -1256,11 +1220,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1272,10 +1236,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1295,22 +1255,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1320,11 +1279,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -1339,14 +1294,6 @@ def begin_restart( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1376,7 +1323,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1385,17 +1332,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1416,22 +1359,21 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1441,11 +1383,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage( @@ -1460,14 +1398,6 @@ def begin_reimage( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1497,7 +1427,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1506,17 +1436,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1537,22 +1463,21 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1562,11 +1487,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_rebuild( @@ -1582,14 +1503,6 @@ def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1619,7 +1532,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1628,17 +1541,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_remote_desktop_file( @@ -1652,8 +1561,7 @@ def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Iterator of the response bytes or the result of cls(response) + :return: Iterator[bytes] or the result of cls(response) :rtype: Iterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -1673,22 +1581,21 @@ def get_remote_desktop_file( ) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1704,10 +1611,6 @@ def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -1741,7 +1644,6 @@ def get( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -1762,22 +1664,21 @@ def get( ) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1789,13 +1690,9 @@ def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1808,7 +1705,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceRole] @@ -1833,17 +1729,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1854,14 +1749,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1871,11 +1766,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1887,10 +1782,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -1916,7 +1807,7 @@ def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1947,7 +1838,7 @@ def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1955,16 +1846,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1984,10 +1874,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload def begin_create_or_update( self, @@ -2010,14 +1896,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2030,7 +1908,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2043,18 +1921,10 @@ def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2067,7 +1937,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -2077,20 +1947,9 @@ def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2125,7 +1984,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2135,23 +1994,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -2182,7 +2039,7 @@ def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2190,16 +2047,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2211,13 +2067,9 @@ def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2240,14 +2092,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2260,7 +2104,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2272,18 +2116,10 @@ def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2296,7 +2132,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -2305,20 +2141,10 @@ def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceUpdate or + IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: @@ -2353,7 +2179,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2363,17 +2189,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2394,21 +2218,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2418,11 +2241,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2432,14 +2251,6 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2468,7 +2279,7 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2477,17 +2288,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -2497,7 +2304,6 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -2518,21 +2324,20 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) ) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2544,13 +2349,9 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -2562,7 +2363,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -2583,21 +2383,20 @@ def get_instance_view( ) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2609,13 +2408,9 @@ def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: @@ -2623,7 +2418,6 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudService] @@ -2648,15 +2442,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2667,14 +2460,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2684,11 +2477,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2700,8 +2493,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -2710,7 +2501,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.CloudService] @@ -2735,16 +2525,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2755,14 +2544,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2772,11 +2561,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2788,10 +2577,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -2811,21 +2596,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2835,11 +2619,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2849,14 +2629,6 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2885,7 +2657,7 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2894,17 +2666,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2925,21 +2693,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements ) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2949,11 +2716,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2964,14 +2727,6 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3000,7 +2755,7 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3009,23 +2764,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3056,7 +2807,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3064,16 +2815,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3083,11 +2833,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -3110,14 +2856,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3128,7 +2866,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3140,18 +2878,10 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3162,7 +2892,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -3172,19 +2902,8 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3216,7 +2935,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3225,23 +2944,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3272,7 +2987,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3280,16 +2995,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3299,11 +3013,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -3327,14 +3037,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3345,7 +3047,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3358,18 +3060,10 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3380,7 +3074,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -3391,19 +3085,8 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3435,7 +3118,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3444,23 +3127,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3491,7 +3170,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3499,16 +3178,15 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3518,11 +3196,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_rebuild( @@ -3547,14 +3221,6 @@ def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3565,7 +3231,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3579,18 +3245,10 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3601,7 +3259,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -3613,19 +3271,8 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3657,7 +3304,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3666,23 +3313,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3713,7 +3356,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3721,16 +3364,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3740,11 +3382,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -3767,14 +3405,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3785,7 +3415,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3797,18 +3427,10 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3819,7 +3441,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -3829,19 +3451,8 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3873,7 +3484,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3882,17 +3493,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3920,7 +3527,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3951,7 +3558,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3960,16 +3567,15 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3979,11 +3585,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_walk_update_domain( @@ -4011,14 +3613,6 @@ def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4030,7 +3624,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4046,18 +3640,10 @@ def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4069,7 +3655,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -4082,20 +3668,9 @@ def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4128,7 +3703,7 @@ def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4137,17 +3712,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_update_domain( @@ -4165,7 +3736,6 @@ def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -4186,22 +3756,21 @@ def get_update_domain( ) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4213,13 +3782,9 @@ def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -4231,7 +3796,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_10_01_preview.models.UpdateDomain] @@ -4256,17 +3820,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4277,14 +3840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -4294,11 +3857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4309,7 +3872,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_compute_management_client.py index 67d3aec312ad..b1f395044850 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -232,7 +252,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -252,7 +272,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_configuration.py index b6d40183034c..576357007ca4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_metadata.json index ef52e68b7775..36da439bc045 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_compute_management_client.py index d5a944ee21cc..6e71da3317fb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -154,7 +156,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -234,7 +254,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2020-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -254,7 +276,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_configuration.py index 4d263e9ccd98..d44c7062a565 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2020-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/operations/_operations.py index 4cd611012da8..deb5c60e8af0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, AsyncIterable, Callable, Dict, IO, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.async_paging import AsyncItemPaged, AsyncList @@ -223,10 +222,6 @@ build_virtual_machines_update_request, ) -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]] @@ -252,7 +247,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -277,7 +272,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -285,16 +280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -314,10 +308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -341,14 +331,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -359,7 +341,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -373,18 +355,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -392,7 +366,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -403,19 +377,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_12_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_12_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -446,7 +409,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -456,20 +419,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -494,7 +455,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -502,16 +463,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -531,10 +491,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -558,14 +514,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -576,7 +524,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -590,18 +538,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -609,7 +549,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -620,19 +560,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_12_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_12_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -663,7 +592,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -673,17 +602,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -695,7 +622,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -714,21 +640,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -740,13 +665,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -765,21 +686,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -789,11 +709,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -805,14 +721,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -839,7 +747,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -848,17 +756,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -866,7 +770,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -888,16 +791,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -908,14 +810,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -925,11 +827,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -941,15 +843,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -971,15 +868,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -990,14 +886,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1007,11 +903,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1023,13 +919,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -1055,7 +949,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1063,16 +957,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1086,13 +979,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -1118,14 +1007,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -1137,7 +1018,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1152,18 +1033,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -1175,7 +1048,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1187,19 +1060,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -1231,7 +1093,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1243,17 +1105,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1272,21 +1132,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1296,11 +1155,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1314,14 +1169,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1348,7 +1195,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1359,17 +1206,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1393,7 +1236,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1418,7 +1261,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1426,16 +1269,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1455,10 +1297,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1482,14 +1320,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1501,7 +1331,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1515,18 +1345,10 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1535,7 +1357,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1546,19 +1368,8 @@ async def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1590,7 +1401,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1600,20 +1411,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1638,7 +1451,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1646,16 +1459,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1675,10 +1487,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1703,14 +1511,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1722,7 +1522,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1737,18 +1537,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1757,7 +1549,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1768,19 +1564,8 @@ async def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -1812,7 +1597,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1822,17 +1607,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1844,7 +1627,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1863,21 +1645,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1889,13 +1670,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1914,21 +1691,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1938,11 +1714,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1954,14 +1726,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1988,7 +1752,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1997,17 +1761,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -2015,7 +1775,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -2038,16 +1797,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2058,14 +1816,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2075,11 +1833,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2091,15 +1849,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -2122,15 +1875,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2141,14 +1893,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2158,11 +1910,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2174,13 +1926,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2206,7 +1956,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2214,16 +1964,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2237,13 +1986,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2269,14 +2014,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -2288,7 +2025,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2303,18 +2040,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -2326,7 +2055,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2338,19 +2067,8 @@ async def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -2382,7 +2100,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2394,17 +2112,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2423,21 +2139,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2447,11 +2162,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2465,14 +2176,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2499,7 +2202,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2510,17 +2213,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2547,7 +2246,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2573,7 +2272,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2581,16 +2280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2610,10 +2308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2638,14 +2332,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2658,7 +2344,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2673,18 +2359,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2697,7 +2375,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2709,19 +2387,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2754,7 +2422,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2764,23 +2432,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2806,7 +2472,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2814,16 +2480,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2843,10 +2508,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2871,14 +2532,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2891,7 +2544,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2906,18 +2559,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2930,7 +2575,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2942,19 +2587,10 @@ async def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2987,7 +2623,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2997,17 +2633,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3021,7 +2655,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3040,21 +2673,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3066,13 +2698,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3091,21 +2719,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3115,11 +2742,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3133,14 +2756,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3167,7 +2782,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3176,17 +2791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3196,7 +2807,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -3219,16 +2829,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3239,14 +2848,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3256,11 +2865,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3272,15 +2881,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -3303,15 +2907,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3322,14 +2925,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3339,11 +2942,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3355,8 +2958,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3369,7 +2970,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3391,17 +2991,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3412,14 +3011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3429,11 +3028,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3445,10 +3044,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -3471,7 +3066,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3496,7 +3095,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3504,16 +3103,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3533,10 +3131,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3561,14 +3155,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3580,7 +3166,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3595,18 +3181,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3615,7 +3193,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3626,19 +3208,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3670,7 +3241,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3680,23 +3251,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3722,7 +3291,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3730,16 +3299,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3759,10 +3327,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3787,14 +3351,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3806,7 +3362,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3821,18 +3377,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3844,7 +3392,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3856,19 +3404,8 @@ async def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -3900,7 +3437,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3910,17 +3447,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3932,7 +3467,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3951,21 +3485,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3977,13 +3510,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4002,21 +3531,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4026,11 +3554,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4044,14 +3568,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4078,7 +3594,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4087,17 +3603,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -4105,7 +3617,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -4128,16 +3639,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4148,14 +3658,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4165,11 +3675,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4181,15 +3691,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -4212,15 +3717,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4231,14 +3735,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4248,11 +3752,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4264,8 +3768,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4278,7 +3780,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4297,21 +3798,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4323,20 +3823,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -4362,7 +3858,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -4371,16 +3867,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4400,12 +3895,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -4433,14 +3924,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4449,12 +3932,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4472,18 +3955,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4492,12 +3967,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -4513,20 +3988,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4560,7 +4024,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4570,17 +4034,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -4596,7 +4058,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -4615,22 +4076,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4642,15 +4102,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -4667,22 +4123,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4692,14 +4147,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -4712,14 +4163,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4747,7 +4190,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4756,17 +4199,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -4780,7 +4219,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4804,17 +4242,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4825,14 +4262,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -4842,11 +4279,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4858,10 +4295,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -4908,7 +4341,6 @@ async def get( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4927,23 +4359,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4955,13 +4386,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4979,7 +4406,6 @@ def list_by_restore_point( belongs. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskRestorePoint] @@ -5002,18 +4428,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5024,14 +4449,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -5041,11 +4466,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5057,17 +4482,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -5093,7 +4514,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -5103,16 +4524,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5126,13 +4546,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5168,14 +4584,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -5189,7 +4597,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5212,18 +4620,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -5237,7 +4637,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5257,19 +4657,8 @@ async def begin_grant_access( Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] @@ -5303,7 +4692,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5315,17 +4704,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5349,23 +4736,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5375,11 +4761,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5406,14 +4788,6 @@ async def begin_revoke_access( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5442,7 +4816,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5453,17 +4827,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class Operations: @@ -5490,7 +4860,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5514,14 +4883,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5532,14 +4900,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5549,11 +4917,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5565,8 +4933,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -5609,7 +4975,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5620,7 +4985,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5632,11 +4997,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5647,7 +5011,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -5657,12 +5021,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5690,7 +5050,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5698,16 +5058,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5719,13 +5078,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -5748,7 +5103,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5759,7 +5113,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5771,11 +5125,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5786,7 +5139,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -5796,12 +5149,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5829,7 +5178,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -5837,16 +5186,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5858,13 +5206,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -5876,7 +5220,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5895,21 +5238,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5919,11 +5261,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -5933,7 +5271,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -5952,21 +5289,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5978,13 +5314,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -5995,7 +5327,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet] @@ -6018,16 +5349,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6038,14 +5368,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -6055,11 +5385,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6071,17 +5401,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet] @@ -6104,16 +5429,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6124,14 +5448,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -6141,11 +5465,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6157,10 +5481,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -6172,7 +5492,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] @@ -6195,17 +5514,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6216,14 +5534,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6233,11 +5551,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6249,10 +5567,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -6296,7 +5610,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6307,7 +5620,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6320,11 +5633,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6335,7 +5647,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -6345,12 +5657,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6378,7 +5686,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6386,16 +5694,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6415,10 +5722,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -6441,7 +5744,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6452,7 +5754,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6465,11 +5767,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6480,7 +5781,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -6490,12 +5791,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6523,7 +5821,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6531,16 +5829,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6552,13 +5849,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -6570,7 +5863,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6589,21 +5881,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6613,11 +5904,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -6637,7 +5924,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6656,22 +5942,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6683,19 +5968,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6719,15 +5999,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6738,14 +6017,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6755,11 +6034,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6771,10 +6050,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -6783,7 +6058,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -6807,16 +6081,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6827,14 +6100,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -6844,11 +6117,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6860,10 +6133,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -6908,7 +6177,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6919,7 +6187,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6933,11 +6201,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6948,7 +6215,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -6960,12 +6227,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6993,7 +6256,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -7001,16 +6264,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7030,10 +6292,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -7055,7 +6313,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7066,7 +6323,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7078,11 +6335,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7093,7 +6349,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -7103,12 +6359,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7136,7 +6388,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -7144,16 +6396,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7165,13 +6416,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -7183,7 +6430,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7202,21 +6448,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7226,11 +6471,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -7251,7 +6492,6 @@ async def get( of instance view of the dedicated hosts under the dedicated host group. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7270,22 +6510,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7297,13 +6536,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -7314,7 +6549,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup] @@ -7337,16 +6571,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7357,14 +6590,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7374,11 +6607,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7390,16 +6623,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup] @@ -7422,15 +6650,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7441,14 +6668,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7458,11 +6685,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7474,8 +6701,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -7502,7 +6727,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7528,7 +6753,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7537,16 +6762,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7566,10 +6790,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -7594,14 +6814,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7615,7 +6827,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7629,18 +6841,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7654,7 +6858,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -7666,19 +6870,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7712,7 +6905,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7722,24 +6915,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7765,7 +6956,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7774,16 +6965,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7795,13 +6985,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -7827,14 +7013,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7848,7 +7026,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7862,18 +7040,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7887,7 +7057,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -7899,19 +7069,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -7945,7 +7104,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7955,17 +7114,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -7984,22 +7141,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8009,11 +7165,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8027,14 +7179,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8062,7 +7206,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8071,17 +7215,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8104,7 +7244,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -8123,23 +7262,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8151,13 +7289,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -8170,7 +7304,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -8193,17 +7326,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8214,14 +7346,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -8231,11 +7363,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8247,10 +7379,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -8277,7 +7405,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -8301,15 +7428,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8320,14 +7446,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -8337,11 +7463,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8353,8 +7479,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -8364,7 +7488,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -8388,16 +7511,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8408,14 +7530,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -8425,11 +7547,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8441,10 +7563,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -8466,7 +7584,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8477,7 +7594,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8489,11 +7606,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8504,7 +7620,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -8514,12 +7630,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8547,7 +7659,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -8555,16 +7667,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8584,10 +7695,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -8609,7 +7716,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8620,7 +7726,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8632,11 +7738,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8647,7 +7752,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -8657,12 +7762,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8690,7 +7792,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -8698,16 +7800,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8719,13 +7820,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -8737,7 +7834,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8756,21 +7852,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8780,11 +7875,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -8796,7 +7887,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8815,21 +7905,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8841,13 +7930,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -8861,7 +7946,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8880,21 +7964,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8906,13 +7989,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -8949,7 +8028,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8968,23 +8046,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8996,13 +8073,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -9014,7 +8087,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9033,21 +8105,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9059,13 +8130,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -9093,7 +8160,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9112,7 +8178,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -9121,16 +8187,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9142,13 +8207,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -9176,7 +8237,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9202,7 +8263,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9211,16 +8272,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9240,10 +8300,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9270,14 +8326,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9291,7 +8339,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9307,18 +8355,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9332,7 +8372,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -9345,20 +8385,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9392,7 +8421,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9402,24 +8431,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9445,7 +8472,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9454,16 +8481,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9475,13 +8501,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9510,14 +8532,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9531,7 +8545,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9547,18 +8561,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9572,7 +8578,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -9585,20 +8591,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9632,7 +8627,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9642,17 +8637,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -9671,22 +8664,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9696,11 +8688,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9715,14 +8703,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9750,7 +8730,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9759,17 +8739,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9791,7 +8767,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9810,23 +8785,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9838,13 +8812,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9858,7 +8828,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9877,22 +8846,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9904,13 +8872,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -9949,7 +8913,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9968,7 +8931,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -9976,16 +8939,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9997,13 +8959,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -10035,7 +8993,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10054,7 +9011,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -10064,16 +9021,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10085,13 +9041,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -10103,7 +9055,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10122,21 +9073,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10148,13 +9098,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -10162,7 +9108,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10181,20 +9126,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10206,13 +9150,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -10226,7 +9166,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10245,22 +9184,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10272,13 +9210,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -10319,7 +9253,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -10338,7 +9271,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -10347,16 +9280,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10368,13 +9300,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -10411,7 +9339,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10430,7 +9357,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -10441,16 +9368,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10462,13 +9388,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -10483,7 +9405,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10502,22 +9423,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10529,13 +9449,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -10547,7 +9463,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10566,21 +9481,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10592,13 +9506,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -10615,7 +9525,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10634,23 +9543,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10662,13 +9570,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -10698,7 +9602,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -10720,16 +9623,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10740,14 +9642,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -10757,11 +9659,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10773,8 +9675,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -10803,7 +9703,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -10826,16 +9725,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10846,14 +9744,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10863,11 +9761,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10879,15 +9777,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -10913,7 +9807,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10921,16 +9815,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10944,13 +9837,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -10974,14 +9863,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10994,7 +9875,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11007,18 +9888,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -11031,7 +9904,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -11042,19 +9915,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -11087,7 +9950,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11099,20 +9962,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -11137,7 +10002,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11145,16 +10010,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11174,10 +10038,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -11200,14 +10060,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11220,7 +10072,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11233,18 +10085,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11254,7 +10098,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -11264,19 +10112,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11309,7 +10146,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11319,20 +10156,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -11357,7 +10196,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11365,16 +10204,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11386,13 +10224,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -11415,14 +10249,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11435,7 +10261,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11447,18 +10273,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11468,7 +10286,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -11477,19 +10299,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -11522,7 +10333,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11532,17 +10343,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -11561,22 +10370,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11586,11 +10394,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -11605,14 +10409,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11640,7 +10436,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11649,17 +10445,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11674,7 +10466,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -11693,22 +10484,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11720,13 +10510,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -11738,7 +10524,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11757,21 +10542,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11783,13 +10567,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11808,21 +10588,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11832,11 +10611,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -11849,14 +10624,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11883,7 +10650,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11892,17 +10659,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11921,21 +10684,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11945,11 +10707,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11960,14 +10718,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11994,7 +10744,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12003,17 +10753,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -12030,7 +10776,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12049,21 +10794,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12073,11 +10817,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -12086,7 +10826,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -12109,16 +10848,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12129,14 +10867,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12146,11 +10884,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12162,10 +10900,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -12174,7 +10908,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -12197,16 +10930,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12217,14 +10949,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12234,11 +10966,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12250,8 +10982,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12263,7 +10993,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] @@ -12286,17 +11015,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12307,14 +11035,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12324,11 +11052,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12340,10 +11068,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -12361,22 +11085,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12386,11 +11109,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -12407,14 +11126,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12442,7 +11153,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12451,17 +11162,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12480,21 +11187,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12504,11 +11210,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12518,14 +11220,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12552,7 +11246,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12561,17 +11255,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12590,21 +11280,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12614,11 +11303,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12628,14 +11313,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12662,7 +11339,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12671,17 +11348,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12700,21 +11373,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12724,11 +11396,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12738,14 +11406,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12772,7 +11432,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12781,17 +11441,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12810,21 +11466,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12834,11 +11489,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12848,14 +11499,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12882,7 +11525,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12891,23 +11534,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12936,7 +11575,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12944,16 +11583,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12963,11 +11601,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -12991,14 +11625,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13009,7 +11635,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13022,18 +11648,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13044,7 +11662,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -13054,19 +11672,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13096,7 +11704,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13105,17 +11713,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -13137,7 +11741,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13156,22 +11759,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13183,13 +11785,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13208,21 +11806,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13232,11 +11829,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -13248,14 +11841,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13282,7 +11867,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13291,17 +11876,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -13313,7 +11894,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13332,21 +11912,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13356,11 +11935,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13379,21 +11954,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13407,13 +11981,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -13425,14 +11995,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -13462,7 +12024,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13474,23 +12036,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -13516,7 +12076,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -13524,16 +12084,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13547,13 +12106,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -13578,14 +12133,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13598,7 +12145,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13611,18 +12158,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13635,7 +12174,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -13645,20 +12184,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13691,7 +12219,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13703,20 +12231,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -13741,7 +12271,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -13749,16 +12279,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13772,13 +12301,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -13801,14 +12326,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13821,7 +12338,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13833,18 +12350,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13854,7 +12363,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -13863,19 +12376,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13908,7 +12410,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13920,17 +12422,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -13960,7 +12460,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13984,16 +12483,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14004,14 +12502,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -14021,11 +12519,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14037,15 +12535,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -14071,7 +12565,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14079,16 +12573,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14108,10 +12601,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -14133,14 +12622,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14153,7 +12634,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14165,18 +12646,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14189,7 +12662,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -14198,20 +12671,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14244,7 +12706,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14254,23 +12716,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -14296,7 +12756,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14304,16 +12764,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14325,13 +12784,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -14354,14 +12809,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14374,7 +12821,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14386,18 +12833,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14410,7 +12849,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -14419,20 +12858,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14465,7 +12894,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14475,17 +12904,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -14504,22 +12931,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14529,11 +12955,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14548,14 +12970,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14583,7 +12997,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14592,17 +13006,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14614,7 +13024,6 @@ async def get( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -14633,21 +13042,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14659,19 +13067,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14700,7 +13104,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14708,16 +13112,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14727,11 +13130,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -14758,14 +13157,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14776,7 +13167,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14791,18 +13182,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14813,7 +13196,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -14825,20 +13208,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14868,7 +13240,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14877,23 +13249,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -14921,7 +13289,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14930,16 +13298,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14949,11 +13316,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -14982,14 +13345,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15000,7 +13355,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -15014,21 +13369,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15039,7 +13386,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15051,23 +13398,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15098,7 +13434,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15107,17 +13443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -15129,7 +13461,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15148,21 +13479,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15174,13 +13504,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -15188,7 +13514,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15212,16 +13537,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15232,14 +13556,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -15249,11 +13573,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15265,17 +13589,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15299,15 +13618,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15318,14 +13636,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -15335,11 +13653,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15351,8 +13669,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -15364,7 +13680,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -15388,17 +13703,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15409,14 +13723,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -15426,11 +13740,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15442,10 +13756,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -15456,7 +13766,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -15480,17 +13789,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15501,14 +13809,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -15518,11 +13826,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15534,15 +13842,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -15573,7 +13877,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15582,16 +13886,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15601,11 +13904,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -15637,14 +13936,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15655,7 +13946,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -15671,7 +13962,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -15679,14 +13970,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15697,7 +13980,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -15711,24 +13994,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15759,7 +14031,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15768,23 +14040,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15813,7 +14081,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15821,16 +14089,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15840,11 +14107,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -15869,14 +14132,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15887,7 +14142,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15900,18 +14155,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15922,7 +14169,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -15932,20 +14179,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15975,7 +14211,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15984,23 +14220,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16029,7 +14261,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16037,16 +14269,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16056,11 +14287,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -16085,14 +14312,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16103,7 +14322,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16116,18 +14335,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16138,7 +14349,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -16148,20 +14359,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16191,7 +14391,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16200,23 +14400,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16245,7 +14441,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16253,16 +14449,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16272,11 +14467,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -16302,14 +14493,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16320,7 +14503,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16334,18 +14517,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16356,7 +14531,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -16367,20 +14542,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16410,7 +14574,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16419,23 +14583,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16464,7 +14624,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16472,16 +14632,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16491,11 +14650,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -16523,14 +14678,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16541,7 +14688,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16557,18 +14704,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16579,7 +14718,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -16592,20 +14731,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16635,7 +14763,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16644,23 +14772,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -16686,7 +14810,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16694,16 +14818,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16713,11 +14836,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -16742,14 +14861,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16760,7 +14871,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16773,18 +14884,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16795,7 +14898,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -16805,20 +14908,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16848,7 +14940,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16857,23 +14949,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16902,7 +14990,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16910,16 +14998,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16929,11 +15016,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -16959,14 +15042,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16977,7 +15052,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16991,18 +15066,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17013,7 +15080,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -17025,20 +15092,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17068,7 +15124,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17077,23 +15133,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17122,7 +15174,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17130,16 +15182,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17149,11 +15200,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -17179,14 +15226,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17197,7 +15236,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17211,18 +15250,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17233,7 +15264,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -17244,20 +15275,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17287,7 +15307,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17296,20 +15316,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -17322,7 +15338,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -17341,22 +15356,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17368,13 +15382,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -17399,7 +15409,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17410,7 +15419,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17423,11 +15432,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17438,7 +15446,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -17449,13 +15457,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2020_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2020_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17483,7 +15488,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17491,16 +15496,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17510,17 +15514,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -17546,7 +15546,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17554,16 +15554,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17573,11 +15572,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -17601,14 +15596,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17619,7 +15606,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17632,18 +15619,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17654,7 +15633,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -17665,19 +15644,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17707,7 +15676,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17716,17 +15685,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -17756,7 +15721,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] @@ -17779,16 +15743,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17799,14 +15762,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -17816,11 +15779,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17832,8 +15795,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -17856,7 +15817,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17881,7 +15842,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17889,16 +15850,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17918,10 +15878,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17943,14 +15899,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17961,7 +15909,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17973,18 +15921,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17992,7 +15932,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -18001,19 +15941,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18044,7 +15973,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18054,20 +15983,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -18092,7 +16023,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -18100,16 +16031,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18129,10 +16059,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -18154,14 +16080,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18172,7 +16090,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18184,18 +16102,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18203,7 +16113,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -18212,19 +16126,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18255,7 +16158,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18265,17 +16168,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -18294,21 +16195,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18318,11 +16218,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -18332,14 +16228,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18366,7 +16254,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18375,17 +16263,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18399,7 +16283,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -18418,22 +16301,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18445,13 +16327,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -18459,7 +16337,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18481,16 +16358,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18501,14 +16377,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -18518,11 +16394,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18534,16 +16410,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18565,15 +16436,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18584,14 +16454,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -18601,11 +16471,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18617,10 +16487,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18645,7 +16513,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -18671,7 +16539,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -18680,16 +16548,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18709,10 +16576,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -18740,14 +16603,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18761,7 +16616,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18777,18 +16632,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18802,7 +16649,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -18815,20 +16662,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18862,7 +16698,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18872,24 +16708,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -18915,7 +16749,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -18924,16 +16758,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18953,10 +16786,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -18984,14 +16813,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -19005,7 +16826,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19021,18 +16842,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -19046,7 +16859,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -19059,20 +16872,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -19106,7 +16909,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19116,17 +16919,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -19145,22 +16946,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19170,11 +16970,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -19189,14 +16985,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19224,7 +17012,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19233,17 +17021,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -19265,7 +17049,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -19284,23 +17067,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19312,13 +17094,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19330,7 +17108,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -19354,17 +17131,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19375,14 +17151,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -19392,11 +17168,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19408,12 +17184,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -19450,21 +17222,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19474,11 +17245,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -19490,14 +17257,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19524,7 +17283,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19533,17 +17292,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19562,21 +17317,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19586,11 +17340,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -19604,14 +17354,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19638,7 +17380,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19647,17 +17389,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19676,21 +17414,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19700,11 +17437,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -19718,14 +17451,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19752,7 +17477,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19761,17 +17486,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -19783,7 +17504,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -19802,21 +17522,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19828,16 +17547,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -19863,7 +17578,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -19889,7 +17604,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19899,16 +17614,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19928,10 +17642,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -19961,14 +17671,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19983,7 +17685,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20000,18 +17702,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -20026,7 +17720,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -20040,20 +17734,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -20088,7 +17771,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20098,17 +17781,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -20116,7 +17797,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -20142,7 +17823,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20152,16 +17833,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20173,13 +17853,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20210,14 +17886,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -20232,7 +17900,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20249,18 +17917,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -20275,7 +17935,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -20289,20 +17949,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -20337,7 +17987,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20347,17 +17997,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -20376,23 +18024,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20402,11 +18049,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20422,14 +18065,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20458,7 +18093,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20467,17 +18102,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20502,7 +18133,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -20521,7 +18151,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20529,16 +18159,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20550,13 +18179,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -20578,7 +18203,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -20597,23 +18221,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20625,13 +18248,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -20659,7 +18278,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -20688,7 +18309,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20697,16 +18318,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20716,11 +18336,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -20748,14 +18364,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20767,7 +18375,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20782,18 +18390,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20805,7 +18405,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -20817,20 +18419,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20861,7 +18453,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20870,17 +18462,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20899,22 +18487,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20924,11 +18511,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -20943,14 +18526,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20978,7 +18553,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20987,17 +18562,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21016,22 +18587,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21041,11 +18611,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -21061,14 +18627,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21096,7 +18654,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21105,24 +18663,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -21148,7 +18702,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21157,16 +18711,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21186,10 +18739,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -21216,14 +18765,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -21237,7 +18778,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21253,18 +18794,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -21278,7 +18811,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -21291,19 +18824,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -21337,7 +18859,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21347,17 +18869,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -21382,23 +18902,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21408,11 +18927,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21435,14 +18950,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21471,7 +18978,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21480,17 +18987,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -21513,7 +19016,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -21532,23 +19034,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21560,13 +19061,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -21580,7 +19077,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -21599,22 +19095,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21626,13 +19121,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -21661,7 +19152,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -21685,7 +19175,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -21693,12 +19183,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21709,14 +19198,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -21726,11 +19215,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21742,10 +19231,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -21769,23 +19254,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21795,11 +19279,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -21825,14 +19305,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21861,7 +19333,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21870,17 +19342,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21899,22 +19367,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21924,11 +19391,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -21942,14 +19405,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21977,7 +19432,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21986,17 +19441,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -22015,22 +19466,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22040,11 +19490,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -22058,14 +19504,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22093,7 +19531,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22102,17 +19540,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -22131,22 +19565,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22156,11 +19589,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -22175,14 +19604,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22210,7 +19631,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22219,17 +19640,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -22255,7 +19672,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22274,23 +19690,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22302,13 +19717,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -22327,22 +19738,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22352,11 +19762,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -22370,14 +19776,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22405,7 +19803,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22414,17 +19812,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -22438,7 +19832,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22457,22 +19850,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22482,18 +19874,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -22519,7 +19907,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22528,16 +19916,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22551,13 +19938,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -22583,14 +19966,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22604,7 +19979,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22618,18 +19993,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22643,7 +20010,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -22655,19 +20022,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22701,7 +20057,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22713,17 +20069,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -22747,7 +20101,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -22772,23 +20126,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22802,13 +20155,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -22830,14 +20179,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22847,7 +20188,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -22856,18 +20197,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22877,7 +20210,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -22885,19 +20218,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22929,7 +20252,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22942,20 +20265,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -22980,23 +20301,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23010,13 +20330,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -23037,14 +20353,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -23054,7 +20362,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -23062,18 +20370,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -23083,7 +20383,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -23091,19 +20391,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -23135,7 +20424,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23148,17 +20437,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -23187,7 +20474,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -23211,16 +20497,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23231,14 +20516,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -23248,11 +20533,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23264,10 +20549,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -23276,7 +20557,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -23295,21 +20575,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23321,20 +20600,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23360,7 +20635,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -23369,16 +20644,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23398,10 +20672,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -23428,14 +20698,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23449,7 +20711,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23465,18 +20727,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23490,7 +20744,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -23503,19 +20757,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23549,7 +20792,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23559,24 +20802,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23602,7 +20843,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -23611,16 +20852,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23632,13 +20872,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -23666,14 +20902,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23687,7 +20915,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23703,18 +20931,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23728,7 +20948,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -23741,19 +20961,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23787,7 +20997,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23797,17 +21007,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -23826,22 +21034,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23851,11 +21058,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -23870,14 +21073,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23905,7 +21100,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23914,17 +21109,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -23946,7 +21137,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -23965,23 +21155,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23993,13 +21182,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -24013,7 +21198,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24037,18 +21221,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24059,14 +21242,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -24076,11 +21259,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24092,12 +21275,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -24123,7 +21302,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -24149,7 +21328,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24159,16 +21338,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24188,10 +21366,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -24220,14 +21394,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24242,7 +21408,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24259,18 +21425,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24285,7 +21443,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -24299,19 +21457,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24346,7 +21493,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24356,17 +21503,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -24374,7 +21519,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -24400,7 +21545,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24410,16 +21555,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24431,13 +21575,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -24467,14 +21607,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24489,7 +21621,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24506,18 +21638,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24532,7 +21656,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -24546,19 +21670,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24593,7 +21707,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24603,17 +21717,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -24632,23 +21744,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24658,11 +21769,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -24678,14 +21785,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24714,7 +21813,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -24723,17 +21822,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -24758,7 +21853,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -24777,7 +21871,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24785,16 +21879,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24806,13 +21899,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24834,7 +21923,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24858,19 +21946,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24881,14 +21968,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -24898,11 +21985,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24913,7 +22000,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_compute_management_client_enums.py index 5a1f96d7b038..fa6ef3131ce4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_compute_management_client_enums.py @@ -80,21 +80,18 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + the disk.""" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -142,10 +139,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -153,7 +150,7 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -182,16 +179,16 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -199,13 +196,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExecutionState(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_models_py3.py index 6c3d021994c0..bf6535020929 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -253,7 +249,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -278,7 +274,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -293,7 +289,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -359,7 +355,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -417,7 +413,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -516,7 +512,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet] @@ -900,11 +896,10 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :vartype create_option: str or ~azure.mgmt.compute.v2020_12_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -965,8 +960,7 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and - "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", and "Upload". :paramtype create_option: str or ~azure.mgmt.compute.v2020_12_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -1008,7 +1002,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1204,7 +1198,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1381,7 +1375,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1497,7 +1491,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup] @@ -1701,7 +1695,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -1924,7 +1918,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2226,7 +2220,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2288,7 +2282,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] @@ -2344,7 +2338,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2447,7 +2441,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -2510,17 +2504,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2673,7 +2656,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.Disk] @@ -2831,7 +2814,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.DiskRestorePoint] @@ -3203,7 +3186,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -3327,7 +3310,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2020_12_01.models.AccessLevel @@ -3476,7 +3459,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3656,7 +3639,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2020_12_01.models.SubResource @@ -3764,7 +3747,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -3802,7 +3785,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.Image] @@ -3836,7 +3819,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2020_12_01.models.SubResource @@ -4035,12 +4018,12 @@ class ImageStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_12_01.models.ImageOSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.ImageDataDisk] :ivar zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4065,12 +4048,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_12_01.models.ImageOSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.ImageDataDisk] :keyword zone_resilient: Specifies whether an image is zone resilient or not. Default is false. Zone resilient images can be created only in regions that provide Zone Redundant Storage (ZRS). @@ -4231,7 +4214,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -4270,7 +4253,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_12_01.models.SourceVault @@ -4303,7 +4286,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2020_12_01.models.SourceVault @@ -4336,7 +4319,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -4369,7 +4352,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4634,7 +4617,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.Usage] @@ -4668,7 +4651,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4981,7 +4964,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2020_12_01.models.OrchestrationServiceNames @@ -5055,7 +5038,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -5198,7 +5181,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5232,7 +5215,7 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. :code:`
    `:code:`
    ` @@ -5244,10 +5227,10 @@ class OSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5260,10 +5243,10 @@ class OSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5274,7 +5257,7 @@ class OSProfile(_serialization.Model): `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5282,10 +5265,10 @@ class OSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2020_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machine. :vartype secrets: list[~azure.mgmt.compute.v2020_12_01.models.VaultSecretGroup] @@ -5331,7 +5314,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword admin_username: Specifies the name of the administrator account. :code:`
    `:code:`
    ` This property cannot be updated after the VM is created. @@ -5344,10 +5327,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -5360,10 +5343,10 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum @@ -5374,7 +5357,7 @@ def __init__( `_ :code:`
    `:code:`
    ` For using cloud-init for your Linux VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -5382,10 +5365,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2020_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machine. @@ -5832,7 +5815,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5926,7 +5909,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup] @@ -5964,22 +5947,11 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -6029,7 +6001,7 @@ def __init__( class PurchasePlanAutoGenerated(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -6100,7 +6072,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6196,7 +6168,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -6471,7 +6443,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6534,7 +6506,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6597,7 +6569,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6671,7 +6643,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6718,7 +6690,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6751,7 +6723,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.RunCommandDocumentBase] @@ -6787,7 +6759,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7051,7 +7023,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7255,7 +7227,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.Snapshot] @@ -7478,7 +7450,7 @@ class SshPublicKey(_serialization.Model): :ivar key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :vartype key_data: str """ @@ -7496,7 +7468,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None :keyword key_data: SSH public key certificate used to authenticate with the VM through ssh. The key needs to be at least 2048-bit and in ssh-rsa format. :code:`
    `:code:`
    ` For creating ssh keys, see `Create SSH keys on Linux and Mac for Linux VMs in Azure - `_. + `_. # pylint: disable=line-too-long :paramtype key_data: str """ super().__init__(**kwargs) @@ -7507,7 +7479,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7517,7 +7489,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7546,7 +7518,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7561,7 +7533,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7617,7 +7589,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource] @@ -7694,12 +7666,12 @@ class StorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_12_01.models.OSDisk :ivar data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.DataDisk] """ @@ -7726,12 +7698,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_12_01.models.OSDisk :keyword data_disks: Specifies the parameters that are used to add a data disk to a virtual machine. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.DataDisk] """ super().__init__(**kwargs) @@ -7833,7 +7805,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7854,67 +7826,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7983,7 +7894,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8126,7 +8037,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8336,7 +8247,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8384,10 +8295,10 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -8574,10 +8485,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -8784,7 +8695,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8874,7 +8785,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9001,7 +8912,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -9045,7 +8956,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9335,7 +9246,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2020_12_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_12_01.models.UserAssignedIdentitiesValue] """ @@ -9367,7 +9278,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2020_12_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_12_01.models.UserAssignedIdentitiesValue] """ @@ -9381,7 +9292,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9444,7 +9355,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9591,7 +9502,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). Required. @@ -9891,7 +9802,7 @@ def __init__( class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -10005,7 +9916,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10252,7 +10163,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand] @@ -10401,7 +10312,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10633,7 +10544,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10867,10 +10778,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension] @@ -11081,7 +10992,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2020_12_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -11120,7 +11031,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -11131,7 +11042,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -11202,7 +11113,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11229,7 +11140,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11384,10 +11295,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -11429,7 +11340,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet] @@ -11465,7 +11376,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetSku] @@ -11501,7 +11412,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet] @@ -11534,7 +11445,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -11578,10 +11489,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11675,7 +11586,7 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -11700,7 +11611,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2020_12_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -11725,7 +11636,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2020_12_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11739,7 +11650,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11879,10 +11790,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :vartype admin_username: str :ivar admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -11895,16 +11806,16 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :vartype admin_password: str :ivar custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :vartype custom_data: str :ivar windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -11912,10 +11823,10 @@ class VirtualMachineScaleSetOSProfile(_serialization.Model): :ivar linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :vartype linux_configuration: ~azure.mgmt.compute.v2020_12_01.models.LinuxConfiguration :ivar secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -11958,10 +11869,10 @@ def __init__( **Max-length (Linux):** 64 characters :code:`
    `:code:`
    ` **Max-length (Windows):** 20 characters :code:`
    `:code:`
    `:code:`
  • ` For root access to the Linux VM, see `Using root privileges on Linux virtual machines in Azure - `_\ + `_\ # pylint: disable=line-too-long :code:`
    `:code:`
  • ` For a list of built-in system users on Linux that should not be used in this field, see `Selecting User Names for Linux on Azure - `_. + `_. # pylint: disable=line-too-long :paramtype admin_username: str :keyword admin_password: Specifies the password of the administrator account. :code:`
    `:code:`
    ` **Minimum-length (Windows):** 8 characters :code:`
    `:code:`
    ` @@ -11974,16 +11885,16 @@ def __init__( "pass@word1", "Password!", "Password1", "Password22", "iloveyou!" :code:`
    `:code:`
    ` For resetting the password, see `How to reset the Remote Desktop service or its login password in a Windows VM - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For resetting root password, see `Manage users, SSH, and check or repair disks on Azure Linux VMs using the VMAccess Extension - `_. + `_. # pylint: disable=line-too-long :paramtype admin_password: str :keyword custom_data: Specifies a base-64 encoded string of custom data. The base-64 encoded string is decoded to a binary array that is saved as a file on the Virtual Machine. The maximum length of the binary array is 65535 bytes. :code:`
    `:code:`
    ` For using cloud-init for your VM, see `Using cloud-init to customize a Linux VM during creation - `_. + `_. # pylint: disable=line-too-long :paramtype custom_data: str :keyword windows_configuration: Specifies Windows operating system settings on the virtual machine. @@ -11991,10 +11902,10 @@ def __init__( :keyword linux_configuration: Specifies the Linux operating system settings on the virtual machine. :code:`
    `:code:`
    `For a list of supported Linux distributions, see `Linux on Azure-Endorsed Distributions - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` For running non-endorsed distributions, see `Information for Non-Endorsed Distributions - `_. + `_. # pylint: disable=line-too-long :paramtype linux_configuration: ~azure.mgmt.compute.v2020_12_01.models.LinuxConfiguration :keyword secrets: Specifies set of certificates that should be installed onto the virtual machines in the scale set. @@ -12010,10 +11921,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12085,10 +11996,12 @@ def __init__( self.public_ip_address_version = public_ip_address_version -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12115,7 +12028,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12123,18 +12036,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -12253,12 +12154,12 @@ class VirtualMachineScaleSetStorageProfile(_serialization.Model): :ivar os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype os_disk: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetOSDisk :ivar data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :vartype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -12286,12 +12187,12 @@ def __init__( :keyword os_disk: Specifies information about the operating system disk used by the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype os_disk: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetOSDisk :keyword data_disks: Specifies the parameters that are used to add data disks to the virtual machines in the scale set. :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines - `_. + `_. # pylint: disable=line-too-long :paramtype data_disks: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetDataDisk] """ @@ -12443,7 +12344,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -12561,7 +12462,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -12653,12 +12554,12 @@ def __init__( self.enable_ip_forwarding = enable_ip_forwarding -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2020_12_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12685,7 +12586,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2020_12_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12815,7 +12716,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -12859,7 +12760,7 @@ def __init__( self.dns_settings = dns_settings -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -12998,7 +12899,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13058,10 +12959,10 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :vartype availability_set: ~azure.mgmt.compute.v2020_12_01.models.SubResource @@ -13191,10 +13092,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. An existing VM cannot be added to an availability set. :paramtype availability_set: ~azure.mgmt.compute.v2020_12_01.models.SubResource @@ -13356,7 +13257,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -13379,7 +13280,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -13537,10 +13438,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -13684,7 +13585,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM] @@ -13717,7 +13618,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -14155,10 +14056,10 @@ class VirtualMachineUpdate(UpdateResource): # pylint: disable=too-many-instance machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -14328,10 +14229,10 @@ def __init__( # pylint: disable=too-many-locals machine should be assigned to. Virtual machines specified in the same availability set are allocated to different nodes to maximize availability. For more information about availability sets, see `Manage the availability of virtual machines - `_. + `_. # pylint: disable=line-too-long :code:`
    `:code:`
    ` For more information on Azure planned maintenance, see `Planned maintenance for virtual machines in Azure - `_ + `_ # pylint: disable=line-too-long :code:`
    `:code:`
    ` Currently, a VM can only be added to availability set at creation time. The availability set to which the VM is being added should be under the same resource group as the availability set resource. An existing VM cannot be added to an availability set. @@ -14424,7 +14325,7 @@ def __init__( # pylint: disable=too-many-locals self.platform_fault_domain = platform_fault_domain -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/operations/_operations.py index 3f19ab08117c..89dd9cfb6387 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2020_12_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -7,8 +7,7 @@ # 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, List, Optional, TypeVar, Union, cast, overload +from typing import Any, Callable, Dict, IO, Iterable, List, Literal, Optional, TypeVar, Union, cast, overload import urllib.parse from azure.core.exceptions import ( @@ -33,10 +32,6 @@ from ..._serialization import Serializer from .._vendor import _convert_request -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports T = TypeVar("T") ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]] @@ -168,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -405,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -518,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -552,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -617,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -648,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -702,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -733,7 +728,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -863,7 +858,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -917,7 +912,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -948,7 +943,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -989,7 +984,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1027,7 +1022,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1065,7 +1060,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1136,7 +1131,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1174,7 +1169,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1217,7 +1212,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1276,7 +1271,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1400,7 +1395,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1456,7 +1451,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1487,7 +1482,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1523,7 +1518,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1559,7 +1554,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1586,7 +1581,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -1628,7 +1623,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1654,7 +1651,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1684,7 +1681,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1718,7 +1715,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1752,7 +1749,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1815,7 +1812,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1845,7 +1842,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1869,7 +1868,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2005,7 +2004,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2036,7 +2035,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2060,7 +2061,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2214,7 +2215,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2245,7 +2246,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2278,7 +2279,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2309,7 +2310,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -2355,7 +2356,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2390,7 +2391,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2425,7 +2426,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2451,7 +2452,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2491,7 +2492,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2558,7 +2559,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -2606,7 +2607,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2637,7 +2638,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2666,7 +2667,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2698,7 +2699,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -2740,7 +2741,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -2790,7 +2791,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2822,7 +2823,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2853,7 +2854,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2913,7 +2914,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2975,7 +2978,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3113,7 +3116,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3144,7 +3147,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3169,7 +3172,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3194,7 +3197,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3275,7 +3278,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3470,7 +3473,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -3510,7 +3513,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3535,7 +3538,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3560,7 +3563,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3591,7 +3594,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3625,7 +3628,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3659,7 +3662,7 @@ def build_virtual_machines_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3689,7 +3692,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3723,7 +3726,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3757,7 +3760,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3789,7 +3792,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3820,7 +3823,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3851,7 +3854,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3889,7 +3892,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3920,7 +3923,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3950,7 +3953,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3976,7 +3981,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4007,7 +4012,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4038,7 +4043,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -4076,7 +4081,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4107,7 +4112,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4138,7 +4143,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4169,7 +4174,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4200,7 +4205,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4231,7 +4236,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4262,7 +4267,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4293,7 +4298,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -4330,7 +4335,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4361,7 +4366,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4545,7 +4550,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4599,7 +4604,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4634,7 +4639,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4669,7 +4674,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4695,7 +4700,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -4735,7 +4740,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4766,7 +4771,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4791,7 +4796,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4816,7 +4821,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4841,7 +4846,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4872,7 +4877,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4913,7 +4918,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4954,7 +4959,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4992,7 +4997,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5034,7 +5039,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5074,7 +5079,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5106,7 +5111,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5132,7 +5137,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5158,7 +5163,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5193,7 +5198,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5227,7 +5232,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5267,7 +5272,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5299,7 +5304,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -5345,7 +5350,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5379,7 +5384,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5405,7 +5410,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5431,7 +5436,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5457,7 +5462,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5499,7 +5504,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5525,7 +5530,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5551,7 +5556,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5586,7 +5591,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5619,7 +5624,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5652,7 +5657,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5679,7 +5686,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5710,7 +5717,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5745,7 +5752,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5780,7 +5787,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5812,7 +5819,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5852,7 +5859,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5885,7 +5892,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5926,7 +5933,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5967,7 +5974,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6005,7 +6012,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6047,7 +6054,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6108,7 +6115,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -6133,7 +6140,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -6141,16 +6148,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6170,10 +6176,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -6197,14 +6199,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6215,7 +6209,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6229,18 +6223,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6248,7 +6234,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -6259,19 +6245,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_12_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_12_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6302,7 +6277,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6312,20 +6287,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -6350,7 +6323,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -6358,16 +6331,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6387,10 +6359,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -6414,14 +6382,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6432,7 +6392,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6446,18 +6406,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6465,7 +6417,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -6476,19 +6428,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2020_12_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2020_12_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6519,7 +6460,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6529,17 +6470,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -6551,7 +6490,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -6570,21 +6508,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6596,13 +6533,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -6621,21 +6554,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6645,11 +6577,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6661,14 +6589,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6695,7 +6615,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6704,17 +6624,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -6722,7 +6638,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6744,16 +6659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6764,14 +6678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -6781,11 +6695,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6797,15 +6711,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -6827,15 +6736,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6846,14 +6754,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -6863,11 +6771,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6879,13 +6787,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6911,7 +6817,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -6919,16 +6825,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6942,13 +6847,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6974,14 +6875,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6992,7 +6885,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7007,18 +6900,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -7029,7 +6914,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -7041,19 +6926,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -7084,7 +6958,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7096,17 +6970,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -7125,21 +6997,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7149,11 +7020,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7165,14 +7032,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7199,7 +7058,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -7210,17 +7069,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -7244,7 +7099,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -7269,7 +7124,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -7277,16 +7132,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7306,10 +7160,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -7333,14 +7183,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7351,7 +7193,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7365,18 +7207,10 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7384,7 +7218,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -7395,19 +7229,8 @@ def begin_create_or_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7438,7 +7261,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7448,20 +7271,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -7486,7 +7311,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -7494,16 +7319,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7523,10 +7347,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -7551,14 +7371,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7569,7 +7381,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7584,18 +7396,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7603,7 +7407,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -7614,19 +7422,8 @@ def begin_update( max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2020_12_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7657,7 +7454,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7667,17 +7464,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -7689,7 +7484,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -7708,21 +7502,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7734,13 +7527,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -7759,21 +7548,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7783,11 +7571,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7799,14 +7583,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7833,7 +7609,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7842,17 +7618,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -7860,7 +7632,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7882,16 +7653,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7902,14 +7672,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -7919,11 +7689,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7935,15 +7705,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -7965,15 +7730,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7984,14 +7748,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -8001,11 +7765,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8017,13 +7781,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -8049,7 +7811,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -8057,16 +7819,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8080,13 +7841,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -8112,14 +7869,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -8130,7 +7879,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8145,18 +7894,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -8167,7 +7908,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -8179,19 +7920,8 @@ def begin_grant_access( max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -8222,7 +7952,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8234,17 +7964,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -8263,21 +7991,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8287,11 +8014,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8303,14 +8026,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8337,7 +8052,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -8348,17 +8063,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -8385,7 +8096,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -8411,7 +8122,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -8419,16 +8130,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8448,10 +8158,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -8476,14 +8182,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8495,7 +8193,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8510,18 +8208,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8533,7 +8223,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -8545,19 +8235,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8589,7 +8269,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8599,23 +8279,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -8641,7 +8319,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -8649,16 +8327,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8678,10 +8355,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -8706,14 +8379,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8725,7 +8390,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8740,18 +8405,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8763,7 +8420,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -8775,19 +8432,10 @@ def begin_update( name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] @@ -8819,7 +8467,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8829,17 +8477,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -8851,7 +8497,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -8870,21 +8515,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8896,13 +8540,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -8921,21 +8561,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8945,11 +8584,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8961,14 +8596,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8995,7 +8622,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9004,17 +8631,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -9022,7 +8645,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -9044,16 +8666,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9064,14 +8685,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -9081,11 +8702,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9097,15 +8718,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -9127,15 +8743,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9146,14 +8761,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -9163,11 +8778,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9179,8 +8794,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -9193,7 +8806,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -9215,17 +8827,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9236,14 +8847,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -9253,11 +8864,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9269,10 +8880,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -9295,7 +8902,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -9320,7 +8931,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -9328,16 +8939,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9357,10 +8967,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -9385,14 +8991,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9403,7 +9001,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9418,18 +9016,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9437,7 +9027,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -9448,19 +9042,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9491,7 +9074,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9501,23 +9084,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -9543,7 +9124,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -9551,16 +9132,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9580,10 +9160,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -9608,14 +9184,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9626,7 +9194,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9641,18 +9209,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9663,7 +9223,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -9675,19 +9235,8 @@ def begin_update( are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2020_12_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9718,7 +9267,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9728,17 +9277,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -9750,7 +9297,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -9769,21 +9315,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9795,13 +9340,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -9820,21 +9361,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9844,11 +9384,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -9860,14 +9396,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9894,7 +9422,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9903,17 +9431,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -9921,7 +9445,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -9943,16 +9466,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9963,14 +9485,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -9980,11 +9502,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9996,15 +9518,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -10026,15 +9543,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10045,14 +9561,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -10062,11 +9578,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10078,8 +9594,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -10092,7 +9606,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -10111,21 +9624,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10137,20 +9649,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -10176,7 +9684,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -10185,16 +9693,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10214,12 +9721,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -10247,14 +9750,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -10263,12 +9758,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10286,18 +9781,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -10306,12 +9793,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -10327,20 +9814,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -10374,7 +9850,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10384,17 +9860,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -10410,7 +9884,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -10429,22 +9902,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10456,15 +9928,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -10481,22 +9949,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10506,14 +9973,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -10526,14 +9989,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10561,7 +10016,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10570,17 +10025,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -10594,7 +10045,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -10618,17 +10068,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10639,14 +10088,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -10656,11 +10105,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10672,10 +10121,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -10722,7 +10167,6 @@ def get( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -10741,23 +10185,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10769,13 +10212,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -10793,7 +10232,6 @@ def list_by_restore_point( belongs. Supported characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -10815,18 +10253,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10837,14 +10274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -10854,11 +10291,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10870,17 +10307,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -10906,7 +10339,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -10916,16 +10349,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10939,13 +10371,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -10981,14 +10409,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11001,7 +10421,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11024,18 +10444,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11048,7 +10460,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -11068,19 +10480,8 @@ def begin_grant_access( Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2020_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -11113,7 +10514,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11125,17 +10526,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -11159,23 +10558,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11185,11 +10583,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -11216,14 +10610,6 @@ def begin_revoke_access( characters for the name are a-z, A-Z, 0-9 and _. The maximum name length is 80 characters. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11252,7 +10638,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -11263,17 +10649,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class Operations: @@ -11300,7 +10682,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -11324,14 +10705,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11342,14 +10722,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -11359,11 +10739,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11375,8 +10755,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -11419,7 +10797,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11430,7 +10807,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11442,11 +10819,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11457,7 +10833,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -11467,12 +10843,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11500,7 +10872,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -11508,16 +10880,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11529,13 +10900,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -11558,7 +10925,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11569,7 +10935,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11581,11 +10947,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11596,7 +10961,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -11606,12 +10971,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11639,7 +11000,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -11647,16 +11008,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11668,13 +11028,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -11686,7 +11042,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11705,21 +11060,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11729,11 +11083,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -11743,7 +11093,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -11762,21 +11111,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11788,13 +11136,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -11805,7 +11149,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -11827,16 +11170,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11847,14 +11189,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -11864,11 +11206,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11880,17 +11222,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -11912,16 +11249,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11932,14 +11268,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -11949,11 +11285,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11965,10 +11301,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -11980,7 +11312,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -12002,17 +11333,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12023,14 +11353,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12040,11 +11370,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12056,10 +11386,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -12103,7 +11429,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12114,7 +11439,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12127,11 +11452,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12142,7 +11466,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -12152,12 +11476,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12185,7 +11505,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -12193,16 +11513,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12222,10 +11541,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -12248,7 +11563,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12259,7 +11573,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12272,11 +11586,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12287,7 +11600,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -12297,12 +11610,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12330,7 +11640,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -12338,16 +11648,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12359,13 +11668,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -12377,7 +11682,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12396,21 +11700,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12420,11 +11723,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -12444,7 +11743,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12463,22 +11761,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12490,19 +11787,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -12526,15 +11818,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12545,14 +11836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -12562,11 +11853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12578,10 +11869,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -12590,7 +11877,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -12614,16 +11900,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12634,14 +11919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -12651,11 +11936,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12667,10 +11952,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -12715,7 +11996,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12726,7 +12006,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12740,11 +12020,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12755,7 +12034,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -12767,12 +12046,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12800,7 +12075,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -12808,16 +12083,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12837,10 +12111,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -12862,7 +12132,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12873,7 +12142,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12885,11 +12154,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12900,7 +12168,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -12910,12 +12178,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -12943,7 +12207,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -12951,16 +12215,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12972,13 +12235,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -12990,7 +12249,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13009,21 +12267,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13033,11 +12290,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -13058,7 +12311,6 @@ def get( of instance view of the dedicated hosts under the dedicated host group. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -13077,22 +12329,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13104,13 +12355,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -13119,7 +12366,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -13141,16 +12387,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13161,14 +12406,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -13178,11 +12423,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13194,16 +12439,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -13225,15 +12465,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13244,14 +12483,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -13261,11 +12500,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13277,8 +12516,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -13305,7 +12542,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -13331,7 +12568,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -13340,16 +12577,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13369,10 +12605,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -13397,14 +12629,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13417,7 +12641,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13431,18 +12655,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13455,7 +12671,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -13467,19 +12683,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13512,7 +12717,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13522,24 +12727,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -13565,7 +12768,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -13574,16 +12777,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13595,13 +12797,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -13627,14 +12825,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13647,7 +12837,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13661,18 +12851,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13685,7 +12867,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -13697,19 +12879,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] @@ -13742,7 +12913,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13752,17 +12923,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -13781,22 +12950,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13806,11 +12974,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13824,14 +12988,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13859,7 +13015,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13868,17 +13024,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13901,7 +13053,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -13920,23 +13071,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13948,13 +13098,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -13967,7 +13113,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -13989,17 +13134,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14010,14 +13154,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -14027,11 +13171,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14043,10 +13187,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -14073,7 +13213,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -14097,15 +13236,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14116,14 +13254,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -14133,11 +13271,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14149,8 +13287,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14160,7 +13296,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -14184,16 +13319,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14204,14 +13338,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -14221,11 +13355,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14237,10 +13371,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -14262,7 +13392,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14273,7 +13402,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14285,11 +13414,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14300,7 +13428,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -14310,12 +13438,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14343,7 +13467,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -14351,16 +13475,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14380,10 +13503,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -14405,7 +13524,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14416,7 +13534,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14428,11 +13546,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14443,7 +13560,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -14453,12 +13570,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14486,7 +13600,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -14494,16 +13608,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14515,13 +13628,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -14533,7 +13642,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14552,21 +13660,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14576,11 +13683,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -14590,7 +13693,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -14609,21 +13711,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14635,13 +13736,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -14655,7 +13752,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14674,21 +13770,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14700,13 +13795,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -14743,7 +13834,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14762,23 +13852,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14790,13 +13879,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -14808,7 +13893,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -14827,21 +13911,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14853,13 +13936,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -14887,7 +13966,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -14906,7 +13984,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -14915,16 +13993,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14936,13 +14013,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -14970,7 +14043,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14996,7 +14069,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -15005,16 +14078,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15034,10 +14106,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -15064,14 +14132,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15085,7 +14145,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15101,18 +14161,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15126,7 +14178,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -15139,20 +14191,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15186,7 +14227,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15196,24 +14237,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -15239,7 +14278,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -15248,16 +14287,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15269,13 +14307,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15304,14 +14338,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15325,7 +14351,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15341,18 +14367,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15366,7 +14384,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -15379,20 +14397,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15426,7 +14433,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15436,17 +14443,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -15465,22 +14470,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15490,11 +14494,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15509,14 +14509,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15544,7 +14536,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15553,17 +14545,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15585,7 +14573,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -15604,23 +14591,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15632,13 +14618,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15652,7 +14634,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -15671,22 +14652,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15698,13 +14678,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -15743,7 +14719,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -15762,7 +14737,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -15770,16 +14745,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15791,13 +14765,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15829,7 +14799,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -15848,7 +14817,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -15858,16 +14827,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15879,13 +14847,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -15897,7 +14861,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -15916,21 +14879,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15942,13 +14904,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -15956,7 +14914,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -15975,20 +14932,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16000,13 +14956,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -16020,7 +14972,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -16039,22 +14990,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16066,13 +15016,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -16113,7 +15059,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -16132,7 +15077,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -16141,16 +15086,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16162,13 +15106,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -16205,7 +15145,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -16224,7 +15163,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -16235,16 +15174,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16256,13 +15194,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -16277,7 +15211,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -16296,22 +15229,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16323,13 +15255,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -16341,7 +15269,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -16360,21 +15287,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16386,13 +15312,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -16409,7 +15331,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -16428,23 +15349,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16456,13 +15376,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -16492,7 +15408,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -16514,16 +15429,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16534,14 +15448,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -16551,11 +15465,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16567,8 +15481,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -16597,7 +15509,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16619,16 +15530,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16639,14 +15549,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16656,11 +15566,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16672,15 +15582,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -16706,7 +15612,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -16714,16 +15620,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16737,13 +15642,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -16767,14 +15668,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16787,7 +15680,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16800,18 +15693,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16824,7 +15709,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -16835,19 +15720,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16880,7 +15755,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16892,20 +15767,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -16930,7 +15807,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -16938,16 +15815,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16967,10 +15843,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -16993,14 +15865,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17012,7 +15876,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17025,18 +15889,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17045,7 +15901,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -17055,19 +15915,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17099,7 +15948,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17109,20 +15958,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -17147,7 +15998,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17155,16 +16006,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17176,13 +16026,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -17205,14 +16051,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17224,7 +16062,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17236,18 +16074,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17256,7 +16086,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -17265,19 +16099,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] @@ -17309,7 +16132,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17319,17 +16142,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -17348,22 +16169,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17373,11 +16193,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -17392,14 +16208,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17427,7 +16235,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17436,17 +16244,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17461,7 +16265,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -17480,22 +16283,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17507,13 +16309,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -17525,7 +16323,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -17544,21 +16341,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17570,13 +16366,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17595,21 +16387,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17619,11 +16410,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17634,14 +16421,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17668,7 +16447,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17677,17 +16456,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17706,21 +16481,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17730,11 +16504,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17745,14 +16515,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17779,7 +16541,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17788,17 +16550,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -17815,7 +16573,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17834,21 +16591,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17858,11 +16614,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -17871,7 +16623,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -17893,16 +16644,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17913,14 +16663,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -17930,11 +16680,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17946,10 +16696,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -17958,7 +16704,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -17980,16 +16725,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18000,14 +16744,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -18017,11 +16761,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18033,8 +16777,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18046,7 +16788,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -18068,17 +16809,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18089,14 +16829,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -18106,11 +16846,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18122,10 +16862,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -18143,22 +16879,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18168,11 +16903,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -18189,14 +16920,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18224,7 +16947,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18233,17 +16956,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18262,21 +16981,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18286,11 +17004,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18300,14 +17014,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18334,7 +17040,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18343,17 +17049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18372,21 +17074,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18396,11 +17097,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18410,14 +17107,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18444,7 +17133,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18453,17 +17142,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18482,21 +17167,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18506,11 +17190,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18520,14 +17200,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18554,7 +17226,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18563,17 +17235,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18592,21 +17260,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18616,11 +17283,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18630,14 +17293,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18664,7 +17319,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18673,23 +17328,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18718,7 +17369,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18726,16 +17377,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18745,11 +17395,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -18773,14 +17419,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18791,7 +17429,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18804,18 +17442,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18826,7 +17456,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -18836,19 +17466,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18878,7 +17498,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18887,17 +17507,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -18919,7 +17535,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18938,22 +17553,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18965,13 +17579,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18990,21 +17600,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19014,11 +17623,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19028,14 +17633,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19062,7 +17659,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19071,17 +17668,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -19093,7 +17686,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19112,21 +17704,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19136,11 +17727,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -19159,21 +17746,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19187,13 +17773,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -19205,14 +17787,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -19242,7 +17816,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19254,23 +17828,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -19296,7 +17868,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -19304,16 +17876,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19327,13 +17898,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -19358,14 +17925,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19378,7 +17937,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19391,18 +17950,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19415,7 +17966,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -19425,20 +17976,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19471,7 +18011,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19483,20 +18023,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -19521,7 +18063,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -19529,16 +18071,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19552,13 +18093,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -19581,14 +18118,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -19600,7 +18129,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19612,18 +18141,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -19632,7 +18153,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -19641,19 +18166,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -19685,7 +18199,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19697,17 +18211,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -19737,7 +18249,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -19761,16 +18272,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19781,14 +18291,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -19798,11 +18308,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19814,15 +18324,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -19848,7 +18354,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -19856,16 +18362,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19885,10 +18390,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -19910,14 +18411,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -19930,7 +18423,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19942,18 +18435,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -19966,7 +18451,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -19975,20 +18460,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -20021,7 +18495,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20031,23 +18505,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -20073,7 +18545,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20081,16 +18553,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20102,13 +18573,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -20131,14 +18598,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -20151,7 +18610,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20163,18 +18622,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -20187,7 +18638,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -20196,20 +18647,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -20242,7 +18683,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20252,17 +18693,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -20281,22 +18720,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20306,11 +18744,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20325,14 +18759,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20360,7 +18786,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20369,17 +18795,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> _models.VirtualMachineScaleSet: @@ -20389,7 +18811,6 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -20408,21 +18829,20 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20434,19 +18854,15 @@ def get(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) - deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -20475,7 +18891,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20483,16 +18899,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20502,11 +18917,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -20533,14 +18944,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20551,7 +18954,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20566,18 +18969,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20588,7 +18983,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -20600,20 +18995,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20643,7 +19027,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20652,23 +19036,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -20696,7 +19076,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20705,16 +19085,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20724,11 +19103,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -20757,14 +19132,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20775,7 +19142,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -20789,21 +19156,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20814,7 +19173,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -20826,23 +19185,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20873,7 +19221,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20882,17 +19230,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -20904,7 +19248,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -20923,21 +19266,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20949,13 +19291,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -20963,7 +19301,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -20987,16 +19324,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21007,14 +19343,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -21024,11 +19360,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21040,17 +19376,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -21074,15 +19405,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21093,14 +19423,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -21110,11 +19440,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21126,8 +19456,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -21139,7 +19467,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -21163,17 +19490,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21184,14 +19510,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -21201,11 +19527,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21217,10 +19543,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -21231,7 +19553,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -21255,17 +19576,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21276,14 +19596,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -21293,11 +19613,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21309,15 +19629,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -21348,7 +19664,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21357,16 +19673,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21376,11 +19691,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -21412,14 +19723,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21430,7 +19733,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -21446,7 +19749,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -21454,14 +19757,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21472,7 +19767,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -21486,24 +19781,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21534,7 +19818,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21543,23 +19827,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -21588,7 +19868,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21596,16 +19876,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21615,11 +19894,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -21644,14 +19919,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21662,7 +19929,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -21675,18 +19942,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21697,7 +19956,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -21707,20 +19966,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -21750,7 +19998,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21759,23 +20007,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -21804,7 +20048,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21812,16 +20056,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21831,11 +20074,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -21860,14 +20099,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21878,7 +20109,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -21891,18 +20122,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21913,7 +20136,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -21923,20 +20146,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -21966,7 +20178,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21975,23 +20187,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -22020,7 +20228,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -22028,16 +20236,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22047,11 +20254,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -22077,14 +20280,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22095,7 +20290,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -22109,18 +20304,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22131,7 +20318,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -22142,20 +20329,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -22185,7 +20361,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22194,23 +20370,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -22239,7 +20411,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -22247,16 +20419,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22266,11 +20437,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -22298,14 +20465,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22316,7 +20475,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -22332,18 +20491,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22354,7 +20505,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -22367,20 +20518,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -22410,7 +20550,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22419,23 +20559,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -22461,7 +20597,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -22469,16 +20605,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22488,11 +20623,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -22517,14 +20648,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22535,7 +20658,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22548,18 +20671,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22570,7 +20685,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -22580,20 +20695,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -22623,7 +20727,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22632,23 +20736,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -22677,7 +20777,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -22685,16 +20785,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22704,11 +20803,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -22734,14 +20829,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22752,7 +20839,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -22766,18 +20853,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22788,7 +20867,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -22800,20 +20879,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -22843,7 +20911,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22852,23 +20920,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -22897,7 +20961,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -22905,16 +20969,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22924,11 +20987,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -22954,14 +21013,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22972,7 +21023,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -22986,18 +21037,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23008,7 +21051,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -23019,20 +21062,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -23062,7 +21094,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23071,20 +21103,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -23097,7 +21125,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -23116,22 +21143,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23143,13 +21169,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -23174,7 +21196,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -23185,7 +21206,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23198,11 +21219,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -23213,7 +21233,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -23224,13 +21244,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2020_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2020_12_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -23258,7 +21275,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -23266,16 +21283,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23285,17 +21301,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -23321,7 +21333,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -23329,16 +21341,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23348,11 +21359,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -23376,14 +21383,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23394,7 +21393,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23407,18 +21406,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23429,7 +21420,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -23440,19 +21431,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -23482,7 +21463,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23491,17 +21472,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -23531,7 +21508,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -23553,16 +21529,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23573,14 +21548,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -23590,11 +21565,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23606,8 +21581,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -23630,7 +21603,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -23655,7 +21628,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -23663,16 +21636,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23692,10 +21664,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -23717,14 +21685,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23735,7 +21695,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23747,18 +21707,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23766,7 +21718,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -23775,19 +21727,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23818,7 +21759,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23828,20 +21769,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -23866,7 +21809,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -23874,16 +21817,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23903,10 +21845,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -23928,14 +21866,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23946,7 +21876,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23958,18 +21888,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23977,7 +21899,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -23986,19 +21912,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24029,7 +21944,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24039,17 +21954,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -24068,21 +21981,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24092,11 +22004,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -24106,14 +22014,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24140,7 +22040,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24149,17 +22049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24173,7 +22069,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -24192,22 +22087,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24219,13 +22113,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -24233,7 +22123,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24255,16 +22144,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24275,14 +22163,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -24292,11 +22180,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24308,16 +22196,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2020_12_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24339,15 +22222,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24358,14 +22240,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -24375,11 +22257,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24391,10 +22273,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -24419,7 +22299,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -24445,7 +22325,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -24454,16 +22334,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24483,10 +22362,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -24514,14 +22389,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24535,7 +22402,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24551,18 +22418,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24576,7 +22435,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -24589,20 +22448,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24636,7 +22484,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24646,24 +22494,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -24689,7 +22535,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -24698,16 +22544,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24727,10 +22572,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -24758,14 +22599,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24779,7 +22612,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24795,18 +22628,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24820,7 +22645,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -24833,20 +22658,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24880,7 +22695,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24890,17 +22705,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -24919,22 +22732,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24944,11 +22756,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24963,14 +22771,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24998,7 +22798,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25007,17 +22807,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25039,7 +22835,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -25058,23 +22853,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25086,13 +22880,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -25104,7 +22894,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -25128,17 +22917,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25149,14 +22937,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -25166,11 +22954,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25182,12 +22970,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -25224,21 +23008,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25248,11 +23031,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -25262,14 +23041,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25296,7 +23067,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25305,17 +23076,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -25334,21 +23101,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25358,11 +23124,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -25376,14 +23138,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25410,7 +23164,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25419,17 +23173,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -25448,21 +23198,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25472,11 +23221,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -25490,14 +23235,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25524,7 +23261,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25533,17 +23270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -25555,7 +23288,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -25574,21 +23306,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25600,16 +23331,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -25635,7 +23362,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -25661,7 +23388,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25671,16 +23398,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25700,10 +23426,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -25733,14 +23455,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25755,7 +23469,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25772,18 +23486,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25798,7 +23504,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -25812,20 +23518,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25860,7 +23555,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25870,17 +23565,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -25888,7 +23581,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -25914,7 +23607,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25924,16 +23617,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25945,13 +23637,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25982,14 +23670,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -26004,7 +23684,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26021,18 +23701,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -26047,7 +23719,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -26061,20 +23733,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -26109,7 +23771,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26119,17 +23781,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -26148,23 +23808,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26174,11 +23833,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26194,14 +23849,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26230,7 +23877,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26239,17 +23886,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -26274,7 +23917,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -26293,7 +23935,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26301,16 +23943,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26322,13 +23963,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -26350,7 +23987,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -26369,23 +24005,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26397,13 +24032,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -26431,7 +24062,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -26460,7 +24093,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26469,16 +24102,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26488,11 +24120,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -26520,14 +24148,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26539,7 +24159,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -26554,18 +24174,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26577,7 +24189,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -26589,20 +24203,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -26633,7 +24237,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26642,17 +24246,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26671,22 +24271,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26696,11 +24295,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -26715,14 +24310,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26750,7 +24337,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26759,17 +24346,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26788,22 +24371,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26813,11 +24395,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -26833,14 +24411,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26868,7 +24438,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26877,24 +24447,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -26920,7 +24486,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26929,16 +24495,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26958,10 +24523,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -26988,14 +24549,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -27009,7 +24562,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27025,18 +24578,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -27050,7 +24595,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -27063,19 +24608,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -27109,7 +24643,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27119,17 +24653,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -27154,23 +24686,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27180,11 +24711,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27207,14 +24734,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27243,7 +24762,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27252,17 +24771,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27285,7 +24800,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are "instanceView" and None. Default value is "instanceView". :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -27304,23 +24818,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27332,13 +24845,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -27352,7 +24861,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -27371,22 +24879,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27398,13 +24905,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -27433,7 +24936,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -27457,7 +24959,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -27465,12 +24967,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27481,14 +24982,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -27498,11 +24999,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27514,10 +25015,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -27541,23 +25038,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27567,11 +25063,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -27597,14 +25089,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27633,7 +25117,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27642,17 +25126,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27671,22 +25151,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27696,11 +25175,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -27714,14 +25189,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27749,7 +25216,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27758,17 +25225,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27787,22 +25250,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27812,11 +25274,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -27830,14 +25288,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27865,7 +25315,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27874,17 +25324,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27903,22 +25349,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27928,11 +25373,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -27947,14 +25388,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27982,7 +25415,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27991,17 +25424,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -28027,7 +25456,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -28046,23 +25474,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28074,13 +25501,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -28099,22 +25522,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28124,11 +25546,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -28142,14 +25560,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28177,7 +25587,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28186,17 +25596,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -28210,7 +25616,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -28229,22 +25634,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28254,18 +25658,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -28291,7 +25691,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28300,16 +25700,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28323,13 +25722,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -28355,14 +25750,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -28375,7 +25762,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28389,18 +25776,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -28413,7 +25792,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -28425,19 +25804,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2020_12_01.models.RunCommandResult] @@ -28470,7 +25838,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28482,17 +25850,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -28516,7 +25882,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -28541,23 +25907,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28571,13 +25936,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -28599,14 +25960,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28616,7 +25969,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -28625,18 +25978,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28646,7 +25991,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -28654,19 +25999,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28698,7 +26033,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28710,20 +26045,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -28748,23 +26081,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28778,13 +26110,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -28805,14 +26133,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28822,7 +26142,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28830,18 +26150,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28851,7 +26163,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28859,19 +26171,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2020_12_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28903,7 +26204,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28915,17 +26216,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -28954,7 +26253,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -28978,16 +26276,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28998,14 +26295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -29015,11 +26312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29031,10 +26328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -29043,7 +26336,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -29062,21 +26354,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29088,20 +26379,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29127,7 +26414,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -29136,16 +26423,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29165,10 +26451,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -29195,14 +26477,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29216,7 +26490,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29232,18 +26506,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29257,7 +26523,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -29270,19 +26536,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29316,7 +26571,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29326,24 +26581,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29369,7 +26622,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -29378,16 +26631,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29399,13 +26651,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -29433,14 +26681,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29454,7 +26694,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29470,18 +26710,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29495,7 +26727,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -29508,19 +26740,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29554,7 +26776,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29564,17 +26786,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -29593,22 +26813,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29618,11 +26837,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -29637,14 +26852,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29672,7 +26879,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -29681,17 +26888,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -29713,7 +26916,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -29732,23 +26934,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29760,13 +26961,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -29780,7 +26977,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29804,18 +27000,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29826,14 +27021,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -29843,11 +27038,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29859,12 +27054,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -29890,7 +27081,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29916,7 +27107,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29926,16 +27117,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29955,10 +27145,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -29987,14 +27173,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30009,7 +27187,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30026,18 +27204,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30052,7 +27222,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -30066,19 +27236,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30113,7 +27272,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30123,17 +27282,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -30141,7 +27298,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -30167,7 +27324,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -30177,16 +27334,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30198,13 +27354,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -30234,14 +27386,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30256,7 +27400,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30273,18 +27417,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30299,7 +27435,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -30313,19 +27449,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30360,7 +27486,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30370,17 +27496,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -30399,23 +27523,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30425,11 +27548,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -30445,14 +27564,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -30481,7 +27592,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -30490,17 +27601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -30525,7 +27632,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2020_12_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -30544,7 +27650,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2020-12-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -30552,16 +27658,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30573,13 +27678,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -30601,7 +27702,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30625,19 +27725,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -30648,14 +27747,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -30665,11 +27764,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30680,7 +27779,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_compute_management_client.py index f7c859ffae43..1718b1c2c5e6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -161,7 +163,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -249,7 +269,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -269,7 +289,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_configuration.py index 3ca59115714e..489db95b5a26 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_metadata.json index a135ce6476b6..1d67412c6370 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "cloud_service_role_instances": "CloudServiceRoleInstancesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_compute_management_client.py index 01506c6fd6ad..ef192856c59e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -163,7 +165,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -251,7 +271,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -271,7 +293,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_configuration.py index cae9c72ebc1c..7a87e1f88600 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/operations/_operations.py index 8537e5402022..fa4c3ce73d94 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -264,22 +264,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -289,11 +288,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -307,14 +302,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -342,7 +329,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -351,17 +338,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -384,7 +367,6 @@ async def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -403,23 +385,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -431,13 +412,9 @@ async def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -451,7 +428,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -470,22 +446,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -497,13 +472,9 @@ async def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -525,7 +496,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.RoleInstance] @@ -548,18 +518,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -570,14 +539,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -587,11 +556,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -603,10 +572,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -624,22 +589,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -649,11 +613,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -668,14 +628,6 @@ async def begin_restart( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -703,7 +655,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -712,17 +664,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -741,22 +689,21 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -766,11 +713,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage( @@ -785,14 +728,6 @@ async def begin_reimage( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -820,7 +755,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -829,17 +764,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -858,22 +789,21 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -883,11 +813,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_rebuild( @@ -903,14 +829,6 @@ async def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -938,7 +856,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -947,17 +865,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_remote_desktop_file( @@ -971,8 +885,7 @@ async def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Async iterator of the response bytes or the result of cls(response) + :return: AsyncIterator[bytes] or the result of cls(response) :rtype: AsyncIterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -990,22 +903,21 @@ async def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1021,10 +933,6 @@ async def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -1058,7 +966,6 @@ async def get( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -1077,22 +984,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1104,13 +1010,9 @@ async def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1123,7 +1025,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudServiceRole] @@ -1146,17 +1047,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1167,14 +1067,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1184,11 +1084,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1200,10 +1100,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -1229,7 +1125,7 @@ async def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1258,7 +1154,7 @@ async def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1266,16 +1162,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1295,10 +1190,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload async def begin_create_or_update( self, @@ -1321,14 +1212,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1340,7 +1223,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1353,18 +1236,10 @@ async def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1376,7 +1251,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -1386,20 +1261,9 @@ async def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudService or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1431,7 +1295,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1441,23 +1305,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1486,7 +1348,7 @@ async def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1494,16 +1356,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1515,13 +1376,9 @@ async def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1544,14 +1401,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1563,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1575,18 +1424,10 @@ async def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1598,7 +1439,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -1607,20 +1448,9 @@ async def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1652,7 +1482,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1662,17 +1492,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1691,21 +1519,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1715,11 +1542,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1731,14 +1554,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1765,7 +1580,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1774,17 +1589,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -1794,7 +1605,6 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -1813,21 +1623,20 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1839,13 +1648,9 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -1857,7 +1662,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1876,21 +1680,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1902,13 +1705,9 @@ async def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: @@ -1916,7 +1715,6 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1939,15 +1737,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1958,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1975,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1991,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -2001,7 +1796,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -2024,16 +1818,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2044,14 +1837,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2061,11 +1854,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2077,10 +1870,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -2098,21 +1887,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2122,11 +1910,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -2138,14 +1922,6 @@ async def begin_start( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2172,7 +1948,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2181,17 +1957,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2210,21 +1982,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2234,11 +2005,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -2251,14 +2018,6 @@ async def begin_power_off( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2285,7 +2044,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2294,23 +2053,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2339,7 +2094,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2347,16 +2102,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2366,11 +2120,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2393,14 +2143,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2411,7 +2153,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2423,18 +2165,10 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2445,7 +2179,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -2455,19 +2189,8 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2497,7 +2220,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2506,23 +2229,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2551,7 +2270,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2559,16 +2278,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2578,11 +2296,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -2606,14 +2320,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2624,7 +2330,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2637,18 +2343,10 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2659,7 +2357,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -2670,19 +2368,8 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2712,7 +2399,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2721,23 +2408,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2766,7 +2449,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2774,16 +2457,15 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2793,11 +2475,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_rebuild( @@ -2822,14 +2500,6 @@ async def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2840,7 +2510,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2854,18 +2524,10 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2876,7 +2538,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -2888,19 +2550,8 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2930,7 +2581,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2939,23 +2590,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2984,7 +2631,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2992,16 +2639,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3011,11 +2657,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -3038,14 +2680,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3056,7 +2690,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3068,18 +2702,10 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3090,7 +2716,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -3100,19 +2726,8 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3142,7 +2757,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3151,17 +2766,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3189,7 +2800,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3218,7 +2829,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3227,16 +2838,15 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3246,11 +2856,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_walk_update_domain( @@ -3278,14 +2884,6 @@ async def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3297,7 +2895,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3313,18 +2911,10 @@ async def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3336,7 +2926,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -3349,20 +2939,9 @@ async def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3393,7 +2972,7 @@ async def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3402,17 +2981,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_update_domain( @@ -3430,7 +3005,6 @@ async def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -3449,22 +3023,21 @@ async def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3476,13 +3049,9 @@ async def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -3494,7 +3063,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.UpdateDomain] @@ -3517,17 +3085,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3538,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -3555,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3571,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -3605,7 +3168,6 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3624,21 +3186,20 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3650,13 +3211,9 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSVersion"]: @@ -3666,7 +3223,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.OSVersion] @@ -3689,16 +3245,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3709,14 +3264,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -3726,11 +3281,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3742,10 +3297,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace_async async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -3755,7 +3306,6 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -3774,21 +3324,20 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3800,13 +3349,9 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSFamily"]: @@ -3816,7 +3361,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.OSFamily] @@ -3839,16 +3383,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3859,14 +3402,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -3876,11 +3419,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3892,10 +3435,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } - class Operations: """ @@ -3921,7 +3460,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -3945,14 +3483,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3963,14 +3500,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -3980,11 +3517,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3996,8 +3533,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -4040,7 +3575,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4051,7 +3585,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4063,11 +3597,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4078,7 +3611,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -4088,12 +3621,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4121,7 +3650,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4129,16 +3658,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4150,13 +3678,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -4179,7 +3703,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4190,7 +3713,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4202,11 +3725,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4217,7 +3739,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -4227,12 +3749,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4260,7 +3778,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -4268,16 +3786,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4289,13 +3806,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -4307,7 +3820,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4326,21 +3838,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4350,11 +3861,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -4364,7 +3871,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -4383,21 +3889,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4409,13 +3914,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -4426,7 +3927,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet] @@ -4449,16 +3949,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4469,14 +3968,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4486,11 +3985,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4502,17 +4001,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet] @@ -4535,16 +4029,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4555,14 +4048,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -4572,11 +4065,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4588,10 +4081,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -4603,7 +4092,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] @@ -4626,17 +4114,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4647,14 +4134,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -4664,11 +4151,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4680,10 +4167,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -4727,7 +4210,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4738,7 +4220,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4751,11 +4233,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4766,7 +4247,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -4776,12 +4257,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4809,7 +4286,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4817,16 +4294,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4846,10 +4322,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -4872,7 +4344,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4883,7 +4354,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4896,11 +4367,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4911,7 +4381,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -4921,12 +4391,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -4954,7 +4421,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -4962,16 +4429,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4983,13 +4449,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -5001,7 +4463,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5020,21 +4481,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5044,11 +4504,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -5068,7 +4524,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5087,22 +4542,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5114,19 +4568,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5150,15 +4599,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5169,14 +4617,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5186,11 +4634,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5202,10 +4650,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -5214,7 +4658,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -5238,16 +4681,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5258,14 +4700,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -5275,11 +4717,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5291,10 +4733,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -5339,7 +4777,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5350,7 +4787,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5364,11 +4801,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5379,7 +4815,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -5391,12 +4827,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5424,7 +4856,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -5432,16 +4864,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5461,10 +4892,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -5486,7 +4913,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5497,7 +4923,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5509,11 +4935,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5524,7 +4949,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -5534,12 +4959,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5567,7 +4988,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -5575,16 +4996,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5596,13 +5016,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -5614,7 +5030,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -5633,21 +5048,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5657,11 +5071,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -5683,7 +5093,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -5702,22 +5111,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5729,13 +5137,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -5746,7 +5150,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup] @@ -5769,16 +5172,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5789,14 +5191,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -5806,11 +5208,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5822,16 +5224,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup] @@ -5854,15 +5251,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5873,14 +5269,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -5890,11 +5286,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5906,8 +5302,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -5934,7 +5328,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -5960,7 +5354,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -5969,16 +5363,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5998,10 +5391,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -6026,14 +5415,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6047,7 +5428,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6061,18 +5442,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6086,7 +5459,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -6098,19 +5471,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6144,7 +5506,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6154,24 +5516,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -6197,7 +5557,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -6206,16 +5566,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6227,13 +5586,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6259,14 +5614,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6280,7 +5627,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6294,18 +5641,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6319,7 +5658,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -6331,19 +5670,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -6377,7 +5705,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6387,17 +5715,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -6416,22 +5742,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6441,11 +5766,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6459,14 +5780,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6494,7 +5807,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6503,17 +5816,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6537,7 +5846,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -6556,23 +5864,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6584,13 +5891,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -6603,7 +5906,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -6626,17 +5928,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6647,14 +5948,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -6664,11 +5965,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6680,10 +5981,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -6710,7 +6007,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -6734,15 +6030,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6753,14 +6048,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -6770,11 +6065,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6786,8 +6081,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -6797,7 +6090,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -6821,16 +6113,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6841,14 +6132,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -6858,11 +6149,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6874,10 +6165,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -6899,7 +6186,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -6910,7 +6196,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6922,11 +6208,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -6937,7 +6222,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -6947,12 +6232,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -6980,7 +6261,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -6988,16 +6269,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7017,10 +6297,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -7042,7 +6318,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7053,7 +6328,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7065,11 +6340,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7080,7 +6354,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -7090,12 +6364,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7123,7 +6394,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -7131,16 +6402,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7152,13 +6422,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -7170,7 +6436,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7189,21 +6454,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7213,11 +6477,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -7229,7 +6489,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -7248,21 +6507,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7274,13 +6532,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -7294,7 +6548,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7313,21 +6566,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7339,13 +6591,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -7382,7 +6630,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -7401,23 +6648,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7429,13 +6675,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -7447,7 +6689,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7466,21 +6707,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7492,13 +6732,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -7526,7 +6762,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -7545,7 +6780,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -7554,16 +6789,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7575,13 +6809,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -7609,7 +6839,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -7635,7 +6865,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -7644,16 +6874,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7673,10 +6902,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -7703,14 +6928,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7724,7 +6941,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7740,18 +6957,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7765,7 +6974,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -7778,20 +6987,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7825,7 +7023,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7835,24 +7033,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -7878,7 +7074,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -7887,16 +7083,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7908,13 +7103,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -7943,14 +7134,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -7964,7 +7147,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7980,18 +7163,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8005,7 +7180,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8018,20 +7193,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8065,7 +7229,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8075,17 +7239,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8104,22 +7266,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8129,11 +7290,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -8148,14 +7305,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8183,7 +7332,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8192,17 +7341,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -8224,7 +7369,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -8243,23 +7387,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8271,13 +7414,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -8291,7 +7430,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8310,22 +7448,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8337,13 +7474,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -8382,7 +7515,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8401,7 +7533,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -8409,16 +7541,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8430,13 +7561,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -8468,7 +7595,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8487,7 +7613,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -8497,16 +7623,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8518,13 +7643,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -8536,7 +7657,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8555,21 +7675,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8581,13 +7700,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -8595,7 +7710,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8614,20 +7728,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8639,13 +7752,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -8659,7 +7768,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8678,22 +7786,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8705,13 +7812,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -8752,7 +7855,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -8771,7 +7873,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -8780,16 +7882,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8802,13 +7903,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -8845,7 +7942,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8864,7 +7960,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -8875,16 +7971,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8897,13 +7992,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -8918,7 +8009,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -8937,22 +8027,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8965,13 +8054,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -8983,7 +8068,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9002,21 +8086,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9029,13 +8112,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -9052,7 +8131,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -9071,23 +8149,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9100,13 +8177,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -9136,7 +8209,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9158,16 +8230,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9178,14 +8249,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -9195,11 +8266,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9211,8 +8282,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -9241,7 +8310,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -9264,16 +8332,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9284,14 +8351,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9301,11 +8368,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9317,15 +8384,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9351,7 +8414,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9359,16 +8422,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9382,13 +8444,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9412,14 +8470,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9432,7 +8482,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9445,18 +8495,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9469,7 +8511,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9480,19 +8522,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9525,7 +8557,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9537,20 +8569,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9575,7 +8609,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9583,16 +8617,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9612,10 +8645,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9638,14 +8667,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9658,7 +8679,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9671,18 +8692,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9692,7 +8705,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9702,19 +8719,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9747,7 +8753,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9757,20 +8763,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9795,7 +8803,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9803,16 +8811,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9824,13 +8831,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9853,14 +8856,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9873,7 +8868,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9885,18 +8880,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9906,7 +8893,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -9915,19 +8906,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9960,7 +8940,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9970,17 +8950,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9999,22 +8977,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10024,11 +9001,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10043,14 +9016,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10078,7 +9043,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10087,17 +9052,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10120,7 +9081,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10139,22 +9099,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10166,13 +9125,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10184,7 +9139,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10203,21 +9157,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10229,13 +9182,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10254,21 +9203,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10278,11 +9226,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10295,14 +9239,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10329,7 +9265,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10338,17 +9274,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10367,21 +9299,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10391,11 +9322,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -10406,14 +9333,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10440,7 +9359,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10449,17 +9368,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10476,7 +9391,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10495,21 +9409,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10519,11 +9432,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -10532,7 +9441,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -10555,16 +9463,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10575,14 +9482,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10592,11 +9499,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10608,10 +9515,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -10620,7 +9523,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -10643,16 +9545,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10663,14 +9564,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10680,11 +9581,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10696,8 +9597,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10709,7 +9608,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] @@ -10732,17 +9630,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10753,14 +9650,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10770,11 +9667,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10786,10 +9683,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10807,22 +9700,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10832,11 +9724,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -10853,14 +9741,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10888,7 +9768,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10897,17 +9777,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10926,21 +9802,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10951,11 +9826,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -10965,14 +9836,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10999,7 +9862,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11008,17 +9871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11037,21 +9896,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11061,11 +9919,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11075,14 +9929,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11109,7 +9955,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11118,17 +9964,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11147,21 +9989,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11171,11 +10012,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11185,14 +10022,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11219,7 +10048,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11228,17 +10057,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11257,21 +10082,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11281,11 +10105,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11295,14 +10115,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11329,7 +10141,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11338,23 +10150,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11383,7 +10191,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11391,16 +10199,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11410,11 +10217,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11438,14 +10241,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11456,7 +10251,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11469,18 +10264,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11491,7 +10278,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11501,19 +10288,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11543,7 +10320,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11552,17 +10329,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11584,7 +10357,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11603,22 +10375,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11631,13 +10402,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11656,21 +10423,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11680,11 +10446,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11696,14 +10458,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11730,7 +10484,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11739,17 +10493,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11761,7 +10511,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11780,21 +10529,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11804,11 +10552,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11827,21 +10571,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11856,13 +10599,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -11874,14 +10613,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -11911,7 +10642,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11923,23 +10654,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -11965,7 +10694,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11973,16 +10702,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11997,13 +10725,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12028,14 +10752,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12048,7 +10764,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12061,18 +10777,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12085,7 +10793,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12095,20 +10803,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12141,7 +10838,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12153,20 +10850,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12191,7 +10890,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12199,16 +10898,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12222,13 +10920,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12251,14 +10945,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12271,7 +10957,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12283,18 +10969,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12304,7 +10982,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12313,19 +10995,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12358,7 +11029,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12370,17 +11041,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -12410,7 +11079,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12434,16 +11102,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12454,14 +11121,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -12471,11 +11138,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12488,15 +11155,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -12522,7 +11185,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12530,16 +11193,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12559,10 +11221,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -12584,14 +11242,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12604,7 +11254,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12616,18 +11266,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12640,7 +11282,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -12649,20 +11291,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12695,7 +11326,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12705,23 +11336,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -12747,7 +11376,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12755,16 +11384,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12776,13 +11404,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -12805,14 +11429,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12825,7 +11441,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12837,18 +11453,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12861,7 +11469,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -12870,20 +11478,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -12916,7 +11514,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12926,17 +11524,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -12955,22 +11551,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12980,11 +11575,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -12999,14 +11590,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13034,7 +11617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13043,17 +11626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13074,7 +11653,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -13093,22 +11671,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13120,19 +11697,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13161,7 +11734,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13169,16 +11742,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13188,11 +11760,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -13219,14 +11787,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13237,7 +11797,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13252,18 +11812,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13274,7 +11826,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -13286,20 +11838,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13329,7 +11870,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13338,23 +11879,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -13382,7 +11919,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -13391,16 +11928,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13410,11 +11946,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -13443,14 +11975,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13461,7 +11985,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -13475,21 +11999,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13500,7 +12016,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -13512,23 +12028,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13559,7 +12064,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13568,17 +12073,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -13590,7 +12091,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13609,21 +12109,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13635,13 +12134,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -13649,7 +12144,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13673,16 +12167,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13693,14 +12186,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -13710,11 +12203,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13726,17 +12219,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -13760,15 +12248,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13779,14 +12266,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -13796,11 +12283,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13812,8 +12299,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13825,7 +12310,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -13849,17 +12333,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13870,14 +12353,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -13887,11 +12370,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13903,10 +12386,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -13917,7 +12396,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -13941,17 +12419,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13962,14 +12439,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -13979,11 +12456,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13995,15 +12472,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14034,7 +12507,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14043,16 +12516,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14062,11 +12534,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -14098,14 +12566,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14116,7 +12576,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -14132,7 +12592,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -14140,14 +12600,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14158,7 +12610,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -14172,24 +12624,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14220,7 +12661,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14229,23 +12670,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14274,7 +12711,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14282,16 +12719,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14301,11 +12737,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -14330,14 +12762,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14348,7 +12772,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14361,18 +12785,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14383,7 +12799,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -14393,20 +12809,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14436,7 +12841,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14445,23 +12850,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14490,7 +12891,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14498,16 +12899,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14517,11 +12917,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -14546,14 +12942,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14564,7 +12952,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14577,18 +12965,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14599,7 +12979,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -14609,20 +12989,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14652,7 +13021,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14661,23 +13030,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14706,7 +13071,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14714,16 +13079,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14733,11 +13097,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -14763,14 +13123,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14781,7 +13133,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14795,18 +13147,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14817,7 +13161,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -14828,20 +13172,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14871,7 +13204,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14880,23 +13213,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14925,7 +13254,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14933,16 +13262,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14952,11 +13280,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -14984,14 +13308,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15002,7 +13318,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15018,18 +13334,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15040,7 +13348,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -15053,20 +13361,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15096,7 +13393,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15105,23 +13402,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -15147,7 +13440,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15155,16 +13448,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15174,11 +13466,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -15203,14 +13491,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15221,7 +13501,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15234,18 +13514,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15256,7 +13528,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -15266,20 +13538,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15309,7 +13570,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15318,23 +13579,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15363,7 +13620,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15371,16 +13628,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15390,11 +13646,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -15420,14 +13672,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15438,7 +13682,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15452,18 +13696,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15474,7 +13710,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -15486,20 +13722,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15529,7 +13754,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15538,23 +13763,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15583,7 +13804,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15591,16 +13812,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15610,11 +13830,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -15640,14 +13856,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15658,7 +13866,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15672,18 +13880,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15694,7 +13894,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -15705,20 +13905,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15748,7 +13937,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15757,20 +13946,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -15783,7 +13968,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -15802,22 +13986,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15829,13 +14012,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -15860,7 +14039,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15871,7 +14049,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15884,11 +14062,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15899,7 +14076,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -15910,13 +14087,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15944,7 +14118,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15952,16 +14126,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15971,17 +14144,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -16007,7 +14176,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16015,16 +14184,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16034,11 +14202,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -16062,14 +14226,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16080,7 +14236,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16093,18 +14249,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16115,7 +14263,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -16126,19 +14274,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16168,7 +14306,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16177,17 +14315,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -16217,7 +14351,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] @@ -16240,16 +14373,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16260,14 +14392,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16277,11 +14409,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16293,8 +14425,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -16317,7 +14447,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -16342,7 +14472,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -16350,16 +14480,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16379,10 +14508,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -16404,14 +14529,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16422,7 +14539,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16434,18 +14551,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16453,7 +14562,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -16462,19 +14571,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16505,7 +14603,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16515,20 +14613,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -16553,7 +14653,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -16561,16 +14661,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16590,10 +14689,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -16615,14 +14710,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16633,7 +14720,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16645,18 +14732,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16664,7 +14743,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -16673,19 +14756,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16716,7 +14788,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16726,17 +14798,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -16755,21 +14825,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16779,11 +14848,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -16793,14 +14858,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16827,7 +14884,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16836,17 +14893,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16860,7 +14913,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -16879,22 +14931,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16906,13 +14957,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -16920,7 +14967,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -16942,16 +14988,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16962,14 +15007,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -16979,11 +15024,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16995,16 +15040,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17026,15 +15066,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17045,14 +15084,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17062,11 +15101,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17078,8 +15117,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -17125,7 +15162,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17136,7 +15172,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17151,11 +15187,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17166,7 +15201,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -17178,12 +15213,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17211,7 +15242,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17219,16 +15250,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17249,10 +15279,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -17275,7 +15301,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17286,7 +15311,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17299,11 +15324,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17314,7 +15338,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -17324,12 +15348,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17357,7 +15378,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17365,16 +15386,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17387,13 +15407,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -17412,21 +15428,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17437,11 +15452,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17454,14 +15465,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17488,7 +15491,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17497,17 +15500,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17529,7 +15528,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17548,22 +15546,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17576,13 +15573,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -17590,7 +15583,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -17614,16 +15606,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17634,14 +15625,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -17651,11 +15642,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17668,17 +15659,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -17702,15 +15688,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17721,14 +15706,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -17738,11 +15723,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17755,8 +15740,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -17783,7 +15766,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -17809,7 +15792,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -17818,16 +15801,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17840,13 +15822,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -17873,14 +15851,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -17893,7 +15863,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17908,18 +15878,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -17932,7 +15894,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -17945,19 +15907,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -17990,7 +15941,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18000,17 +15951,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18029,22 +15978,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18055,11 +16003,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18073,14 +16017,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18108,7 +16044,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18117,17 +16053,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18141,7 +16073,6 @@ async def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -18160,22 +16091,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18188,16 +16118,12 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -18222,7 +16148,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -18248,7 +16174,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -18257,16 +16183,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18286,10 +16211,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -18317,14 +16238,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18338,7 +16251,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18354,18 +16267,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18379,7 +16284,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -18392,20 +16297,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18439,7 +16333,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18449,24 +16343,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -18492,7 +16384,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -18501,16 +16393,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18530,10 +16421,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -18561,14 +16448,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18582,7 +16461,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18598,18 +16477,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18623,7 +16494,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -18636,20 +16507,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18683,7 +16544,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18693,17 +16554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -18722,22 +16581,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18747,11 +16605,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18766,14 +16620,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18801,7 +16647,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18810,17 +16656,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18842,7 +16684,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -18861,23 +16702,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18889,13 +16729,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18907,7 +16743,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -18931,17 +16766,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18952,14 +16786,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -18969,11 +16803,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18985,12 +16819,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -19027,21 +16857,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19051,11 +16880,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -19067,14 +16892,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19101,7 +16918,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19110,17 +16927,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19139,21 +16952,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19163,11 +16975,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -19181,14 +16989,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19215,7 +17015,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19224,17 +17024,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19253,21 +17049,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19277,11 +17072,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -19295,14 +17086,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19329,7 +17112,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19338,17 +17121,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -19360,7 +17139,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -19379,21 +17157,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19405,16 +17182,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -19440,7 +17213,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -19466,7 +17239,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19476,16 +17249,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19506,10 +17278,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -19539,14 +17307,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19561,7 +17321,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19578,18 +17338,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19604,7 +17356,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -19618,20 +17370,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19666,7 +17407,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19676,17 +17417,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -19694,7 +17433,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -19720,7 +17459,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19730,16 +17469,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19752,13 +17490,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19789,14 +17523,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19811,7 +17537,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19828,18 +17554,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19854,7 +17572,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -19868,20 +17586,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -19916,7 +17624,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19926,17 +17634,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -19955,23 +17661,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19982,11 +17687,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20002,14 +17703,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20038,7 +17731,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20047,17 +17740,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20082,7 +17771,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -20101,7 +17789,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20109,16 +17797,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20131,13 +17818,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -20159,7 +17842,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -20178,23 +17860,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20207,13 +17888,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -20241,7 +17918,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -20270,7 +17949,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20279,16 +17958,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20298,11 +17976,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -20330,14 +18004,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20349,7 +18015,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20364,18 +18030,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20387,7 +18045,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -20399,20 +18059,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20443,7 +18093,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20452,17 +18102,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20481,22 +18127,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20506,11 +18151,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -20525,14 +18166,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20560,7 +18193,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20569,17 +18202,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -20598,22 +18227,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20623,11 +18251,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -20643,14 +18267,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20678,7 +18294,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20687,24 +18303,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -20730,7 +18342,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20739,16 +18351,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20768,10 +18379,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -20798,14 +18405,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -20819,7 +18418,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20835,18 +18434,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -20860,7 +18451,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -20873,19 +18464,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -20919,7 +18499,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20929,17 +18509,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20964,23 +18542,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20990,11 +18567,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21017,14 +18590,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21053,7 +18618,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21062,17 +18627,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -21096,7 +18657,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -21115,23 +18675,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21143,13 +18702,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -21163,7 +18718,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -21182,22 +18736,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21209,13 +18762,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -21244,7 +18793,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -21268,7 +18816,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -21276,12 +18824,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21292,14 +18839,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -21309,11 +18856,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21325,10 +18872,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -21352,23 +18895,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21378,11 +18920,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -21408,14 +18946,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21444,7 +18974,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21453,17 +18983,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21482,22 +19008,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21507,11 +19032,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -21525,14 +19046,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21560,7 +19073,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21569,17 +19082,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21598,22 +19107,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21623,11 +19131,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -21641,14 +19145,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21676,7 +19172,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21685,17 +19181,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21714,22 +19206,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21739,11 +19230,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -21758,14 +19245,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21793,7 +19272,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21802,17 +19281,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -21838,7 +19313,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -21857,23 +19331,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21886,13 +19359,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -21911,22 +19380,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21936,11 +19404,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -21954,14 +19418,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21989,7 +19445,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21998,17 +19454,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -22022,7 +19474,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22041,22 +19492,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22066,18 +19516,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -22103,7 +19549,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22112,16 +19558,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22135,13 +19580,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -22167,14 +19608,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22188,7 +19621,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22202,18 +19635,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22227,7 +19652,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -22239,19 +19664,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -22285,7 +19699,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22297,17 +19711,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -22331,7 +19743,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -22356,23 +19768,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22386,13 +19797,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -22414,14 +19821,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22431,7 +19830,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -22440,18 +19839,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22461,7 +19852,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -22469,19 +19860,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22513,7 +19894,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22526,20 +19907,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -22564,23 +19943,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22594,13 +19972,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -22621,14 +19995,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22638,7 +20004,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -22646,18 +20012,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22667,7 +20025,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -22675,19 +20033,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22719,7 +20066,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22732,17 +20079,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -22771,7 +20116,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -22795,16 +20139,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22815,14 +20158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -22832,11 +20175,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22848,10 +20191,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -22860,7 +20199,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -22879,21 +20217,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22905,20 +20242,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22944,7 +20277,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -22953,16 +20286,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22983,10 +20315,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -23013,14 +20341,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23034,7 +20354,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23050,18 +20370,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23075,7 +20387,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -23088,19 +20400,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23134,7 +20435,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23144,24 +20445,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23187,7 +20486,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -23196,16 +20495,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23218,13 +20516,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -23252,14 +20546,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23273,7 +20559,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23289,18 +20575,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23314,7 +20592,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -23327,19 +20605,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23373,7 +20641,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23383,17 +20651,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -23412,22 +20678,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23438,11 +20703,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -23457,14 +20718,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23492,7 +20745,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23501,17 +20754,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -23533,7 +20782,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -23552,23 +20800,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23581,13 +20828,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -23601,7 +20844,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23625,18 +20867,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23647,14 +20888,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -23664,11 +20905,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23681,12 +20922,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23712,7 +20949,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23738,7 +20975,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23748,16 +20985,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23778,10 +21014,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -23810,14 +21042,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23832,7 +21056,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23849,18 +21073,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23875,7 +21091,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -23889,19 +21105,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23936,7 +21141,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23946,17 +21151,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -23964,7 +21167,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23990,7 +21193,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24000,16 +21203,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24022,13 +21224,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -24058,14 +21256,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24080,7 +21270,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24097,18 +21287,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24123,7 +21305,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -24137,19 +21319,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24184,7 +21356,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24194,17 +21366,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -24223,23 +21393,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24250,11 +21419,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -24270,14 +21435,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24306,7 +21463,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -24315,17 +21472,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -24350,7 +21503,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -24369,7 +21521,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24377,16 +21529,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24399,13 +21550,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24427,7 +21574,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -24451,19 +21597,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24474,14 +21619,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -24491,11 +21636,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24507,7 +21652,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_compute_management_client_enums.py index c9d7244ab346..2e578577fd07 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_compute_management_client_enums.py @@ -365,7 +365,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_models_py3.py index ee30ba08cb6b..68fdbdd06ccb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -336,7 +332,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -361,7 +357,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -376,7 +372,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -442,7 +438,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -499,7 +495,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -598,7 +594,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet] @@ -951,7 +947,7 @@ class CloudService(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1206,7 +1202,7 @@ def __init__(self, *, role_instance: Optional["_models.InstanceViewStatusesSumma class CloudServiceListResult(_serialization.Model): """CloudServiceListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -1506,7 +1502,7 @@ def __init__( class CloudServiceRoleListResult(_serialization.Model): """CloudServiceRoleListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.CloudServiceRole] @@ -1836,7 +1832,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -2047,7 +2043,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2224,7 +2220,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2340,7 +2336,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup] @@ -2544,7 +2540,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -2792,17 +2788,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -3071,7 +3056,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3251,7 +3236,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_03_01.models.SubResource @@ -3359,7 +3344,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.Image] @@ -3393,7 +3378,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_03_01.models.SubResource @@ -3839,7 +3824,7 @@ def __init__(self, **kwargs: Any) -> None: class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3872,7 +3857,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4158,7 +4143,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.Usage] @@ -4192,7 +4177,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LoadBalancerConfiguration(_serialization.Model): """Describes the load balancer configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4239,7 +4224,7 @@ def __init__( class LoadBalancerConfigurationProperties(_serialization.Model): """LoadBalancerConfigurationProperties. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar frontend_ip_configurations: Specifies the frontend IP to be used for the load balancer. Only IPv4 frontend IP address is supported. Each load balancer configuration must have exactly @@ -4276,7 +4261,7 @@ def __init__( class LoadBalancerFrontendIPConfiguration(_serialization.Model): """LoadBalancerFrontendIPConfiguration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the resource that is unique within the set of frontend IP configurations used by the load balancer. This name can be used to access the resource. @@ -4314,7 +4299,7 @@ def __init__( self.properties = properties -class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): +class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): # pylint: disable=name-too-long """Describes a cloud service IP Configuration. :ivar public_ip_address: The reference to the public ip address resource. @@ -4356,7 +4341,7 @@ def __init__( class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4704,7 +4689,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2021_03_01.models.OrchestrationServiceNames @@ -4777,7 +4762,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -4939,7 +4924,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5012,7 +4997,7 @@ def __init__(self, *, properties: Optional["_models.OSFamilyProperties"] = None, class OSFamilyListResult(_serialization.Model): """OSFamilyListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.OSFamily] @@ -5301,7 +5286,7 @@ def __init__(self, *, properties: Optional["_models.OSVersionProperties"] = None class OSVersionListResult(_serialization.Model): """OSVersionListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.OSVersion] @@ -5599,7 +5584,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5693,7 +5678,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup] @@ -5731,17 +5716,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5782,7 +5756,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2021_03_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2021_03_01.models.PublicIPAddressSkuTier """ @@ -5801,8 +5775,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2021_03_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2021_03_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5813,7 +5786,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5884,7 +5857,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6096,7 +6069,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6622,7 +6595,7 @@ def __init__( class RoleInstanceListResult(_serialization.Model): """RoleInstanceListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.RoleInstance] @@ -6710,7 +6683,7 @@ def __init__( class RoleInstances(_serialization.Model): """Specifies a list of role instances from the cloud service. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar role_instances: List of cloud service role instance names. Value of '*' will signify all role instances of the cloud service. Required. @@ -6989,7 +6962,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7052,7 +7025,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7115,7 +7088,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7189,7 +7162,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -7236,7 +7209,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7269,7 +7242,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.RunCommandDocumentBase] @@ -7305,7 +7278,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7601,7 +7574,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7611,7 +7584,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7640,7 +7613,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7655,7 +7628,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7711,7 +7684,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource] @@ -7955,7 +7928,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7976,67 +7949,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -8102,7 +8014,7 @@ def __init__(self, **kwargs: Any) -> None: class UpdateDomainListResult(_serialization.Model): """UpdateDomainListResult. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.UpdateDomain] @@ -8165,7 +8077,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8308,7 +8220,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8526,7 +8438,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8992,7 +8904,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -9082,7 +8994,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9209,7 +9121,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -9253,7 +9165,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9543,7 +9455,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_03_01.models.UserAssignedIdentitiesValue] """ @@ -9575,7 +9487,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2021_03_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_03_01.models.UserAssignedIdentitiesValue] """ @@ -9589,7 +9501,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9652,7 +9564,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9799,7 +9711,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). Required. @@ -10125,7 +10037,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -10158,10 +10070,10 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -10268,7 +10180,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -10288,10 +10200,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -10457,10 +10369,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10559,10 +10471,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -10617,7 +10529,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10864,7 +10776,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand] @@ -11013,7 +10925,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11245,7 +11157,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11479,10 +11391,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension] @@ -11693,7 +11605,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -11732,7 +11644,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -11743,7 +11655,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -11814,7 +11726,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11841,7 +11753,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11996,10 +11908,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -12041,7 +11953,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet] @@ -12077,7 +11989,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetSku] @@ -12113,7 +12025,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet] @@ -12146,7 +12058,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -12190,10 +12102,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -12296,7 +12208,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12321,7 +12233,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12352,7 +12264,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12371,7 +12283,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12632,10 +12544,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12723,10 +12635,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12753,7 +12667,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12761,18 +12675,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -13081,7 +12983,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -13199,7 +13101,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -13300,12 +13202,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -13338,7 +13240,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13473,7 +13375,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -13526,7 +13428,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -13674,7 +13576,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14041,7 +13943,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -14064,7 +13966,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -14222,10 +14124,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -14369,7 +14271,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM] @@ -14402,7 +14304,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -15146,7 +15048,7 @@ def __init__( # pylint: disable=too-many-locals self.user_data = user_data -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/operations/_operations.py index 241a9d6ae4b3..fd5393cf0180 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_03_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -39,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_cloud_service_role_instances_delete_request( +def build_cloud_service_role_instances_delete_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -71,7 +71,7 @@ def build_cloud_service_role_instances_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_request( +def build_cloud_service_role_instances_get_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, @@ -111,7 +111,7 @@ def build_cloud_service_role_instances_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_instance_view_request( +def build_cloud_service_role_instances_get_instance_view_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -143,7 +143,7 @@ def build_cloud_service_role_instances_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_list_request( +def build_cloud_service_role_instances_list_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, @@ -181,7 +181,7 @@ def build_cloud_service_role_instances_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_restart_request( +def build_cloud_service_role_instances_restart_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -213,7 +213,7 @@ def build_cloud_service_role_instances_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_reimage_request( +def build_cloud_service_role_instances_reimage_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -245,7 +245,7 @@ def build_cloud_service_role_instances_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_rebuild_request( +def build_cloud_service_role_instances_rebuild_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -277,7 +277,7 @@ def build_cloud_service_role_instances_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_remote_desktop_file_request( +def build_cloud_service_role_instances_get_remote_desktop_file_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -372,7 +372,7 @@ def build_cloud_service_roles_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_create_or_update_request( +def build_cloud_services_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -502,7 +502,7 @@ def build_cloud_services_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_get_instance_view_request( +def build_cloud_services_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -749,7 +749,7 @@ def build_cloud_services_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_delete_instances_request( +def build_cloud_services_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -783,7 +783,7 @@ def build_cloud_services_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_walk_update_domain_request( +def build_cloud_services_update_domain_walk_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -818,7 +818,7 @@ def build_cloud_services_update_domain_walk_update_domain_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_get_update_domain_request( +def build_cloud_services_update_domain_get_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -850,7 +850,7 @@ def build_cloud_services_update_domain_get_update_domain_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_list_update_domains_request( +def build_cloud_services_update_domain_list_update_domains_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -881,7 +881,7 @@ def build_cloud_services_update_domain_list_update_domains_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_version_request( +def build_cloud_service_operating_systems_get_os_version_request( # pylint: disable=name-too-long location: str, os_version_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_cloud_service_operating_systems_get_os_version_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_versions_request( +def build_cloud_service_operating_systems_list_os_versions_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +942,7 @@ def build_cloud_service_operating_systems_list_os_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_family_request( +def build_cloud_service_operating_systems_get_os_family_request( # pylint: disable=name-too-long location: str, os_family_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -973,7 +973,7 @@ def build_cloud_service_operating_systems_get_os_family_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_families_request( +def build_cloud_service_operating_systems_list_os_families_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1022,7 +1022,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1146,7 +1146,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1202,7 +1202,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1233,7 +1233,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1269,7 +1269,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1305,7 +1305,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1332,7 +1332,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -1374,7 +1374,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1400,7 +1402,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1430,7 +1432,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1464,7 +1466,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1498,7 +1500,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1561,7 +1563,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1591,7 +1593,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1615,7 +1619,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1751,7 +1755,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1782,7 +1786,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1806,7 +1812,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1960,7 +1966,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1991,7 +1997,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2024,7 +2030,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2055,7 +2061,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -2101,7 +2107,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2136,7 +2142,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2171,7 +2177,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2197,7 +2203,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2237,7 +2243,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2304,7 +2310,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -2352,7 +2358,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2383,7 +2389,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2412,7 +2418,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2444,7 +2450,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -2486,7 +2492,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -2536,7 +2542,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2568,7 +2574,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2599,7 +2605,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2659,7 +2665,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2721,7 +2729,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2859,7 +2867,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2890,7 +2898,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2915,7 +2923,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2940,7 +2948,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3021,7 +3029,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3216,7 +3224,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -3256,7 +3264,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3281,7 +3289,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3306,7 +3314,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3337,7 +3345,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3371,7 +3379,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3405,7 +3413,7 @@ def build_virtual_machines_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3435,7 +3443,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3469,7 +3477,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3503,7 +3511,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3535,7 +3543,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3573,7 +3581,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3604,7 +3612,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3642,7 +3650,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3673,7 +3681,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3703,7 +3711,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3729,7 +3739,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3760,7 +3770,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3791,7 +3801,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3829,7 +3839,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3860,7 +3870,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3891,7 +3901,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3922,7 +3932,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3953,7 +3963,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3984,7 +3994,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4015,7 +4025,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4046,7 +4056,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -4083,7 +4093,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4114,7 +4124,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4298,7 +4308,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4352,7 +4362,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4388,7 +4398,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4424,7 +4434,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4457,7 +4467,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4497,7 +4507,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4527,7 +4537,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4670,7 +4682,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4705,7 +4717,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4740,7 +4752,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4766,7 +4778,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -4806,7 +4818,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4837,7 +4849,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4862,7 +4874,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4887,7 +4899,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4912,7 +4924,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4943,7 +4955,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4984,7 +4996,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5025,7 +5037,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5063,7 +5075,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5105,7 +5117,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5145,7 +5157,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5177,7 +5189,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5203,7 +5215,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5229,7 +5241,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5264,7 +5276,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5298,7 +5310,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5338,7 +5350,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5370,7 +5382,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -5416,7 +5428,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5450,7 +5462,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5476,7 +5488,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5502,7 +5514,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5528,7 +5540,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5570,7 +5582,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5596,7 +5608,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5622,7 +5634,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5657,7 +5669,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5690,7 +5702,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5723,7 +5735,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5750,7 +5764,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5781,7 +5795,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5816,7 +5830,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5851,7 +5865,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5883,7 +5897,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5923,7 +5937,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5956,7 +5970,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5997,7 +6011,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6038,7 +6052,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6076,7 +6090,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6118,7 +6132,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6195,22 +6209,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6220,11 +6233,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6238,14 +6247,6 @@ def begin_delete( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6273,7 +6274,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6282,17 +6283,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6315,7 +6312,6 @@ def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -6334,23 +6330,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6362,13 +6357,9 @@ def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -6382,7 +6373,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6401,22 +6391,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6428,13 +6417,9 @@ def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6456,7 +6441,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.RoleInstance] :raises ~azure.core.exceptions.HttpResponseError: @@ -6478,18 +6462,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6500,14 +6483,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -6517,11 +6500,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6533,10 +6516,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -6554,22 +6533,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6579,11 +6557,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -6598,14 +6572,6 @@ def begin_restart( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6633,7 +6599,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6642,17 +6608,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -6671,22 +6633,21 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6696,11 +6657,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage( @@ -6715,14 +6672,6 @@ def begin_reimage( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6750,7 +6699,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6759,17 +6708,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -6788,22 +6733,21 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6813,11 +6757,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_rebuild( @@ -6833,14 +6773,6 @@ def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6868,7 +6800,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6877,17 +6809,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_remote_desktop_file( @@ -6901,8 +6829,7 @@ def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Iterator of the response bytes or the result of cls(response) + :return: Iterator[bytes] or the result of cls(response) :rtype: Iterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -6920,22 +6847,21 @@ def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6951,10 +6877,6 @@ def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -6988,7 +6910,6 @@ def get( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -7007,22 +6928,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7034,13 +6954,9 @@ def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7053,7 +6969,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudServiceRole] :raises ~azure.core.exceptions.HttpResponseError: @@ -7075,17 +6990,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7096,14 +7010,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -7113,11 +7027,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7129,10 +7043,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -7158,7 +7068,7 @@ def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -7187,7 +7097,7 @@ def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -7195,16 +7105,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7224,10 +7133,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload def begin_create_or_update( self, @@ -7250,14 +7155,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7269,7 +7166,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7282,18 +7179,10 @@ def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7305,7 +7194,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -7315,20 +7204,9 @@ def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudService or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7360,7 +7238,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7370,23 +7248,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -7415,7 +7291,7 @@ def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -7423,16 +7299,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7444,13 +7319,9 @@ def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7473,14 +7344,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7492,7 +7355,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7504,18 +7367,10 @@ def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7527,7 +7382,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -7536,20 +7391,9 @@ def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceUpdate or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.CloudService] @@ -7581,7 +7425,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7591,17 +7435,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -7620,21 +7462,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7644,11 +7485,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -7658,14 +7495,6 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7692,7 +7521,7 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7701,17 +7530,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -7721,7 +7546,6 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -7740,21 +7564,20 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7766,13 +7589,9 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -7784,7 +7603,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7803,21 +7621,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7829,13 +7646,9 @@ def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: @@ -7843,7 +7656,6 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -7865,15 +7677,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7884,14 +7695,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -7901,11 +7712,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7917,8 +7728,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -7927,7 +7736,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -7949,16 +7757,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7969,14 +7776,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -7986,11 +7793,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8002,10 +7809,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -8023,21 +7826,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8047,11 +7849,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8061,14 +7859,6 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8095,7 +7885,7 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8104,17 +7894,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -8133,21 +7919,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8157,11 +7942,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8172,14 +7953,6 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8206,7 +7979,7 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8215,23 +7988,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8260,7 +8029,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -8268,16 +8037,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8287,11 +8055,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8314,14 +8078,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8332,7 +8088,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8344,18 +8100,10 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8366,7 +8114,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -8376,19 +8124,8 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8418,7 +8155,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8427,23 +8164,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8472,7 +8205,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -8480,16 +8213,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8499,11 +8231,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -8527,14 +8255,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8545,7 +8265,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8558,18 +8278,10 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8580,7 +8292,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -8591,19 +8303,8 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8633,7 +8334,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8642,23 +8343,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8687,7 +8384,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -8695,16 +8392,15 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8714,11 +8410,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_rebuild( @@ -8743,14 +8435,6 @@ def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8761,7 +8445,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8775,18 +8459,10 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8797,7 +8473,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -8809,19 +8485,8 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8851,7 +8516,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8860,23 +8525,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8905,7 +8566,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -8913,16 +8574,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8932,11 +8592,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -8959,14 +8615,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8977,7 +8625,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8989,18 +8637,10 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9011,7 +8651,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -9021,19 +8661,8 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9063,7 +8692,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9072,17 +8701,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -9110,7 +8735,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9139,7 +8764,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -9148,16 +8773,15 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9167,11 +8791,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_walk_update_domain( @@ -9199,14 +8819,6 @@ def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9218,7 +8830,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9234,18 +8846,10 @@ def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9257,7 +8861,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -9270,20 +8874,9 @@ def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9314,7 +8907,7 @@ def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9323,17 +8916,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_update_domain( @@ -9351,7 +8940,6 @@ def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -9370,22 +8958,21 @@ def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9397,13 +8984,9 @@ def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -9415,7 +8998,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.UpdateDomain] :raises ~azure.core.exceptions.HttpResponseError: @@ -9437,17 +9019,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9458,14 +9039,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -9475,11 +9056,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9491,10 +9072,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -9525,7 +9102,6 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -9544,21 +9120,20 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9570,13 +9145,9 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OSVersion"]: @@ -9586,7 +9157,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.OSVersion] :raises ~azure.core.exceptions.HttpResponseError: @@ -9608,16 +9178,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9628,14 +9197,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -9645,11 +9214,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9661,10 +9230,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -9674,7 +9239,6 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -9693,21 +9257,20 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9719,13 +9282,9 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OSFamily"]: @@ -9735,7 +9294,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.OSFamily] :raises ~azure.core.exceptions.HttpResponseError: @@ -9757,16 +9315,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9777,14 +9334,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -9794,11 +9351,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9810,10 +9367,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } - class Operations: """ @@ -9839,7 +9392,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -9863,14 +9415,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9881,14 +9432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -9898,11 +9449,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9914,8 +9465,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -9958,7 +9507,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -9969,7 +9517,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9981,11 +9529,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -9996,7 +9543,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -10006,12 +9553,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -10039,7 +9582,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -10047,16 +9590,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10068,13 +9610,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -10097,7 +9635,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -10108,7 +9645,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10120,11 +9657,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -10135,7 +9671,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -10145,12 +9681,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -10178,7 +9710,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -10186,16 +9718,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10207,13 +9738,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -10225,7 +9752,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10244,21 +9770,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10268,11 +9793,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -10282,7 +9803,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -10301,21 +9821,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10327,13 +9846,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -10344,7 +9859,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -10366,16 +9880,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10386,14 +9899,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -10403,11 +9916,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10419,17 +9932,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -10451,16 +9959,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10471,14 +9978,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -10488,11 +9995,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10504,10 +10011,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -10519,7 +10022,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -10541,17 +10043,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10562,14 +10063,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10579,11 +10080,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10595,10 +10096,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -10642,7 +10139,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10653,7 +10149,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10666,11 +10162,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10681,7 +10176,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -10691,12 +10186,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10724,7 +10215,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -10732,16 +10223,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10761,10 +10251,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -10787,7 +10273,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10798,7 +10283,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10811,11 +10296,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10826,7 +10310,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -10836,12 +10320,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -10869,7 +10350,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -10877,16 +10358,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10898,13 +10378,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -10916,7 +10392,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10935,21 +10410,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10959,11 +10433,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -10983,7 +10453,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11002,22 +10471,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11029,19 +10497,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -11065,15 +10528,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11084,14 +10546,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -11101,11 +10563,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11117,10 +10579,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -11129,7 +10587,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -11153,16 +10610,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11173,14 +10629,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -11190,11 +10646,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11206,10 +10662,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -11254,7 +10706,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11265,7 +10716,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11279,11 +10730,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11294,7 +10744,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -11306,12 +10756,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11339,7 +10785,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -11347,16 +10793,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11376,10 +10821,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -11401,7 +10842,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11412,7 +10852,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11424,11 +10864,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11439,7 +10878,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -11449,12 +10888,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11482,7 +10917,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -11490,16 +10925,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11511,13 +10945,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -11529,7 +10959,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11548,21 +10977,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11572,11 +11000,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -11598,7 +11022,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -11617,22 +11040,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11644,13 +11066,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -11659,7 +11077,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -11681,16 +11098,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11701,14 +11117,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -11718,11 +11134,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11734,16 +11150,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -11765,15 +11176,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11784,14 +11194,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -11801,11 +11211,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11817,8 +11227,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -11845,7 +11253,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -11871,7 +11279,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -11880,16 +11288,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11909,10 +11316,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -11937,14 +11340,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -11957,7 +11352,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11971,18 +11366,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -11995,7 +11382,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -12007,19 +11394,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -12052,7 +11428,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12062,24 +11438,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -12105,7 +11479,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -12114,16 +11488,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12135,13 +11508,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -12167,14 +11536,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -12187,7 +11548,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12201,18 +11562,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -12225,7 +11578,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -12237,19 +11590,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] @@ -12282,7 +11624,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12292,17 +11634,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -12321,22 +11661,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12346,11 +11685,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12364,14 +11699,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12399,7 +11726,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12408,17 +11735,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12442,7 +11765,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -12461,23 +11783,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12489,13 +11810,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -12508,7 +11825,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -12530,17 +11846,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12551,14 +11866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -12568,11 +11883,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12584,10 +11899,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -12614,7 +11925,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -12638,15 +11948,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12657,14 +11966,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -12674,11 +11983,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12690,8 +11999,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -12701,7 +12008,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -12725,16 +12031,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12745,14 +12050,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -12762,11 +12067,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12778,10 +12083,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -12803,7 +12104,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -12814,7 +12114,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12826,11 +12126,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -12841,7 +12140,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -12851,12 +12150,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -12884,7 +12179,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -12892,16 +12187,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12921,10 +12215,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -12946,7 +12236,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -12957,7 +12246,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12969,11 +12258,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -12984,7 +12272,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -12994,12 +12282,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -13027,7 +12312,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -13035,16 +12320,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13056,13 +12340,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -13074,7 +12354,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13093,21 +12372,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13117,11 +12395,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -13131,7 +12405,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -13150,21 +12423,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13176,13 +12448,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -13196,7 +12464,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13215,21 +12482,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13241,13 +12507,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13284,7 +12546,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13303,23 +12564,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13331,13 +12591,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -13349,7 +12605,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13368,21 +12623,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13394,13 +12648,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -13428,7 +12678,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13447,7 +12696,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13456,16 +12705,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13477,13 +12725,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -13511,7 +12755,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -13537,7 +12781,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -13546,16 +12790,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13575,10 +12818,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -13605,14 +12844,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13626,7 +12857,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13642,18 +12873,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13667,7 +12890,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -13680,20 +12903,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13727,7 +12939,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13737,24 +12949,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -13780,7 +12990,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -13789,16 +12999,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13810,13 +13019,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -13845,14 +13050,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13866,7 +13063,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13882,18 +13079,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13907,7 +13096,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -13920,20 +13109,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -13967,7 +13145,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13977,17 +13155,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14006,22 +13182,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14031,11 +13206,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14050,14 +13221,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14085,7 +13248,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14094,17 +13257,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14126,7 +13285,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14145,23 +13303,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14173,13 +13330,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14193,7 +13346,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14212,22 +13364,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14239,13 +13390,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -14284,7 +13431,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14303,7 +13449,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -14311,16 +13457,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14332,13 +13477,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14370,7 +13511,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14389,7 +13529,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -14399,16 +13539,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14420,13 +13559,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -14438,7 +13573,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14457,21 +13591,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14483,13 +13616,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -14497,7 +13626,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14516,20 +13644,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14541,13 +13668,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -14561,7 +13684,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14580,22 +13702,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14607,13 +13728,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -14654,7 +13771,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14673,7 +13789,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -14682,16 +13798,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14704,13 +13819,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14747,7 +13858,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14766,7 +13876,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -14777,16 +13887,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14799,13 +13908,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -14820,7 +13925,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14839,22 +13943,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14867,13 +13970,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -14885,7 +13984,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14904,21 +14002,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14931,13 +14028,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -14954,7 +14047,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14973,23 +14065,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15002,13 +14093,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -15038,7 +14125,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -15060,16 +14146,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15080,14 +14165,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -15097,11 +14182,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15113,8 +14198,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -15143,7 +14226,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -15165,16 +14247,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15185,14 +14266,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -15202,11 +14283,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15218,15 +14299,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15252,7 +14329,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15260,16 +14337,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15283,13 +14359,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15313,14 +14385,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15333,7 +14397,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15346,18 +14410,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15370,7 +14426,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15381,19 +14437,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15426,7 +14472,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15438,20 +14484,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15476,7 +14524,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15484,16 +14532,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15513,10 +14560,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15539,14 +14582,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15558,7 +14593,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15571,18 +14606,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15591,7 +14618,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15601,19 +14632,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15645,7 +14665,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15655,20 +14675,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15693,7 +14715,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15701,16 +14723,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15722,13 +14743,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15751,14 +14768,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15770,7 +14779,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15782,18 +14791,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15802,7 +14803,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15811,19 +14816,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] @@ -15855,7 +14849,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15865,17 +14859,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15894,22 +14886,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15919,11 +14910,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15938,14 +14925,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15973,7 +14952,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15982,17 +14961,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -16015,7 +14990,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -16034,22 +15008,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16061,13 +15034,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -16079,7 +15048,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -16098,21 +15066,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16124,13 +15091,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16149,21 +15112,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16173,11 +15135,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16188,14 +15146,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16222,7 +15172,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16231,17 +15181,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16260,21 +15206,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16284,11 +15229,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16299,14 +15240,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16333,7 +15266,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16342,17 +15275,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16369,7 +15298,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16388,21 +15316,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16412,11 +15339,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -16425,7 +15348,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16447,16 +15369,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16467,14 +15388,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16484,11 +15405,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16500,10 +15421,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -16512,7 +15429,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16534,16 +15450,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16554,14 +15469,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16571,11 +15486,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16587,8 +15502,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16600,7 +15513,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16622,17 +15534,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16643,14 +15554,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16660,11 +15571,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16676,10 +15587,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16697,22 +15604,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16722,11 +15628,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16743,14 +15645,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16778,7 +15672,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16787,17 +15681,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16816,21 +15706,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16841,11 +15730,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16855,14 +15740,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16889,7 +15766,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16898,17 +15775,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16927,21 +15800,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16951,11 +15823,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16965,14 +15833,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16999,7 +15859,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17008,17 +15868,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17037,21 +15893,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17061,11 +15916,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17075,14 +15926,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17109,7 +15952,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17118,17 +15961,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17147,21 +15986,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17171,11 +16009,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17185,14 +16019,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17219,7 +16045,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17228,23 +16054,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17273,7 +16095,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17281,16 +16103,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17300,11 +16121,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17328,14 +16145,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17346,7 +16155,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17359,18 +16168,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17381,7 +16182,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -17391,19 +16192,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17433,7 +16224,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17442,17 +16233,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17474,7 +16261,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17493,22 +16279,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17521,13 +16306,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17546,21 +16327,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17570,11 +16350,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17584,14 +16360,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17618,7 +16386,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17627,17 +16395,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17649,7 +16413,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17668,21 +16431,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17692,11 +16454,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17715,21 +16473,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17744,13 +16501,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17762,14 +16515,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17799,7 +16544,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17811,23 +16556,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17853,7 +16596,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17861,16 +16604,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17885,13 +16627,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17916,14 +16654,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17936,7 +16666,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17949,18 +16679,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17973,7 +16695,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -17983,20 +16705,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18029,7 +16740,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18041,20 +16752,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -18079,7 +16792,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18087,16 +16800,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18110,13 +16822,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -18139,14 +16847,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -18158,7 +16858,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18170,18 +16870,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -18190,7 +16882,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18199,19 +16895,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -18243,7 +16928,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18255,17 +16940,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -18295,7 +16978,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18319,16 +17001,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18339,14 +17020,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -18356,11 +17037,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18373,15 +17054,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -18407,7 +17084,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18415,16 +17092,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18444,10 +17120,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -18469,14 +17141,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18489,7 +17153,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18501,18 +17165,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18525,7 +17181,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -18534,20 +17190,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18580,7 +17225,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18590,23 +17235,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -18632,7 +17275,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18640,16 +17283,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18661,13 +17303,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -18690,14 +17328,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18710,7 +17340,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18722,18 +17352,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18746,7 +17368,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -18755,20 +17377,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -18801,7 +17413,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18811,17 +17423,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -18840,22 +17450,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18865,11 +17474,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -18884,14 +17489,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18919,7 +17516,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18928,17 +17525,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -18959,7 +17552,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -18978,22 +17570,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19005,19 +17596,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -19046,7 +17633,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -19054,16 +17641,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19073,11 +17659,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -19104,14 +17686,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19122,7 +17696,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -19137,18 +17711,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19159,7 +17725,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -19171,20 +17737,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -19214,7 +17769,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19223,23 +17778,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -19267,7 +17818,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -19276,16 +17827,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19295,11 +17845,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -19328,14 +17874,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19346,7 +17884,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -19360,21 +17898,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19385,7 +17915,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -19397,23 +17927,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19444,7 +17963,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19453,17 +17972,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -19475,7 +17990,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -19494,21 +18008,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19520,13 +18033,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -19534,7 +18043,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -19558,16 +18066,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19578,14 +18085,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -19595,11 +18102,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19611,17 +18118,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -19645,15 +18147,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19664,14 +18165,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -19681,11 +18182,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19697,8 +18198,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19710,7 +18209,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -19734,17 +18232,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19755,14 +18252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -19772,11 +18269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19788,10 +18285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -19802,7 +18295,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -19826,17 +18318,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19847,14 +18338,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -19864,11 +18355,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19880,15 +18371,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -19919,7 +18406,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -19928,16 +18415,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19947,11 +18433,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -19983,14 +18465,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20001,7 +18475,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -20017,7 +18491,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -20025,14 +18499,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20043,7 +18509,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -20057,24 +18523,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20105,7 +18560,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20114,23 +18569,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -20159,7 +18610,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20167,16 +18618,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20186,11 +18636,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -20215,14 +18661,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20233,7 +18671,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20246,18 +18684,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20268,7 +18698,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -20278,20 +18708,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20321,7 +18740,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20330,23 +18749,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -20375,7 +18790,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20383,16 +18798,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20402,11 +18816,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -20431,14 +18841,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20449,7 +18851,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20462,18 +18864,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20484,7 +18878,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -20494,20 +18888,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20537,7 +18920,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20546,23 +18929,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -20591,7 +18970,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20599,16 +18978,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20618,11 +18996,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -20648,14 +19022,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20666,7 +19032,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20680,18 +19046,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20702,7 +19060,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -20713,20 +19071,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20756,7 +19103,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20765,23 +19112,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -20810,7 +19153,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -20818,16 +19161,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20837,11 +19179,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -20869,14 +19207,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20887,7 +19217,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -20903,18 +19233,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20925,7 +19247,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -20938,20 +19260,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -20981,7 +19292,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20990,23 +19301,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -21032,7 +19339,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21040,16 +19347,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21059,11 +19365,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -21088,14 +19390,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21106,7 +19400,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21119,18 +19413,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21141,7 +19427,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -21151,20 +19437,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -21194,7 +19469,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21203,23 +19478,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -21248,7 +19519,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21256,16 +19527,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21275,11 +19545,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -21305,14 +19571,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21323,7 +19581,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -21337,18 +19595,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21359,7 +19609,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -21371,20 +19621,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -21414,7 +19653,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21423,23 +19662,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -21468,7 +19703,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21476,16 +19711,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21495,11 +19729,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -21525,14 +19755,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21543,7 +19765,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -21557,18 +19779,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21579,7 +19793,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -21590,20 +19804,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -21633,7 +19836,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21642,20 +19845,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -21668,7 +19867,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -21687,22 +19885,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21714,13 +19911,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -21745,7 +19938,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21756,7 +19948,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21769,11 +19961,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21784,7 +19975,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -21795,13 +19986,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21829,7 +20017,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21837,16 +20025,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21856,17 +20043,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -21892,7 +20075,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -21900,16 +20083,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21919,11 +20101,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -21947,14 +20125,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21965,7 +20135,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21978,18 +20148,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22000,7 +20162,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -22011,19 +20173,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -22053,7 +20205,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22062,17 +20214,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -22102,7 +20250,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -22124,16 +20271,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22144,14 +20290,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -22161,11 +20307,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22177,8 +20323,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -22201,7 +20345,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22226,7 +20370,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22234,16 +20378,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22263,10 +20406,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -22288,14 +20427,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22306,7 +20437,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22318,18 +20449,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22337,7 +20460,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -22346,19 +20469,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22389,7 +20501,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22399,20 +20511,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22437,7 +20551,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22445,16 +20559,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22474,10 +20587,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -22499,14 +20608,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22517,7 +20618,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22529,18 +20630,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22548,7 +20641,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -22557,19 +20654,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22600,7 +20686,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22610,17 +20696,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -22639,21 +20723,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22663,11 +20746,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22677,14 +20756,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22711,7 +20782,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22720,17 +20791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -22744,7 +20811,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -22763,22 +20829,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22790,13 +20855,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -22804,7 +20865,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22826,16 +20886,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22846,14 +20905,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -22863,11 +20922,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22879,16 +20938,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22910,15 +20964,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22929,14 +20982,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -22946,11 +20999,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22962,8 +21015,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23009,7 +21060,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23020,7 +21070,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23035,11 +21085,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23050,7 +21099,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23062,12 +21111,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23095,7 +21140,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23103,16 +21148,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23133,10 +21177,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23159,7 +21199,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23170,7 +21209,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23183,11 +21222,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23198,7 +21236,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23208,12 +21246,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23241,7 +21276,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23249,16 +21284,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23271,13 +21305,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -23296,21 +21326,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23321,11 +21350,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23338,14 +21363,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23372,7 +21389,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23381,17 +21398,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23413,7 +21426,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23432,22 +21444,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23460,13 +21471,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -23474,7 +21481,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23498,16 +21504,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23518,14 +21523,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -23535,11 +21540,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23552,17 +21557,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23586,15 +21586,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23605,14 +21604,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -23622,11 +21621,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23639,8 +21638,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -23667,7 +21664,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -23693,7 +21690,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -23702,16 +21699,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23724,13 +21720,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -23757,14 +21749,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -23777,7 +21761,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23792,18 +21776,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -23816,7 +21792,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -23829,19 +21805,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RestorePoint] @@ -23874,7 +21839,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23884,17 +21849,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -23913,22 +21876,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23939,11 +21901,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23957,14 +21915,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23992,7 +21942,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24001,17 +21951,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24025,7 +21971,6 @@ def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24044,22 +21989,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24072,16 +22016,12 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -24106,7 +22046,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -24132,7 +22072,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -24141,16 +22081,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24170,10 +22109,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -24201,14 +22136,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24222,7 +22149,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24238,18 +22165,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24263,7 +22182,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -24276,20 +22195,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24323,7 +22231,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24333,24 +22241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -24376,7 +22282,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -24385,16 +22291,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24414,10 +22319,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -24445,14 +22346,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24466,7 +22359,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24482,18 +22375,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24507,7 +22392,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -24520,20 +22405,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24567,7 +22442,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24577,17 +22452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -24606,22 +22479,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24631,11 +22503,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24650,14 +22518,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24685,7 +22545,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24694,17 +22554,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24726,7 +22582,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -24745,23 +22600,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24773,13 +22627,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24791,7 +22641,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -24815,17 +22664,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24836,14 +22684,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -24853,11 +22701,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24869,12 +22717,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -24911,21 +22755,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24935,11 +22778,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -24949,14 +22788,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24983,7 +22814,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24992,17 +22823,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -25021,21 +22848,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25045,11 +22871,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -25063,14 +22885,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25097,7 +22911,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25106,17 +22920,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -25135,21 +22945,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25159,11 +22968,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -25177,14 +22982,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25211,7 +23008,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25220,17 +23017,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -25242,7 +23035,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -25261,21 +23053,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25287,16 +23078,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -25322,7 +23109,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -25348,7 +23135,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25358,16 +23145,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25388,10 +23174,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -25421,14 +23203,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25443,7 +23217,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25460,18 +23234,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25486,7 +23252,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -25500,20 +23266,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25548,7 +23303,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25558,17 +23313,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -25576,7 +23329,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -25602,7 +23355,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25612,16 +23365,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25634,13 +23386,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25671,14 +23419,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25693,7 +23433,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25710,18 +23450,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25736,7 +23468,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -25750,20 +23482,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -25798,7 +23520,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25808,17 +23530,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -25837,23 +23557,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25864,11 +23583,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25884,14 +23599,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25920,7 +23627,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25929,17 +23636,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25964,7 +23667,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -25983,7 +23685,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25991,16 +23693,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26013,13 +23714,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -26041,7 +23738,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -26060,23 +23756,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26089,13 +23784,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -26123,7 +23814,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -26152,7 +23845,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26161,16 +23854,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26180,11 +23872,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -26212,14 +23900,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26231,7 +23911,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -26246,18 +23926,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26269,7 +23941,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -26281,20 +23955,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -26325,7 +23989,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26334,17 +23998,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26363,22 +24023,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26388,11 +24047,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -26407,14 +24062,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26442,7 +24089,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26451,17 +24098,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26480,22 +24123,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26505,11 +24147,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -26525,14 +24163,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26560,7 +24190,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26569,24 +24199,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -26612,7 +24238,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26621,16 +24247,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26650,10 +24275,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -26680,14 +24301,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -26701,7 +24314,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26717,18 +24330,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -26742,7 +24347,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -26755,19 +24360,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -26801,7 +24395,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26811,17 +24405,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -26846,23 +24438,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26872,11 +24463,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26899,14 +24486,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26935,7 +24514,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26944,17 +24523,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -26978,7 +24553,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -26997,23 +24571,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27025,13 +24598,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -27045,7 +24614,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -27064,22 +24632,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27091,13 +24658,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -27126,7 +24689,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -27150,7 +24712,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -27158,12 +24720,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27174,14 +24735,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -27191,11 +24752,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27207,10 +24768,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -27234,23 +24791,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27260,11 +24816,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -27290,14 +24842,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27326,7 +24870,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27335,17 +24879,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27364,22 +24904,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27389,11 +24928,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -27407,14 +24942,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27442,7 +24969,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27451,17 +24978,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27480,22 +25003,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27505,11 +25027,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -27523,14 +25041,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27558,7 +25068,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27567,17 +25077,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27596,22 +25102,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27621,11 +25126,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -27640,14 +25141,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27675,7 +25168,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27684,17 +25177,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -27720,7 +25209,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -27739,23 +25227,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27768,13 +25255,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -27793,22 +25276,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27818,11 +25300,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -27836,14 +25314,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27871,7 +25341,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27880,17 +25350,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -27904,7 +25370,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -27923,22 +25388,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27948,18 +25412,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -27985,7 +25445,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27994,16 +25454,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28017,13 +25476,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -28049,14 +25504,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -28069,7 +25516,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28083,18 +25530,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -28107,7 +25546,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -28119,19 +25558,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_03_01.models.RunCommandResult] @@ -28164,7 +25592,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28176,17 +25604,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -28210,7 +25636,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -28235,23 +25661,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28265,13 +25690,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -28293,14 +25714,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28310,7 +25723,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -28319,18 +25732,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28340,7 +25745,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -28348,19 +25753,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28392,7 +25787,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28404,20 +25799,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -28442,23 +25835,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28472,13 +25864,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -28499,14 +25887,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28516,7 +25896,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28524,18 +25904,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28545,7 +25917,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28553,19 +25925,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28597,7 +25958,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28609,17 +25970,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -28648,7 +26007,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -28672,16 +26030,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28692,14 +26049,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -28709,11 +26066,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28725,10 +26082,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -28737,7 +26090,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -28756,21 +26108,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28782,20 +26133,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28821,7 +26168,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -28830,16 +26177,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28860,10 +26206,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -28890,14 +26232,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28911,7 +26245,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28927,18 +26261,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28952,7 +26278,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -28965,19 +26291,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29011,7 +26326,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29021,24 +26336,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29064,7 +26377,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -29073,16 +26386,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29095,13 +26407,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -29129,14 +26437,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29150,7 +26450,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29166,18 +26466,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29191,7 +26483,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -29204,19 +26496,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29250,7 +26532,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29260,17 +26542,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -29289,22 +26569,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29315,11 +26594,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -29334,14 +26609,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29369,7 +26636,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -29378,17 +26645,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -29410,7 +26673,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -29429,23 +26691,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29458,13 +26719,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -29478,7 +26735,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29502,18 +26758,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29524,14 +26779,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -29541,11 +26796,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29558,12 +26813,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -29589,7 +26840,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29615,7 +26866,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29625,16 +26876,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29655,10 +26905,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -29687,14 +26933,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29709,7 +26947,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29726,18 +26964,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29752,7 +26982,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -29766,19 +26996,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29813,7 +27032,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29823,17 +27042,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -29841,7 +27058,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29867,7 +27084,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29877,16 +27094,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29899,13 +27115,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -29935,14 +27147,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29957,7 +27161,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29974,18 +27178,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30000,7 +27196,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -30014,19 +27210,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30061,7 +27247,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30071,17 +27257,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -30100,23 +27284,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30127,11 +27310,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -30147,14 +27326,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -30183,7 +27354,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -30192,17 +27363,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -30227,7 +27394,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -30246,7 +27412,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -30254,16 +27420,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30276,13 +27441,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -30304,7 +27465,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -30328,19 +27488,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -30351,14 +27510,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -30368,11 +27527,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30384,7 +27543,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_compute_management_client.py index b62933d29d70..8c9cb16c13b6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -167,7 +169,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -259,7 +279,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -279,7 +299,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_configuration.py index ea52c3f4fefc..fd62180d002b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_metadata.json index c995022f0f0a..f61f52fe32fc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_compute_management_client.py index 82ade0f7b7c7..cf57d25f42f3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -169,7 +171,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -261,7 +281,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-04-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -281,7 +303,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_configuration.py index 321a798a3c95..69fcab1a0a58 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-04-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/operations/_operations.py index e6550197776c..d77fdd0daa38 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -270,7 +270,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -294,14 +293,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -312,14 +310,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -329,11 +327,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -345,8 +343,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -389,7 +385,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -400,7 +395,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -412,11 +407,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -427,7 +421,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -437,12 +431,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -470,7 +460,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -478,16 +468,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -499,13 +488,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -528,7 +513,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -539,7 +523,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -551,11 +535,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -566,7 +549,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -576,12 +559,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -609,7 +588,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -617,16 +596,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -638,13 +616,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -656,7 +630,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -675,21 +648,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -699,11 +671,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -713,7 +681,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -732,21 +699,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -758,13 +724,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -775,7 +737,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet] @@ -798,16 +759,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -818,14 +778,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -835,11 +795,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -851,17 +811,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet] @@ -884,16 +839,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -904,14 +858,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -921,11 +875,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -937,10 +891,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -952,7 +902,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] @@ -975,17 +924,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -996,14 +944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -1013,11 +961,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1029,10 +977,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1076,7 +1020,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1087,7 +1030,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1100,11 +1043,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1115,7 +1057,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1125,12 +1067,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1158,7 +1096,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1166,16 +1104,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1195,10 +1132,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1221,7 +1154,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1232,7 +1164,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1245,11 +1177,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1260,7 +1191,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1270,12 +1201,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1303,7 +1231,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1311,16 +1239,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1332,13 +1259,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1350,7 +1273,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1369,21 +1291,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1393,11 +1314,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1417,7 +1334,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1436,22 +1352,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1463,19 +1378,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1499,15 +1409,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1518,14 +1427,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1535,11 +1444,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1551,10 +1460,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1563,7 +1468,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1587,16 +1491,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1607,14 +1510,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1624,11 +1527,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1640,10 +1543,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1688,7 +1587,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1699,7 +1597,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1713,11 +1611,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1728,7 +1625,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1740,12 +1637,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1773,7 +1666,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1781,16 +1674,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,10 +1702,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1835,7 +1723,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1846,7 +1733,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1858,11 +1745,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1873,7 +1759,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1883,12 +1769,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1916,7 +1798,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1924,16 +1806,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1945,13 +1826,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1963,7 +1840,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1982,21 +1858,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2006,11 +1881,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -2032,7 +1903,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -2051,22 +1921,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2078,13 +1947,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2095,7 +1960,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup] @@ -2118,16 +1982,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2138,14 +2001,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2155,11 +2018,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2171,16 +2034,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup] @@ -2203,15 +2061,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2222,14 +2079,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2239,11 +2096,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2255,8 +2112,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2283,7 +2138,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2309,7 +2164,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2318,16 +2173,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2347,10 +2201,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2375,14 +2225,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2396,7 +2238,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2410,18 +2252,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2435,7 +2269,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2447,19 +2281,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2493,7 +2316,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2503,24 +2326,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2546,7 +2367,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2555,16 +2376,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2576,13 +2396,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2608,14 +2424,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2629,7 +2437,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2643,18 +2451,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2668,7 +2468,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2680,19 +2480,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2726,7 +2515,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2736,17 +2525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2765,22 +2552,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2790,11 +2576,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2808,14 +2590,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2843,7 +2617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2852,17 +2626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2886,7 +2656,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2905,23 +2674,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2933,13 +2701,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2952,7 +2716,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -2975,17 +2738,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2996,14 +2758,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -3013,11 +2775,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3029,10 +2791,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -3059,7 +2817,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3083,15 +2840,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3102,14 +2858,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3119,11 +2875,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3135,8 +2891,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -3146,7 +2900,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3170,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3190,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3207,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3223,10 +2975,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -3248,7 +2996,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3259,7 +3006,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3271,11 +3018,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3286,7 +3032,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -3296,12 +3042,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3329,7 +3071,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3337,16 +3079,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3366,10 +3107,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -3391,7 +3128,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3402,7 +3138,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3414,11 +3150,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3429,7 +3164,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -3439,12 +3174,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3472,7 +3204,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3480,16 +3212,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3501,13 +3232,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -3519,7 +3246,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3538,21 +3264,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3562,11 +3287,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -3578,7 +3299,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3597,21 +3317,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3623,13 +3342,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -3643,7 +3358,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3662,21 +3376,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3688,13 +3401,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -3731,7 +3440,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3750,23 +3458,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3778,13 +3485,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3796,7 +3499,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3815,21 +3517,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3841,13 +3542,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3875,7 +3572,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3894,7 +3590,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3903,16 +3599,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3924,13 +3619,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3958,7 +3649,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3984,7 +3675,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3993,16 +3684,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4022,10 +3712,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4052,14 +3738,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4073,7 +3751,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4089,18 +3767,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4114,7 +3784,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4127,20 +3797,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4174,7 +3833,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4184,24 +3843,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4227,7 +3884,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4236,16 +3893,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4257,13 +3913,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -4292,14 +3944,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4313,7 +3957,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4329,18 +3973,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4354,7 +3990,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4367,20 +4003,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4414,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4424,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4453,22 +4076,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4478,11 +4100,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4497,14 +4115,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4532,7 +4142,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4541,17 +4151,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4573,7 +4179,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4592,23 +4197,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4620,13 +4224,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4640,7 +4240,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4659,22 +4258,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4686,13 +4284,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4731,7 +4325,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4750,7 +4343,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4758,16 +4351,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4779,13 +4371,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4817,7 +4405,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,7 +4423,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4846,16 +4433,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4867,13 +4453,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4885,7 +4467,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4904,21 +4485,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4930,13 +4510,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4944,7 +4520,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4963,20 +4538,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4988,13 +4562,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5008,7 +4578,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5027,22 +4596,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5054,13 +4622,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -5101,7 +4665,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5120,7 +4683,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5129,16 +4692,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5150,13 +4712,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -5193,7 +4751,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5212,7 +4769,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5223,16 +4780,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5244,13 +4800,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -5265,7 +4817,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5284,22 +4835,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5311,13 +4861,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -5329,7 +4875,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5348,21 +4893,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5374,13 +4918,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5397,7 +4937,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5416,23 +4955,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5444,13 +4982,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5480,7 +5014,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5502,16 +5035,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5522,14 +5054,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5539,11 +5071,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5555,8 +5087,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -5585,7 +5115,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -5608,16 +5137,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5628,14 +5156,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5645,11 +5173,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5661,15 +5189,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -5695,7 +5219,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5703,16 +5227,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5726,13 +5249,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -5756,14 +5275,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5776,7 +5287,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5789,18 +5300,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5813,7 +5316,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5824,19 +5327,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5869,7 +5362,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5881,20 +5374,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5919,7 +5414,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5927,16 +5422,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5956,10 +5450,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -5982,14 +5472,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6002,7 +5484,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6015,18 +5497,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6036,7 +5510,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -6046,19 +5524,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6091,7 +5558,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6101,20 +5568,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6139,7 +5608,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6147,16 +5616,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6168,13 +5636,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6197,14 +5661,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6217,7 +5673,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6229,18 +5685,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6250,7 +5698,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -6259,19 +5711,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6304,7 +5745,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6314,17 +5755,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6343,22 +5782,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6368,11 +5806,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6387,14 +5821,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6422,7 +5848,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6431,17 +5857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6464,7 +5886,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6483,22 +5904,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6510,13 +5930,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -6528,7 +5944,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6547,21 +5962,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6573,13 +5987,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6598,21 +6008,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6622,11 +6031,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -6639,14 +6044,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6673,7 +6070,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6682,17 +6079,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6711,21 +6104,20 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6735,11 +6127,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -6750,14 +6138,6 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6784,7 +6164,7 @@ async def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6793,17 +6173,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -6820,7 +6196,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6839,21 +6214,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6863,11 +6237,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -6876,7 +6246,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -6899,16 +6268,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6919,14 +6287,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6936,11 +6304,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6952,10 +6320,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -6964,7 +6328,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -6987,16 +6350,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7007,14 +6369,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7024,11 +6386,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7040,8 +6402,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7053,7 +6413,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] @@ -7076,17 +6435,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7097,14 +6455,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7114,11 +6472,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7130,10 +6488,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -7151,22 +6505,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7176,11 +6529,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7197,14 +6546,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7232,7 +6573,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7241,17 +6582,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7270,21 +6607,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7294,11 +6630,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7308,14 +6640,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7342,7 +6666,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7351,17 +6675,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7380,21 +6700,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7404,11 +6723,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7418,14 +6733,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7452,7 +6759,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7461,17 +6768,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7490,21 +6793,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7514,11 +6816,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7528,14 +6826,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7562,7 +6852,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7571,17 +6861,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7600,21 +6886,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7624,11 +6909,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7638,14 +6919,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7672,7 +6945,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7681,23 +6954,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7726,7 +6995,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7734,16 +7003,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7753,11 +7021,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -7781,14 +7045,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7799,7 +7055,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7812,18 +7068,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7834,7 +7082,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -7844,19 +7092,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7886,7 +7124,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7895,17 +7133,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7927,7 +7161,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7946,22 +7179,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7973,13 +7205,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7998,21 +7226,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8022,11 +7249,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8038,14 +7261,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8072,7 +7287,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8081,17 +7296,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8103,7 +7314,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8122,21 +7332,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8146,11 +7355,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8169,21 +7374,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8197,13 +7401,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -8215,14 +7415,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -8252,7 +7444,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8264,23 +7456,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -8306,7 +7496,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8314,16 +7504,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8337,13 +7526,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -8368,14 +7553,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8388,7 +7565,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8401,18 +7578,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8425,7 +7594,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -8435,20 +7604,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8481,7 +7639,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8493,20 +7651,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -8531,7 +7691,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8539,16 +7699,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8562,13 +7721,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8591,14 +7746,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8611,7 +7758,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8623,18 +7770,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8644,7 +7783,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -8653,19 +7796,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8698,7 +7830,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8710,17 +7842,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -8750,7 +7880,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8774,16 +7903,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8794,14 +7922,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -8811,11 +7939,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8827,15 +7955,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8861,7 +7985,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8869,16 +7993,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8898,10 +8021,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -8923,14 +8042,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8943,7 +8054,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8955,18 +8066,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8979,7 +8082,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8988,20 +8091,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9034,7 +8126,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9044,23 +8136,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9086,7 +8176,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9094,16 +8184,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9115,13 +8204,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9144,14 +8229,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9164,7 +8241,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9176,18 +8253,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9200,7 +8269,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9209,20 +8278,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9255,7 +8314,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9265,17 +8324,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9294,22 +8351,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9319,11 +8375,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9338,14 +8390,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9373,7 +8417,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9382,17 +8426,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9413,7 +8453,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9432,22 +8471,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9459,19 +8497,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9500,7 +8534,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9508,16 +8542,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9527,11 +8560,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -9558,14 +8587,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9576,7 +8597,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9591,18 +8612,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9613,7 +8626,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9625,20 +8638,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9668,7 +8670,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9677,23 +8679,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9721,7 +8719,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9730,16 +8728,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9749,11 +8746,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -9782,14 +8775,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9800,7 +8785,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -9814,21 +8799,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9839,7 +8816,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9851,23 +8828,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9898,7 +8864,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9907,17 +8873,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -9929,7 +8891,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9948,21 +8909,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9974,13 +8934,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -9988,7 +8944,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10012,16 +8967,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10032,14 +8986,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10049,11 +9003,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10065,17 +9019,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10099,15 +9048,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10118,14 +9066,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10135,11 +9083,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10151,8 +9099,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10164,7 +9110,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10188,17 +9133,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10209,14 +9153,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10226,11 +9170,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10242,10 +9186,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10256,7 +9196,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10280,17 +9219,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10301,14 +9239,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10318,11 +9256,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10334,15 +9272,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10373,7 +9307,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10382,16 +9316,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10401,11 +9334,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -10437,14 +9366,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10455,7 +9376,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -10471,7 +9392,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -10479,14 +9400,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10497,7 +9410,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10511,24 +9424,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10559,7 +9461,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10568,23 +9470,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10613,7 +9511,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10621,16 +9519,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10640,11 +9537,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -10669,14 +9562,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10687,7 +9572,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10700,18 +9585,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10722,7 +9599,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -10732,20 +9609,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10775,7 +9641,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10784,23 +9650,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10829,7 +9691,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10837,16 +9699,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10856,11 +9717,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10885,14 +9742,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10903,7 +9752,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10916,18 +9765,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10938,7 +9779,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -10948,20 +9789,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10991,7 +9821,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11000,23 +9830,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11045,7 +9871,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11053,16 +9879,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11072,11 +9897,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -11102,14 +9923,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11120,7 +9933,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11134,18 +9947,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11156,7 +9961,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11167,20 +9972,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11210,7 +10004,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11219,23 +10013,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11264,7 +10054,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11272,16 +10062,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11291,11 +10080,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -11323,14 +10108,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11341,7 +10118,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11357,18 +10134,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11379,7 +10148,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11392,20 +10161,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11435,7 +10193,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11444,23 +10202,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11486,7 +10240,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11494,16 +10248,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11513,11 +10266,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -11542,14 +10291,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11560,7 +10301,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11573,18 +10314,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11595,7 +10328,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11605,20 +10338,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11648,7 +10370,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11657,23 +10379,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11702,7 +10420,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11710,16 +10428,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11729,11 +10446,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11759,14 +10472,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11777,7 +10482,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11791,18 +10496,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11813,7 +10510,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11825,20 +10522,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11868,7 +10554,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11877,23 +10563,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11922,7 +10604,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11930,16 +10612,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11949,11 +10630,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -11979,14 +10656,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11997,7 +10666,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12011,18 +10680,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12033,7 +10694,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12044,20 +10705,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12087,7 +10737,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12096,20 +10746,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12122,7 +10768,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12141,22 +10786,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12168,13 +10812,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -12199,7 +10839,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12210,7 +10849,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12223,11 +10862,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12238,7 +10876,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -12249,13 +10887,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_04_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_04_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12283,7 +10918,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12291,16 +10926,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12310,17 +10944,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12346,7 +10976,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12354,16 +10984,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12373,11 +11002,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -12401,14 +11026,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12419,7 +11036,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12432,18 +11049,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12454,7 +11063,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -12465,19 +11074,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12507,7 +11106,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12516,17 +11115,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -12556,7 +11151,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] @@ -12579,16 +11173,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12599,14 +11192,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12616,11 +11209,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12632,8 +11225,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12656,7 +11247,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12681,7 +11272,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12689,16 +11280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12718,10 +11308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -12743,14 +11329,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12761,7 +11339,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12773,18 +11351,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12792,7 +11362,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -12801,19 +11371,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12844,7 +11403,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12854,20 +11413,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12892,7 +11453,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12900,16 +11461,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12929,10 +11489,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -12954,14 +11510,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12972,7 +11520,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12984,18 +11532,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13003,7 +11543,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -13012,19 +11556,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13055,7 +11588,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13065,17 +11598,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -13094,21 +11625,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13118,11 +11648,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -13132,14 +11658,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13166,7 +11684,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13175,17 +11693,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13199,7 +11713,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -13218,22 +11731,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13245,13 +11757,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -13259,7 +11767,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13281,16 +11788,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13301,14 +11807,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13318,11 +11824,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13334,16 +11840,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13365,15 +11866,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13384,14 +11884,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13401,11 +11901,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13417,8 +11917,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -13464,7 +11962,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13475,7 +11972,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13490,11 +11987,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13505,7 +12001,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -13517,12 +12013,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13550,7 +12042,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13558,16 +12050,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13587,10 +12078,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -13613,7 +12100,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13624,7 +12110,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13637,11 +12123,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13652,7 +12137,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -13662,12 +12147,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13695,7 +12177,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13703,16 +12185,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13724,13 +12205,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -13749,21 +12226,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13773,11 +12249,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13790,14 +12262,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13824,7 +12288,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13833,17 +12297,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13865,7 +12325,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13884,22 +12343,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13911,13 +12369,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -13925,7 +12379,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -13949,16 +12402,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13969,14 +12421,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -13986,11 +12438,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14002,17 +12454,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -14036,15 +12483,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14055,14 +12501,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -14072,11 +12518,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14088,8 +12534,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -14116,7 +12560,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -14142,7 +12586,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -14151,16 +12595,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14172,13 +12615,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -14205,14 +12644,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -14225,7 +12656,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14240,18 +12671,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -14264,7 +12687,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -14277,19 +12700,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -14322,7 +12734,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14332,17 +12744,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -14361,22 +12771,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14386,11 +12795,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14404,14 +12809,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14439,7 +12836,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14448,17 +12845,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14472,7 +12865,6 @@ async def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -14491,22 +12883,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14518,13 +12909,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -14570,7 +12957,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14581,7 +12967,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14595,11 +12981,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14610,7 +12995,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -14622,12 +13007,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14655,7 +13036,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14663,16 +13044,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14692,10 +13072,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -14719,7 +13095,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14730,7 +13105,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14744,11 +13119,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14759,7 +13133,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -14770,12 +13144,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14803,7 +13174,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14811,16 +13182,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14832,13 +13202,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14853,7 +13219,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14872,21 +13237,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14896,11 +13260,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14924,7 +13284,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14943,22 +13302,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14970,13 +13328,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -14997,7 +13351,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15021,17 +13374,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15042,14 +13394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15059,11 +13411,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15075,10 +13427,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -15092,7 +13440,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15116,16 +13463,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15136,14 +13482,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15153,11 +13499,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15169,10 +13515,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -15199,7 +13541,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -15225,7 +13567,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15234,16 +13576,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15263,10 +13604,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -15293,14 +13630,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15314,7 +13643,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15330,18 +13659,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15355,7 +13676,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -15369,19 +13690,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15415,7 +13725,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15425,24 +13735,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -15468,7 +13776,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15477,16 +13785,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15500,13 +13807,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15532,14 +13835,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15553,7 +13848,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15567,18 +13862,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15592,7 +13879,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -15604,19 +13891,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15650,7 +13926,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15660,17 +13936,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -15693,22 +13967,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15718,11 +13991,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15742,14 +14011,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15777,7 +14038,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15786,17 +14047,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15821,7 +14078,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -15840,23 +14096,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15868,13 +14123,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -15887,7 +14138,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.CapacityReservation] @@ -15910,17 +14160,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15931,14 +14180,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -15948,11 +14197,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15964,12 +14213,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -15994,7 +14239,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16020,7 +14265,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16029,16 +14274,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16058,10 +14302,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -16089,14 +14329,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16110,7 +14342,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16126,18 +14358,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16151,7 +14375,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -16164,20 +14388,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16211,7 +14424,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16221,24 +14434,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16264,7 +14475,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16273,16 +14484,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16302,10 +14512,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -16333,14 +14539,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16354,7 +14552,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16370,18 +14568,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16395,7 +14585,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -16408,20 +14598,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16455,7 +14635,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16465,17 +14645,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -16494,22 +14672,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16519,11 +14696,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16538,14 +14711,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16573,7 +14738,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16582,17 +14747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16614,7 +14775,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -16633,23 +14793,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16661,13 +14820,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -16679,7 +14834,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16703,17 +14857,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16724,14 +14877,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -16741,11 +14894,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16757,12 +14910,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16799,21 +14948,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16823,11 +14971,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -16839,14 +14983,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16873,7 +15009,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16882,17 +15018,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16911,21 +15043,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16935,11 +15066,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -16953,14 +15080,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16987,7 +15106,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16996,17 +15115,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17025,21 +15140,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17049,11 +15163,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -17067,14 +15177,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17101,7 +15203,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17110,17 +15212,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -17132,7 +15230,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -17151,21 +15248,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17177,16 +15273,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17212,7 +15304,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17238,7 +15330,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17248,16 +15340,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17277,10 +15368,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -17310,14 +15397,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17332,7 +15411,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17349,18 +15428,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17375,7 +15446,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -17389,20 +15460,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17437,7 +15497,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17447,17 +15507,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -17465,7 +15523,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17491,7 +15549,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17501,16 +15559,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17522,13 +15579,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -17559,14 +15612,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17581,7 +15626,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17598,18 +15643,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17624,7 +15661,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -17638,20 +15675,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17686,7 +15713,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17696,17 +15723,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -17725,23 +15750,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17751,11 +15775,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17771,14 +15791,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17807,7 +15819,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17816,17 +15828,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17851,7 +15859,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -17870,7 +15877,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17878,16 +15885,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17899,13 +15905,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -17927,7 +15929,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17946,23 +15947,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17974,13 +15974,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -18008,7 +16004,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -18037,7 +16035,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18046,16 +16044,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18065,11 +16062,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -18097,14 +16090,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18116,7 +16101,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18131,18 +16116,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18154,7 +16131,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -18166,20 +16145,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18210,7 +16179,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18219,17 +16188,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18248,22 +16213,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18273,11 +16237,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -18292,14 +16252,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18327,7 +16279,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18336,17 +16288,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18365,22 +16313,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18390,11 +16337,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -18410,14 +16353,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18445,7 +16380,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18454,24 +16389,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -18497,7 +16428,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18506,16 +16437,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18535,10 +16465,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -18565,14 +16491,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18586,7 +16504,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18602,18 +16520,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18627,7 +16537,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -18640,19 +16550,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18686,7 +16585,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18696,17 +16595,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -18731,23 +16628,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18757,11 +16653,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18784,14 +16676,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18820,7 +16704,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18829,17 +16713,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18863,7 +16743,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -18882,23 +16761,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18910,13 +16788,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -18930,7 +16804,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -18949,22 +16822,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18976,13 +16848,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19011,7 +16879,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19035,7 +16902,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -19043,12 +16910,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19059,14 +16925,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -19076,11 +16942,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19092,10 +16958,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -19119,23 +16981,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19145,11 +17006,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -19175,14 +17032,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19211,7 +17060,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19220,17 +17069,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19249,22 +17094,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19274,11 +17118,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -19292,14 +17132,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19327,7 +17159,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19336,17 +17168,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19365,22 +17193,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19390,11 +17217,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -19408,14 +17231,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19443,7 +17258,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19452,17 +17267,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19481,22 +17292,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19506,11 +17316,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -19525,14 +17331,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19560,7 +17358,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19569,17 +17367,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -19605,7 +17399,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -19624,23 +17417,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19652,13 +17444,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19677,22 +17465,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19702,11 +17489,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -19720,14 +17503,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19755,7 +17530,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19764,17 +17539,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -19788,7 +17559,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19807,22 +17577,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19832,18 +17601,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -19869,7 +17634,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19878,16 +17643,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19901,13 +17665,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -19933,14 +17693,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -19954,7 +17706,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19968,18 +17720,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -19993,7 +17737,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -20005,19 +17749,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -20051,7 +17784,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20063,17 +17796,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -20097,7 +17828,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20122,23 +17853,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20152,13 +17882,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20180,14 +17906,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20197,7 +17915,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20206,18 +17924,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20227,7 +17937,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20235,19 +17945,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20279,7 +17979,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20292,20 +17992,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20330,23 +18028,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20360,13 +18057,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20387,14 +18080,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20404,7 +18089,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20412,18 +18097,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20433,7 +18110,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20441,19 +18118,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20485,7 +18151,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20498,17 +18164,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20537,7 +18201,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20561,16 +18224,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20581,14 +18243,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20598,11 +18260,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20614,10 +18276,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20626,7 +18284,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20645,21 +18302,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20671,20 +18327,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20710,7 +18362,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20719,16 +18371,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20748,10 +18399,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20778,14 +18425,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20799,7 +18438,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20815,18 +18454,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20840,7 +18471,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -20853,19 +18484,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20899,7 +18519,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20909,24 +18529,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20952,7 +18570,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20961,16 +18579,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20982,13 +18599,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21016,14 +18629,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21037,7 +18642,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21053,18 +18658,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21078,7 +18675,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21091,19 +18688,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21137,7 +18724,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21147,17 +18734,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21176,22 +18761,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21201,11 +18785,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21220,14 +18800,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21255,7 +18827,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21264,17 +18836,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21296,7 +18864,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21315,23 +18882,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21343,13 +18909,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21363,7 +18925,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21387,18 +18948,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21409,14 +18969,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21426,11 +18986,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21442,12 +19002,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21473,7 +19029,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21499,7 +19055,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21509,16 +19065,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21538,10 +19093,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21570,14 +19121,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21592,7 +19135,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21609,18 +19152,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21635,7 +19170,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21649,19 +19184,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21696,7 +19220,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21706,17 +19230,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21724,7 +19246,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21750,7 +19272,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21760,16 +19282,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21781,13 +19302,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21817,14 +19334,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21839,7 +19348,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21856,18 +19365,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21882,7 +19383,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -21896,19 +19397,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21943,7 +19434,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21953,17 +19444,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -21982,23 +19471,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22008,11 +19496,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22028,14 +19512,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22064,7 +19540,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22073,17 +19549,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22108,7 +19580,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22127,7 +19598,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22135,16 +19606,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22156,13 +19626,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22184,7 +19650,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22208,19 +19673,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22231,14 +19695,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22248,11 +19712,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22264,10 +19728,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } - class DisksOperations: """ @@ -22290,7 +19750,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -22315,7 +19775,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -22323,16 +19783,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22352,10 +19811,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -22379,14 +19834,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22397,7 +19844,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22411,18 +19858,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22430,7 +19869,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -22441,19 +19880,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_04_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_04_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22484,7 +19912,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22494,20 +19922,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -22532,7 +19958,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -22540,16 +19966,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22569,10 +19994,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -22596,14 +20017,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22614,7 +20027,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22628,18 +20041,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22647,7 +20052,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -22658,19 +20063,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_04_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_04_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22701,7 +20095,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22711,17 +20105,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -22733,7 +20125,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -22752,21 +20143,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22778,13 +20168,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -22803,21 +20189,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22827,11 +20212,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -22843,14 +20224,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22877,7 +20250,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22886,17 +20259,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -22904,7 +20273,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -22926,16 +20294,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22946,14 +20313,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -22963,11 +20330,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22979,15 +20346,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -23009,15 +20371,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23028,14 +20389,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -23045,11 +20406,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23061,13 +20422,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -23093,7 +20452,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -23101,16 +20460,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23124,13 +20482,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -23156,14 +20510,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -23175,7 +20521,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23190,18 +20536,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -23213,7 +20551,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -23225,19 +20563,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -23269,7 +20596,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23281,17 +20608,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -23310,21 +20635,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23334,11 +20658,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -23352,14 +20672,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23386,7 +20698,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -23397,17 +20709,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -23431,7 +20739,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -23456,7 +20764,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -23464,16 +20772,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23493,10 +20800,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -23520,14 +20823,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23539,7 +20834,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23553,18 +20848,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23573,7 +20860,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -23584,19 +20871,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23628,7 +20904,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23638,20 +20914,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -23676,7 +20954,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -23684,16 +20962,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23713,10 +20990,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -23741,14 +21014,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23760,7 +21025,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23775,18 +21040,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23795,7 +21052,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -23806,19 +21067,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -23850,7 +21100,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23860,17 +21110,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -23882,7 +21130,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -23901,21 +21148,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23927,13 +21173,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -23952,21 +21194,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23976,11 +21217,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -23992,14 +21229,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24026,7 +21255,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -24035,17 +21264,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -24053,7 +21278,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -24076,16 +21300,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24096,14 +21319,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -24113,11 +21336,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24129,15 +21352,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -24160,15 +21378,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24179,14 +21396,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -24196,11 +21413,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24212,13 +21429,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -24244,7 +21459,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -24252,16 +21467,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24275,13 +21489,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -24307,14 +21517,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -24326,7 +21528,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24341,18 +21543,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -24364,7 +21558,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -24376,19 +21570,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -24420,7 +21603,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24432,17 +21615,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -24461,21 +21642,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24485,11 +21665,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -24503,14 +21679,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24537,7 +21705,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -24548,17 +21716,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -24585,7 +21749,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -24611,7 +21775,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -24619,16 +21783,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24648,10 +21811,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -24676,14 +21835,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -24696,7 +21847,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24711,18 +21862,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -24735,7 +21878,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -24747,19 +21890,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -24792,7 +21925,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24802,23 +21935,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -24844,7 +21975,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -24852,16 +21983,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24881,10 +22011,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -24909,14 +22035,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -24929,7 +22047,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24944,18 +22062,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -24968,7 +22078,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -24980,19 +22090,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -25025,7 +22126,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25035,17 +22136,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -25059,7 +22158,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -25078,21 +22176,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25104,13 +22201,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -25129,21 +22222,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25153,11 +22245,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -25171,14 +22259,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25205,7 +22285,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -25214,17 +22294,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -25234,7 +22310,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -25257,16 +22332,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25277,14 +22351,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -25294,11 +22368,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25310,15 +22384,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -25341,15 +22410,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25360,14 +22428,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -25377,11 +22445,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25393,8 +22461,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -25407,7 +22473,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -25429,17 +22494,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25450,14 +22514,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -25467,11 +22531,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25483,10 +22547,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -25509,7 +22569,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -25534,7 +22598,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -25542,16 +22606,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25571,10 +22634,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -25599,14 +22658,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25618,7 +22669,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25633,18 +22684,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25653,7 +22696,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -25664,19 +22711,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25708,7 +22744,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25718,23 +22754,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -25760,7 +22794,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -25768,16 +22802,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25797,10 +22830,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -25825,14 +22854,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25844,7 +22865,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25859,18 +22880,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25882,7 +22895,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -25894,19 +22907,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -25938,7 +22940,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25948,17 +22950,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -25970,7 +22970,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -25989,21 +22988,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26015,13 +23013,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -26040,21 +23034,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26064,11 +23057,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -26082,14 +23071,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26116,7 +23097,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -26125,17 +23106,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -26143,7 +23120,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -26166,16 +23142,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26186,14 +23161,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -26203,11 +23178,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26219,15 +23194,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -26250,15 +23220,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26269,14 +23238,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -26286,11 +23255,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26302,8 +23271,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -26316,7 +23283,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -26335,21 +23301,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26361,20 +23326,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -26400,7 +23361,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -26409,16 +23370,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26438,12 +23398,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -26471,14 +23427,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -26487,12 +23435,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26510,18 +23458,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -26530,12 +23470,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -26551,20 +23491,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -26598,7 +23527,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26608,17 +23537,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -26634,7 +23561,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -26653,22 +23579,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26680,15 +23605,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -26705,22 +23626,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26730,14 +23650,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -26750,14 +23666,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26785,7 +23693,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -26794,17 +23702,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -26818,7 +23722,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -26842,17 +23745,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26863,14 +23765,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -26880,11 +23782,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26896,10 +23798,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -26942,7 +23840,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -26961,23 +23858,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26989,13 +23885,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -27011,7 +23903,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskRestorePoint] @@ -27034,18 +23925,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27056,14 +23946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -27073,11 +23963,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27089,17 +23979,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -27125,7 +24011,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -27135,16 +24021,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27158,13 +24043,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -27196,14 +24077,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -27217,7 +24090,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27236,18 +24109,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -27261,7 +24126,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -27277,19 +24142,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] @@ -27323,7 +24177,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27335,17 +24189,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -27369,23 +24221,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27395,11 +24246,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -27422,14 +24269,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27458,7 +24297,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -27469,14 +24308,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_compute_management_client_enums.py index 8f2671904be8..aa93092ed6a6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_compute_management_client_enums.py @@ -109,24 +109,21 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + complete only after all data has been copied from the source.""" class DiskCreateOptionTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -186,10 +183,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -197,7 +194,7 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -230,16 +227,16 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -247,13 +244,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExecutionState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -526,7 +523,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" @@ -542,12 +538,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class ResourceIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_models_py3.py index 404c9edca9e7..eb4979545bda 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -253,7 +249,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -278,7 +274,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -293,7 +289,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -359,7 +355,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -416,7 +412,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -515,7 +511,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet] @@ -828,7 +824,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -940,7 +936,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1042,7 +1038,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup] @@ -1202,7 +1198,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.CapacityReservation] @@ -1431,11 +1427,11 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known - values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - and "Import". + values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and + "CopyStart". :vartype create_option: str or ~azure.mgmt.compute.v2021_04_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -1496,8 +1492,8 @@ def __init__( ) -> None: """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. - Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", and "Import". + Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", and + "CopyStart". :paramtype create_option: str or ~azure.mgmt.compute.v2021_04_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -1539,7 +1535,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1750,7 +1746,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1927,7 +1923,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2043,7 +2039,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup] @@ -2247,7 +2243,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -2470,7 +2466,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2801,7 +2797,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2878,7 +2874,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] @@ -2934,7 +2930,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3044,7 +3040,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -3107,17 +3103,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -3270,7 +3255,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.Disk] @@ -3476,7 +3461,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.DiskRestorePoint] @@ -3868,7 +3853,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -3992,7 +3977,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2021_04_01.models.AccessLevel @@ -4141,7 +4126,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4321,7 +4306,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_04_01.models.SubResource @@ -4429,7 +4414,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -4467,7 +4452,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.Image] @@ -4501,7 +4486,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_04_01.models.SubResource @@ -4896,7 +4881,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -4935,7 +4920,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_04_01.models.SourceVault @@ -4968,7 +4953,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_04_01.models.SourceVault @@ -5001,7 +4986,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", secret_url: str, **kw class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -5034,7 +5019,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -5320,7 +5305,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.Usage] @@ -5354,7 +5339,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5701,7 +5686,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2021_04_01.models.OrchestrationServiceNames @@ -5774,7 +5759,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -5936,7 +5921,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -6582,7 +6567,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6676,7 +6661,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup] @@ -6714,17 +6699,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -6765,7 +6739,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2021_04_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2021_04_01.models.PublicIPAddressSkuTier """ @@ -6784,8 +6758,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2021_04_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2021_04_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -6796,7 +6769,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -6838,7 +6811,7 @@ def __init__(self, *, publisher: str, name: str, product: str, **kwargs: Any) -> class PurchasePlanAutoGenerated(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -6917,7 +6890,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7013,7 +6986,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -7120,7 +7093,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7795,7 +7768,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7858,7 +7831,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7921,7 +7894,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7995,7 +7968,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -8042,7 +8015,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -8075,7 +8048,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.RunCommandDocumentBase] @@ -8111,7 +8084,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -8375,7 +8348,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8608,7 +8581,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.Snapshot] @@ -8905,7 +8878,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -8915,7 +8888,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -8944,7 +8917,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -8959,7 +8932,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9015,7 +8988,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource] @@ -9254,7 +9227,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -9275,67 +9248,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -9404,7 +9316,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -9547,7 +9459,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -9765,7 +9677,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10242,7 +10154,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -10332,7 +10244,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10459,7 +10371,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -10503,7 +10415,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10793,7 +10705,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_04_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_04_01.models.UserAssignedIdentitiesValue] """ @@ -10825,7 +10737,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2021_04_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_04_01.models.UserAssignedIdentitiesValue] """ @@ -10839,7 +10751,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10902,7 +10814,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11049,7 +10961,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). Required. @@ -11375,7 +11287,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -11408,10 +11320,10 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -11518,7 +11430,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -11538,10 +11450,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -11707,10 +11619,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -11809,10 +11721,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -11867,7 +11779,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -12114,7 +12026,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand] @@ -12263,7 +12175,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -12508,7 +12420,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12742,10 +12654,10 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension] @@ -12956,7 +12868,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_04_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -12995,7 +12907,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -13006,7 +12918,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -13077,7 +12989,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -13104,7 +13016,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13259,10 +13171,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -13304,7 +13216,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet] @@ -13340,7 +13252,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetSku] @@ -13376,7 +13288,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet] @@ -13409,7 +13321,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -13453,10 +13365,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13559,7 +13471,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -13584,7 +13496,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_04_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -13615,7 +13527,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_04_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13634,7 +13546,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -13895,10 +13807,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -13986,10 +13898,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -14016,7 +13930,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -14024,18 +13938,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -14344,7 +14246,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -14462,7 +14364,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -14563,12 +14465,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_04_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -14601,7 +14503,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_04_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -14736,7 +14638,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -14789,7 +14691,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -14937,7 +14839,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -15304,7 +15206,7 @@ def __init__( self.instance_view = instance_view -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -15327,7 +15229,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -15485,10 +15387,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -15632,7 +15534,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM] @@ -15665,7 +15567,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -16431,7 +16333,7 @@ def __init__( # pylint: disable=too-many-locals self.capacity_reservation = capacity_reservation -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/operations/_operations.py index ff40ad51a181..81b876f30b6b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_04_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -58,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -182,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -238,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -269,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -305,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -341,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -368,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -410,7 +410,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -436,7 +438,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -466,7 +468,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -500,7 +502,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -534,7 +536,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -597,7 +599,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +629,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -651,7 +655,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -787,7 +791,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -818,7 +822,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -842,7 +848,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -996,7 +1002,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1027,7 +1033,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1060,7 +1066,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1091,7 +1097,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -1137,7 +1143,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1172,7 +1178,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1207,7 +1213,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1233,7 +1239,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1273,7 +1279,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1340,7 +1346,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1388,7 +1394,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1419,7 +1425,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1448,7 +1454,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1480,7 +1486,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1522,7 +1528,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1572,7 +1578,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1604,7 +1610,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1635,7 +1641,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1695,7 +1701,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1757,7 +1765,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1895,7 +1903,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1926,7 +1934,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1951,7 +1959,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1976,7 +1984,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2057,7 +2065,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2252,7 +2260,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2292,7 +2300,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2317,7 +2325,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2342,7 +2350,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2373,7 +2381,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2407,7 +2415,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2441,7 +2449,7 @@ def build_virtual_machines_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2471,7 +2479,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2505,7 +2513,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2539,7 +2547,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2571,7 +2579,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2609,7 +2617,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2640,7 +2648,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2678,7 +2686,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2709,7 +2717,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2739,7 +2747,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2765,7 +2775,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2796,7 +2806,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2827,7 +2837,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2865,7 +2875,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2896,7 +2906,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2927,7 +2937,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2958,7 +2968,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2989,7 +2999,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3020,7 +3030,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3051,7 +3061,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3082,7 +3092,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3119,7 +3129,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3150,7 +3160,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3334,7 +3344,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3388,7 +3398,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3424,7 +3434,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3460,7 +3470,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3493,7 +3503,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -3533,7 +3543,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3563,7 +3573,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3706,7 +3718,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3742,7 +3754,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3778,7 +3790,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3811,7 +3823,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -3851,7 +3863,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -3887,7 +3899,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -3920,7 +3932,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -3961,7 +3973,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4002,7 +4014,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4082,7 +4094,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4115,7 +4127,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4150,7 +4162,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4185,7 +4197,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4211,7 +4223,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -4251,7 +4263,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4282,7 +4294,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4307,7 +4319,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4332,7 +4344,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4357,7 +4369,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4388,7 +4400,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4429,7 +4441,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4470,7 +4482,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4508,7 +4520,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4550,7 +4562,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4590,7 +4602,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4622,7 +4634,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4648,7 +4660,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4674,7 +4686,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4709,7 +4721,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4743,7 +4755,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4783,7 +4795,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4815,7 +4827,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -4861,7 +4873,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4895,7 +4907,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4921,7 +4933,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4947,7 +4959,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4973,7 +4985,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5015,7 +5027,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5041,7 +5053,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5067,7 +5079,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5102,7 +5114,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5135,7 +5147,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5168,7 +5180,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5195,7 +5209,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5226,7 +5240,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5261,7 +5275,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5296,7 +5310,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5328,7 +5342,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5368,7 +5382,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5401,7 +5415,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5442,7 +5456,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5483,7 +5497,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5521,7 +5535,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5563,7 +5577,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5727,7 +5741,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5964,7 +5978,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6077,7 +6091,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6111,7 +6125,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6176,7 +6190,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6207,7 +6221,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6261,7 +6275,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6292,7 +6306,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6422,7 +6436,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6476,7 +6490,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6507,7 +6521,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -6548,7 +6562,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -6586,7 +6600,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -6624,7 +6638,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6695,7 +6709,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -6733,7 +6747,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -6776,7 +6790,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -6840,7 +6854,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -6864,14 +6877,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6882,14 +6894,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -6899,11 +6911,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6915,8 +6927,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -6959,7 +6969,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6970,7 +6979,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6982,11 +6991,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6997,7 +7005,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -7007,12 +7015,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7040,7 +7044,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -7048,16 +7052,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7069,13 +7072,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -7098,7 +7097,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7109,7 +7107,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7121,11 +7119,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7136,7 +7133,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -7146,12 +7143,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7179,7 +7172,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -7187,16 +7180,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7208,13 +7200,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7226,7 +7214,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7245,21 +7232,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7269,11 +7255,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -7283,7 +7265,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7302,21 +7283,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7328,13 +7308,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -7345,7 +7321,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -7367,16 +7342,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7387,14 +7361,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -7404,11 +7378,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7420,17 +7394,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -7452,16 +7421,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7472,14 +7440,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -7489,11 +7457,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7505,10 +7473,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -7520,7 +7484,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7542,17 +7505,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7563,14 +7525,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7580,11 +7542,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7596,10 +7558,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -7643,7 +7601,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7654,7 +7611,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7667,11 +7624,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7682,7 +7638,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -7692,12 +7648,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7725,7 +7677,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -7733,16 +7685,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7762,10 +7713,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -7788,7 +7735,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7799,7 +7745,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7812,11 +7758,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7827,7 +7772,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -7837,12 +7782,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7870,7 +7812,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -7878,16 +7820,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7899,13 +7840,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7917,7 +7854,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7936,21 +7872,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7960,11 +7895,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -7984,7 +7915,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8003,22 +7933,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8030,19 +7959,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -8066,15 +7990,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8085,14 +8008,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -8102,11 +8025,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8118,10 +8041,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -8130,7 +8049,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -8154,16 +8072,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8174,14 +8091,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -8191,11 +8108,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8207,10 +8124,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -8255,7 +8168,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8266,7 +8178,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8280,11 +8192,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8295,7 +8206,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -8307,12 +8218,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8340,7 +8247,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -8348,16 +8255,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8377,10 +8283,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -8402,7 +8304,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8413,7 +8314,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8425,11 +8326,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8440,7 +8340,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -8450,12 +8350,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8483,7 +8379,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -8491,16 +8387,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8512,13 +8407,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -8530,7 +8421,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8549,21 +8439,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8573,11 +8462,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -8599,7 +8484,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8618,22 +8502,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8645,13 +8528,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -8660,7 +8539,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -8682,16 +8560,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8702,14 +8579,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -8719,11 +8596,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8735,16 +8612,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -8766,15 +8638,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8785,14 +8656,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -8802,11 +8673,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8818,8 +8689,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -8846,7 +8715,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -8872,7 +8741,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -8881,16 +8750,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8910,10 +8778,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -8938,14 +8802,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -8958,7 +8814,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8972,18 +8828,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -8996,7 +8844,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -9008,19 +8856,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -9053,7 +8890,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9063,24 +8900,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -9106,7 +8941,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -9115,16 +8950,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9136,13 +8970,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9168,14 +8998,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -9188,7 +9010,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9202,18 +9024,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -9226,7 +9040,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -9238,19 +9052,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] @@ -9283,7 +9086,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9293,17 +9096,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -9322,22 +9123,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9347,11 +9147,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -9365,14 +9161,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9400,7 +9188,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9409,17 +9197,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9443,7 +9227,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -9462,23 +9245,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9490,13 +9272,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -9509,7 +9287,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -9531,17 +9308,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9552,14 +9328,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -9569,11 +9345,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9585,10 +9361,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -9615,7 +9387,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -9639,15 +9410,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9658,14 +9428,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -9675,11 +9445,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9691,8 +9461,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -9702,7 +9470,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -9726,16 +9493,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9746,14 +9512,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -9763,11 +9529,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9779,10 +9545,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -9804,7 +9566,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9815,7 +9576,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9827,11 +9588,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9842,7 +9602,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -9852,12 +9612,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9885,7 +9641,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -9893,16 +9649,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9922,10 +9677,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -9947,7 +9698,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9958,7 +9708,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9970,11 +9720,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9985,7 +9734,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -9995,12 +9744,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10028,7 +9774,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -10036,16 +9782,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10057,13 +9802,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -10075,7 +9816,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10094,21 +9834,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10118,11 +9857,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -10132,7 +9867,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10151,21 +9885,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10177,13 +9910,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -10197,7 +9926,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -10216,21 +9944,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10242,13 +9969,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -10285,7 +10008,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -10304,23 +10026,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10332,13 +10053,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -10350,7 +10067,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -10369,21 +10085,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10395,13 +10110,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -10429,7 +10140,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -10448,7 +10158,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -10457,16 +10167,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10478,13 +10187,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -10512,7 +10217,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -10538,7 +10243,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -10547,16 +10252,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10576,10 +10280,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10606,14 +10306,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10627,7 +10319,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10643,18 +10335,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10668,7 +10352,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -10681,20 +10365,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10728,7 +10401,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10738,24 +10411,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -10781,7 +10452,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -10790,16 +10461,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10811,13 +10481,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -10846,14 +10512,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10867,7 +10525,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10883,18 +10541,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10908,7 +10558,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -10921,20 +10571,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10968,7 +10607,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10978,17 +10617,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -11007,22 +10644,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11032,11 +10668,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11051,14 +10683,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11086,7 +10710,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11095,17 +10719,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11127,7 +10747,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11146,23 +10765,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11174,13 +10792,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11194,7 +10808,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11213,22 +10826,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11240,13 +10852,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -11285,7 +10893,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -11304,7 +10911,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -11312,16 +10919,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11333,13 +10939,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11371,7 +10973,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11390,7 +10991,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -11400,16 +11001,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11421,13 +11021,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -11439,7 +11035,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11458,21 +11053,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11484,13 +11078,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -11498,7 +11088,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11517,20 +11106,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11542,13 +11130,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -11562,7 +11146,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11581,22 +11164,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11608,13 +11190,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -11655,7 +11233,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -11674,7 +11251,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -11683,16 +11260,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11704,13 +11280,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11747,7 +11319,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11766,7 +11337,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -11777,16 +11348,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11798,13 +11368,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -11819,7 +11385,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11838,22 +11403,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11865,13 +11429,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -11883,7 +11443,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11902,21 +11461,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11928,13 +11486,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -11951,7 +11505,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11970,23 +11523,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11998,13 +11550,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -12034,7 +11582,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -12056,16 +11603,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12076,14 +11622,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -12093,11 +11639,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12109,8 +11655,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -12139,7 +11683,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -12161,16 +11704,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12181,14 +11723,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12198,11 +11740,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12214,15 +11756,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -12248,7 +11786,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12256,16 +11794,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12279,13 +11816,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -12309,14 +11842,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12329,7 +11854,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12342,18 +11867,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12366,7 +11883,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -12377,19 +11894,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12422,7 +11929,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12434,20 +11941,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -12472,7 +11981,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12480,16 +11989,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12509,10 +12017,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -12535,14 +12039,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12554,7 +12050,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12567,18 +12063,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12587,7 +12075,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -12597,19 +12089,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12641,7 +12122,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12651,20 +12132,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -12689,7 +12172,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12697,16 +12180,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12718,13 +12200,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -12747,14 +12225,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12766,7 +12236,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12778,18 +12248,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12798,7 +12260,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -12807,19 +12273,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] @@ -12851,7 +12306,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12861,17 +12316,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -12890,22 +12343,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12915,11 +12367,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12934,14 +12382,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12969,7 +12409,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12978,17 +12418,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13011,7 +12447,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -13030,22 +12465,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13057,13 +12491,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -13075,7 +12505,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13094,21 +12523,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13120,13 +12548,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13145,21 +12569,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13169,11 +12592,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13184,14 +12603,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13218,7 +12629,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13227,17 +12638,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13256,21 +12663,20 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13280,11 +12686,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13295,14 +12697,6 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13329,7 +12723,7 @@ def begin_deallocate(self, resource_group_name: str, vm_name: str, **kwargs: Any def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13338,17 +12732,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -13365,7 +12755,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13384,21 +12773,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13408,11 +12796,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -13421,7 +12805,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -13443,16 +12826,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13463,14 +12845,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -13480,11 +12862,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13496,10 +12878,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -13508,7 +12886,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -13530,16 +12907,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13550,14 +12926,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -13567,11 +12943,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13583,8 +12959,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13596,7 +12970,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -13618,17 +12991,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13639,14 +13011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -13656,11 +13028,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13672,10 +13044,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -13693,22 +13061,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13718,11 +13085,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13739,14 +13102,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13774,7 +13129,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13783,17 +13138,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13812,21 +13163,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13836,11 +13186,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13850,14 +13196,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13884,7 +13222,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13893,17 +13231,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13922,21 +13256,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13946,11 +13279,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13960,14 +13289,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13994,7 +13315,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14003,17 +13324,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14032,21 +13349,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14056,11 +13372,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14070,14 +13382,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14104,7 +13408,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14113,17 +13417,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14142,21 +13442,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14166,11 +13465,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14180,14 +13475,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14214,7 +13501,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14223,23 +13510,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14268,7 +13551,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14276,16 +13559,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14295,11 +13577,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -14323,14 +13601,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14341,7 +13611,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14354,18 +13624,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14376,7 +13638,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -14386,19 +13648,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14428,7 +13680,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14437,17 +13689,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -14469,7 +13717,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14488,22 +13735,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14515,13 +13761,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14540,21 +13782,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14564,11 +13805,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14578,14 +13815,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14612,7 +13841,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14621,17 +13850,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -14643,7 +13868,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14662,21 +13886,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14686,11 +13909,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14709,21 +13928,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14737,13 +13955,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -14755,14 +13969,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -14792,7 +13998,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14804,23 +14010,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -14846,7 +14050,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14854,16 +14058,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14877,13 +14080,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -14908,14 +14107,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -14928,7 +14119,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14941,18 +14132,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -14965,7 +14148,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -14975,20 +14158,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -15021,7 +14193,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15033,20 +14205,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -15071,7 +14245,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15079,16 +14253,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15102,13 +14275,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -15131,14 +14300,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -15150,7 +14311,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15162,18 +14323,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -15182,7 +14335,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -15191,19 +14348,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -15235,7 +14381,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15247,17 +14393,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -15287,7 +14431,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15311,16 +14454,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15331,14 +14473,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -15348,11 +14490,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15364,15 +14506,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -15398,7 +14536,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15406,16 +14544,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15435,10 +14572,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -15460,14 +14593,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15480,7 +14605,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15492,18 +14617,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15516,7 +14633,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -15525,20 +14642,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15571,7 +14677,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15581,23 +14687,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -15623,7 +14727,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15631,16 +14735,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15652,13 +14755,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15681,14 +14780,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15701,7 +14792,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15713,18 +14804,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15737,7 +14820,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -15746,20 +14829,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15792,7 +14865,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15802,17 +14875,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15831,22 +14902,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15856,11 +14926,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15875,14 +14941,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15910,7 +14968,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15919,17 +14977,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15950,7 +15004,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -15969,22 +15022,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15996,19 +15048,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16037,7 +15085,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16045,16 +15093,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16064,11 +15111,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -16095,14 +15138,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16113,7 +15148,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16128,18 +15163,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16150,7 +15177,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -16162,20 +15189,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16205,7 +15221,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16214,23 +15230,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -16258,7 +15270,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16267,16 +15279,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16286,11 +15297,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -16319,14 +15326,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16337,7 +15336,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -16351,21 +15350,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16376,7 +15367,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -16388,23 +15379,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16435,7 +15415,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16444,17 +15424,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -16466,7 +15442,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -16485,21 +15460,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16511,13 +15485,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -16525,7 +15495,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -16549,16 +15518,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16569,14 +15537,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -16586,11 +15554,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16602,17 +15570,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -16636,15 +15599,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16655,14 +15617,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -16672,11 +15634,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16688,8 +15650,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16701,7 +15661,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -16725,17 +15684,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16746,14 +15704,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -16763,11 +15721,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16779,10 +15737,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16793,7 +15747,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -16817,17 +15770,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16838,14 +15790,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -16855,11 +15807,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16871,15 +15823,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -16910,7 +15858,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16919,16 +15867,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16938,11 +15885,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -16974,14 +15917,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16992,7 +15927,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -17008,7 +15943,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -17016,14 +15951,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17034,7 +15961,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -17048,24 +15975,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17096,7 +16012,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17105,23 +16021,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17150,7 +16062,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17158,16 +16070,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17177,11 +16088,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -17206,14 +16113,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17224,7 +16123,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17237,18 +16136,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17259,7 +16150,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -17269,20 +16160,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17312,7 +16192,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17321,23 +16201,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17366,7 +16242,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17374,16 +16250,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17393,11 +16268,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -17422,14 +16293,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17440,7 +16303,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17453,18 +16316,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17475,7 +16330,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -17485,20 +16340,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17528,7 +16372,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17537,23 +16381,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17582,7 +16422,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17590,16 +16430,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17609,11 +16448,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -17639,14 +16474,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17657,7 +16484,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17671,18 +16498,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17693,7 +16512,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -17704,20 +16523,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17747,7 +16555,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17756,23 +16564,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17801,7 +16605,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17809,16 +16613,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17828,11 +16631,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -17860,14 +16659,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17878,7 +16669,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17894,18 +16685,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17916,7 +16699,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -17929,20 +16712,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17972,7 +16744,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17981,23 +16753,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -18023,7 +16791,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18031,16 +16799,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18050,11 +16817,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -18079,14 +16842,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18097,7 +16852,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18110,18 +16865,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18132,7 +16879,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -18142,20 +16889,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18185,7 +16921,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18194,23 +16930,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18239,7 +16971,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18247,16 +16979,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18266,11 +16997,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -18296,14 +17023,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18314,7 +17033,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18328,18 +17047,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18350,7 +17061,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -18362,20 +17073,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18405,7 +17105,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18414,23 +17114,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18459,7 +17155,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18467,16 +17163,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18486,11 +17181,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -18516,14 +17207,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18534,7 +17217,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18548,18 +17231,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18570,7 +17245,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -18581,20 +17256,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18624,7 +17288,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18633,20 +17297,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -18659,7 +17319,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -18678,22 +17337,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18705,13 +17363,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -18736,7 +17390,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18747,7 +17400,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18760,11 +17413,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18775,7 +17427,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -18786,13 +17438,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_04_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_04_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18820,7 +17469,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18828,16 +17477,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18847,17 +17495,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -18883,7 +17527,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18891,16 +17535,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18910,11 +17553,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -18938,14 +17577,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18956,7 +17587,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18969,18 +17600,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18991,7 +17614,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -19002,19 +17625,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -19044,7 +17657,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19053,17 +17666,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -19093,7 +17702,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -19115,16 +17723,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19135,14 +17742,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -19152,11 +17759,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19168,8 +17775,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -19192,7 +17797,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -19217,7 +17822,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -19225,16 +17830,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19254,10 +17858,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -19279,14 +17879,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19297,7 +17889,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19309,18 +17901,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19328,7 +17912,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -19337,19 +17921,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19380,7 +17953,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19390,20 +17963,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -19428,7 +18003,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -19436,16 +18011,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19465,10 +18039,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -19490,14 +18060,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19508,7 +18070,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19520,18 +18082,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19539,7 +18093,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -19548,19 +18106,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19591,7 +18138,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19601,17 +18148,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -19630,21 +18175,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19654,11 +18198,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19668,14 +18208,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19702,7 +18234,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19711,17 +18243,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19735,7 +18263,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -19754,22 +18281,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19781,13 +18307,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -19795,7 +18317,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19817,16 +18338,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19837,14 +18357,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19854,11 +18374,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19870,16 +18390,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19901,15 +18416,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19920,14 +18434,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19937,11 +18451,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19953,8 +18467,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -20000,7 +18512,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20011,7 +18522,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20026,11 +18537,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20041,7 +18551,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -20053,12 +18563,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20086,7 +18592,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -20094,16 +18600,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20123,10 +18628,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -20149,7 +18650,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20160,7 +18660,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20173,11 +18673,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20188,7 +18687,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -20198,12 +18697,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20231,7 +18727,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -20239,16 +18735,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20260,13 +18755,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -20285,21 +18776,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20309,11 +18799,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20326,14 +18812,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20360,7 +18838,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20369,17 +18847,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -20401,7 +18875,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20420,22 +18893,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20447,13 +18919,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -20461,7 +18929,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -20485,16 +18952,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20505,14 +18971,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -20522,11 +18988,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20538,17 +19004,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -20572,15 +19033,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20591,14 +19051,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -20608,11 +19068,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20624,8 +19084,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -20652,7 +19110,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -20678,7 +19136,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -20687,16 +19145,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20708,13 +19165,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -20741,14 +19194,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -20761,7 +19206,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20776,18 +19221,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -20800,7 +19237,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -20813,19 +19250,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RestorePoint] @@ -20858,7 +19284,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20868,17 +19294,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -20897,22 +19321,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20922,11 +19345,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20940,14 +19359,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20975,7 +19386,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20984,17 +19395,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21008,7 +19415,6 @@ def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -21027,22 +19433,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21054,13 +19459,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -21106,7 +19507,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21117,7 +19517,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21131,11 +19531,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21146,7 +19545,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -21158,12 +19557,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21191,7 +19586,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -21199,16 +19594,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21228,10 +19622,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -21255,7 +19645,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21266,7 +19655,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21280,11 +19669,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21295,7 +19683,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -21306,12 +19694,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21339,7 +19724,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -21347,16 +19732,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21368,13 +19752,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -21389,7 +19769,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21408,21 +19787,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21432,11 +19810,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -21460,7 +19834,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21479,22 +19852,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21506,13 +19878,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -21533,7 +19901,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -21557,17 +19924,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21578,14 +19944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -21595,11 +19961,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21611,10 +19977,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -21628,7 +19990,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -21652,16 +20013,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21672,14 +20032,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -21689,11 +20049,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21705,10 +20065,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -21735,7 +20091,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -21761,7 +20117,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21770,16 +20126,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21799,10 +20154,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -21829,14 +20180,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21850,7 +20193,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21866,18 +20209,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21891,7 +20226,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -21905,19 +20240,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21951,7 +20275,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21961,24 +20285,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -22004,7 +20326,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -22013,16 +20335,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22036,13 +20357,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22068,14 +20385,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22089,7 +20398,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22103,18 +20412,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22128,7 +20429,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -22140,19 +20441,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22186,7 +20476,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22196,17 +20486,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -22229,22 +20517,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22254,11 +20541,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -22278,14 +20561,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22313,7 +20588,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22322,17 +20597,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -22357,7 +20628,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -22376,23 +20646,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22404,13 +20673,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -22423,7 +20688,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.CapacityReservation] @@ -22446,17 +20710,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22467,14 +20730,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -22484,11 +20747,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22500,12 +20763,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -22530,7 +20789,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -22556,7 +20815,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -22565,16 +20824,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22594,10 +20852,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -22625,14 +20879,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22646,7 +20892,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22662,18 +20908,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22687,7 +20925,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -22700,20 +20938,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22747,7 +20974,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22757,24 +20984,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -22800,7 +21025,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -22809,16 +21034,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22838,10 +21062,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -22869,14 +21089,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22890,7 +21102,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22906,18 +21118,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22931,7 +21135,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -22944,20 +21148,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22991,7 +21185,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23001,17 +21195,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -23030,22 +21222,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23055,11 +21246,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23074,14 +21261,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23109,7 +21288,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23118,17 +21297,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23150,7 +21325,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -23169,23 +21343,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23197,13 +21370,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -23215,7 +21384,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -23239,17 +21407,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23260,14 +21427,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -23277,11 +21444,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23293,12 +21460,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23335,21 +21498,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23359,11 +21521,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23373,14 +21531,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23407,7 +21557,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23416,17 +21566,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -23445,21 +21591,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23469,11 +21614,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -23487,14 +21628,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23521,7 +21654,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23530,17 +21663,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -23559,21 +21688,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23583,11 +21711,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -23601,14 +21725,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23635,7 +21751,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23644,17 +21760,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -23666,7 +21778,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -23685,21 +21796,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23711,16 +21821,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23746,7 +21852,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -23772,7 +21878,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23782,16 +21888,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23811,10 +21916,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -23844,14 +21945,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23866,7 +21959,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23883,18 +21976,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23909,7 +21994,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -23923,20 +22008,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23971,7 +22045,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23981,17 +22055,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -23999,7 +22071,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -24025,7 +22097,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24035,16 +22107,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24056,13 +22127,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -24093,14 +22160,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24115,7 +22174,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24132,18 +22191,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24158,7 +22209,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -24172,20 +22223,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24220,7 +22261,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24230,17 +22271,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -24259,23 +22298,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24285,11 +22323,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24305,14 +22339,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24341,7 +22367,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24350,17 +22376,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24385,7 +22407,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -24404,7 +22425,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24412,16 +22433,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24433,13 +22453,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24461,7 +22477,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -24480,23 +22495,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24508,13 +22522,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -24542,7 +22552,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -24571,7 +22583,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24580,16 +22592,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24599,11 +22610,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -24631,14 +22638,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24650,7 +22649,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -24665,18 +22664,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24688,7 +22679,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -24700,20 +22693,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -24744,7 +22727,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24753,17 +22736,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24782,22 +22761,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24807,11 +22785,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -24826,14 +22800,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24861,7 +22827,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24870,17 +22836,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24899,22 +22861,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24924,11 +22885,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -24944,14 +22901,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24979,7 +22928,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24988,24 +22937,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -25031,7 +22976,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25040,16 +22985,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25069,10 +23013,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -25099,14 +23039,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25120,7 +23052,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25136,18 +23068,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25161,7 +23085,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -25174,19 +23098,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25220,7 +23133,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25230,17 +23143,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25265,23 +23176,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25291,11 +23201,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25318,14 +23224,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25354,7 +23252,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25363,17 +23261,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25397,7 +23291,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_04_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -25416,23 +23309,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25444,13 +23336,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -25464,7 +23352,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -25483,22 +23370,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25510,13 +23396,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -25545,7 +23427,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25569,7 +23450,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -25577,12 +23458,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25593,14 +23473,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -25610,11 +23490,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25626,10 +23506,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -25653,23 +23529,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25679,11 +23554,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -25709,14 +23580,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25745,7 +23608,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25754,17 +23617,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25783,22 +23642,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25808,11 +23666,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -25826,14 +23680,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25861,7 +23707,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25870,17 +23716,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25899,22 +23741,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25924,11 +23765,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -25942,14 +23779,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25977,7 +23806,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25986,17 +23815,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26015,22 +23840,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26040,11 +23864,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -26059,14 +23879,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26094,7 +23906,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26103,17 +23915,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -26139,7 +23947,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -26158,23 +23965,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26186,13 +23992,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26211,22 +24013,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26236,11 +24037,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -26254,14 +24051,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26289,7 +24078,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26298,17 +24087,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -26322,7 +24107,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -26341,22 +24125,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26366,18 +24149,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -26403,7 +24182,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26412,16 +24191,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26435,13 +24213,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -26467,14 +24241,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -26487,7 +24253,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26501,18 +24267,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -26525,7 +24283,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -26537,19 +24295,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.RunCommandResult] @@ -26582,7 +24329,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26594,17 +24341,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -26628,7 +24373,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26653,23 +24398,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26683,13 +24427,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26711,14 +24451,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26728,7 +24460,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26737,18 +24469,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26758,7 +24482,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26766,19 +24490,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26810,7 +24524,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26822,20 +24536,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26860,23 +24572,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26890,13 +24601,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26917,14 +24624,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26934,7 +24633,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26942,18 +24641,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26963,7 +24654,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26971,19 +24662,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_04_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27015,7 +24695,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27027,17 +24707,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -27066,7 +24744,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -27090,16 +24767,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27110,14 +24786,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -27127,11 +24803,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27143,10 +24819,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -27155,7 +24827,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -27174,21 +24845,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27200,20 +24870,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27239,7 +24905,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27248,16 +24914,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27277,10 +24942,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27307,14 +24968,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27328,7 +24981,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27344,18 +24997,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27369,7 +25014,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -27382,19 +25027,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27428,7 +25062,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27438,24 +25072,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27481,7 +25113,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27490,16 +25122,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27511,13 +25142,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27545,14 +25172,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27566,7 +25185,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27582,18 +25201,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27607,7 +25218,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -27620,19 +25231,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27666,7 +25267,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27676,17 +25277,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27705,22 +25304,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27730,11 +25328,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27749,14 +25343,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27784,7 +25370,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27793,17 +25379,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27825,7 +25407,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27844,23 +25425,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27872,13 +25452,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27892,7 +25468,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27916,18 +25491,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27938,14 +25512,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27955,11 +25529,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27971,12 +25545,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -28002,7 +25572,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28028,7 +25598,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28038,16 +25608,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28067,10 +25636,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -28099,14 +25664,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28121,7 +25678,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28138,18 +25695,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28164,7 +25713,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -28178,19 +25727,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28225,7 +25763,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28235,17 +25773,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -28253,7 +25789,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28279,7 +25815,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28289,16 +25825,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28310,13 +25845,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -28346,14 +25877,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28368,7 +25891,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28385,18 +25908,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28411,7 +25926,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -28425,19 +25940,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28472,7 +25977,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28482,17 +25987,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -28511,23 +26014,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28537,11 +26039,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28557,14 +26055,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28593,7 +26083,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28602,17 +26092,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -28637,7 +26123,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -28656,7 +26141,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28664,16 +26149,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28685,13 +26169,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28713,7 +26193,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28737,19 +26216,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28760,14 +26238,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28777,11 +26255,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28793,10 +26271,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } - class DisksOperations: """ @@ -28819,7 +26293,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -28844,7 +26318,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -28852,16 +26326,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28881,10 +26354,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -28908,14 +26377,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -28926,7 +26387,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28940,18 +26401,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -28959,7 +26412,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -28970,19 +26423,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_04_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_04_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29013,7 +26455,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29023,20 +26465,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -29061,7 +26501,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -29069,16 +26509,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29098,10 +26537,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -29125,14 +26560,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29143,7 +26570,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29157,18 +26584,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29176,7 +26595,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -29187,19 +26606,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_04_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_04_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29230,7 +26638,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29240,17 +26648,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -29262,7 +26668,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -29281,21 +26686,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29307,13 +26711,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -29332,21 +26732,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29356,11 +26755,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -29372,14 +26767,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29406,7 +26793,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -29415,17 +26802,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -29433,7 +26816,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29455,16 +26837,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29475,14 +26856,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -29492,11 +26873,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29508,15 +26889,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -29538,15 +26914,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29557,14 +26932,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -29574,11 +26949,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29590,13 +26965,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -29622,7 +26995,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -29630,16 +27003,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29653,13 +27025,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -29685,14 +27053,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -29703,7 +27063,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29718,18 +27078,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -29740,7 +27092,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -29752,19 +27104,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -29795,7 +27136,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29807,17 +27148,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -29836,21 +27175,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29860,11 +27198,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -29876,14 +27210,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29910,7 +27236,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -29921,17 +27247,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -29955,7 +27277,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -29980,7 +27302,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -29988,16 +27310,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30017,10 +27338,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -30044,14 +27361,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30062,7 +27371,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30076,18 +27385,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30095,7 +27396,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -30106,19 +27407,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30149,7 +27439,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30159,20 +27449,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -30197,7 +27489,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -30205,16 +27497,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30234,10 +27525,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -30262,14 +27549,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30280,7 +27559,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30295,18 +27574,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30314,7 +27585,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -30325,19 +27600,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_04_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30368,7 +27632,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30378,17 +27642,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -30400,7 +27662,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -30419,21 +27680,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30445,13 +27705,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -30470,21 +27726,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30494,11 +27749,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -30510,14 +27761,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -30544,7 +27787,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -30553,17 +27796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -30571,7 +27810,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30593,16 +27831,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -30613,14 +27850,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -30630,11 +27867,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30646,15 +27883,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -30676,15 +27908,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -30695,14 +27926,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -30712,11 +27943,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30728,13 +27959,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -30760,7 +27989,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -30768,16 +27997,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30791,13 +28019,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -30823,14 +28047,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -30841,7 +28057,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30856,18 +28072,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -30878,7 +28086,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -30890,19 +28098,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -30933,7 +28130,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30945,17 +28142,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -30974,21 +28169,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30998,11 +28192,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -31014,14 +28204,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -31048,7 +28230,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -31059,17 +28241,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -31096,7 +28274,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -31122,7 +28300,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -31130,16 +28308,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31159,10 +28336,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -31187,14 +28360,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31206,7 +28371,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -31221,18 +28386,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31244,7 +28401,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -31256,19 +28413,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31300,7 +28447,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -31310,23 +28457,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -31352,7 +28497,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -31360,16 +28505,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31389,10 +28533,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -31417,14 +28557,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31436,7 +28568,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -31451,18 +28583,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31474,7 +28598,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -31486,19 +28610,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] @@ -31530,7 +28645,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -31540,17 +28655,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -31562,7 +28675,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -31581,21 +28693,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31607,13 +28718,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -31632,21 +28739,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31656,11 +28762,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -31672,14 +28774,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -31706,7 +28800,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -31715,17 +28809,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -31733,7 +28823,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -31755,16 +28844,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -31775,14 +28863,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -31792,11 +28880,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31808,15 +28896,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -31838,15 +28921,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -31857,14 +28939,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -31874,11 +28956,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31890,8 +28972,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -31904,7 +28984,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -31926,17 +29005,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -31947,14 +29025,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -31964,11 +29042,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31980,10 +29058,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -32006,7 +29080,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -32031,7 +29109,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -32039,16 +29117,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32068,10 +29145,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -32096,14 +29169,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32114,7 +29179,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -32129,18 +29194,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32148,7 +29205,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -32159,19 +29220,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32202,7 +29252,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -32212,23 +29262,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -32254,7 +29302,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -32262,16 +29310,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32291,10 +29338,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -32319,14 +29362,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32337,7 +29372,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -32352,18 +29387,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32374,7 +29401,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -32386,19 +29413,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_04_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32429,7 +29445,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -32439,17 +29455,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -32461,7 +29475,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -32480,21 +29493,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32506,13 +29518,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -32531,21 +29539,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32555,11 +29562,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -32571,14 +29574,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -32605,7 +29600,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -32614,17 +29609,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -32632,7 +29623,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32654,16 +29644,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -32674,14 +29663,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -32691,11 +29680,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32707,15 +29696,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -32737,15 +29721,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -32756,14 +29739,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -32773,11 +29756,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32789,8 +29772,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -32803,7 +29784,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -32822,21 +29802,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32848,20 +29827,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -32887,7 +29862,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -32896,16 +29871,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32925,12 +29899,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -32958,14 +29928,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -32974,12 +29936,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -32997,18 +29959,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -33017,12 +29971,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -33038,20 +29992,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -33085,7 +30028,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -33095,17 +30038,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -33121,7 +30062,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -33140,22 +30080,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33167,15 +30106,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -33192,22 +30127,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33217,14 +30151,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -33237,14 +30167,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -33272,7 +30194,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -33281,17 +30203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -33305,7 +30223,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -33329,17 +30246,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33350,14 +30266,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -33367,11 +30283,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33383,10 +30299,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -33429,7 +30341,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_04_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -33448,23 +30359,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33476,13 +30386,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -33498,7 +30404,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_04_01.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -33520,18 +30425,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33542,14 +30446,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -33559,11 +30463,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33575,17 +30479,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -33611,7 +30511,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -33621,16 +30521,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33644,13 +30543,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -33682,14 +30577,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -33702,7 +30589,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -33721,18 +30608,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -33745,7 +30624,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -33761,19 +30640,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_04_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_04_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -33806,7 +30674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -33818,17 +30686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -33852,23 +30718,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-04-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33878,11 +30743,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -33905,14 +30766,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -33941,7 +30794,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -33952,14 +30805,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_compute_management_client.py index 284ad2843b27..e26641623818 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -198,7 +200,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -316,7 +336,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -336,7 +356,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_configuration.py index ede9522f15e8..c3a833034682 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_metadata.json index 4e309e65cb80..a7b998c89213 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_compute_management_client.py index d87f3d0c24bf..085948dc7107 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -201,7 +203,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -319,7 +339,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -339,7 +361,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_configuration.py index bc115b1f63cf..e50af76cac79 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/operations/_operations.py index 509731909529..88fcc78f0326 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -269,7 +269,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -293,14 +292,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -311,14 +309,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -328,11 +326,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -344,8 +342,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -388,7 +384,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -399,7 +394,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -411,11 +406,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -426,7 +420,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -436,12 +430,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -469,7 +459,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -477,16 +467,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -498,13 +487,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -527,7 +512,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -538,7 +522,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -550,11 +534,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -565,7 +548,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -575,12 +558,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -608,7 +587,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -616,16 +595,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -637,13 +615,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -655,7 +629,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -674,21 +647,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -698,11 +670,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -712,7 +680,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -731,21 +698,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -757,13 +723,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -774,7 +736,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet] @@ -797,16 +758,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -817,14 +777,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -834,11 +794,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -850,17 +810,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet] @@ -883,16 +838,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -903,14 +857,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -920,11 +874,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -936,10 +890,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -951,7 +901,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] @@ -974,17 +923,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -995,14 +943,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -1012,11 +960,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1028,10 +976,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1075,7 +1019,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1086,7 +1029,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1099,11 +1042,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1114,7 +1056,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1124,12 +1066,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1157,7 +1095,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1165,16 +1103,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1194,10 +1131,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1220,7 +1153,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1231,7 +1163,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1244,11 +1176,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1259,7 +1190,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1269,12 +1200,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1302,7 +1230,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1310,16 +1238,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1331,13 +1258,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1349,7 +1272,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1368,21 +1290,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1392,11 +1313,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1416,7 +1333,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1435,22 +1351,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1462,19 +1377,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1498,15 +1408,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1517,14 +1426,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1534,11 +1443,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1550,10 +1459,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1562,7 +1467,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1586,16 +1490,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1606,14 +1509,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1623,11 +1526,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1639,10 +1542,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1687,7 +1586,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1698,7 +1596,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1712,11 +1610,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1727,7 +1624,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1739,12 +1636,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1772,7 +1665,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1780,16 +1673,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1809,10 +1701,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1834,7 +1722,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1845,7 +1732,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1857,11 +1744,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1872,7 +1758,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1882,12 +1768,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1915,7 +1797,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1923,16 +1805,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1944,13 +1825,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1962,7 +1839,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1981,21 +1857,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2005,11 +1880,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -2031,7 +1902,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -2050,22 +1920,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2077,13 +1946,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2094,7 +1959,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup] @@ -2117,16 +1981,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2137,14 +2000,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2154,11 +2017,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2170,16 +2033,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup] @@ -2202,15 +2060,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2221,14 +2078,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2238,11 +2095,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2254,8 +2111,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2282,7 +2137,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2308,7 +2163,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2317,16 +2172,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2346,10 +2200,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2374,14 +2224,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2395,7 +2237,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2409,18 +2251,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2434,7 +2268,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2446,19 +2280,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2492,7 +2315,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2502,24 +2325,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2545,7 +2366,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2554,16 +2375,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2575,13 +2395,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2607,14 +2423,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2628,7 +2436,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2642,18 +2450,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2667,7 +2467,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2679,19 +2479,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2725,7 +2514,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2735,17 +2524,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2764,22 +2551,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2789,11 +2575,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2807,14 +2589,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2842,7 +2616,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2851,17 +2625,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2885,7 +2655,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2904,23 +2673,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2932,13 +2700,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2951,7 +2715,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -2974,17 +2737,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2995,14 +2757,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -3012,11 +2774,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3028,10 +2790,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -3058,7 +2816,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3082,15 +2839,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3101,14 +2857,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3118,11 +2874,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3134,8 +2890,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -3145,7 +2899,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3169,16 +2922,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3189,14 +2941,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3206,11 +2958,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,10 +2974,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -3247,7 +2995,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3258,7 +3005,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3270,11 +3017,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3285,7 +3031,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -3295,12 +3041,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3328,7 +3070,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3336,16 +3078,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3365,10 +3106,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -3390,7 +3127,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3401,7 +3137,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3413,11 +3149,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3428,7 +3163,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -3438,12 +3173,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3471,7 +3203,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3479,16 +3211,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3500,13 +3231,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -3518,7 +3245,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3537,21 +3263,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3561,11 +3286,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -3577,7 +3298,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3596,21 +3316,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3622,13 +3341,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -3642,7 +3357,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3661,21 +3375,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3687,13 +3400,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -3730,7 +3439,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3749,23 +3457,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3777,13 +3484,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3795,7 +3498,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3814,21 +3516,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3840,13 +3541,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3874,7 +3571,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3893,7 +3589,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3902,16 +3598,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3923,13 +3618,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -3957,7 +3648,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -3983,7 +3674,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -3992,16 +3683,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4021,10 +3711,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4051,14 +3737,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4072,7 +3750,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4088,18 +3766,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4113,7 +3783,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4126,20 +3796,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4173,7 +3832,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4183,24 +3842,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4226,7 +3883,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4235,16 +3892,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4256,13 +3912,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -4291,14 +3943,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4312,7 +3956,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4328,18 +3972,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4353,7 +3989,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4366,20 +4002,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4413,7 +4038,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4423,17 +4048,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4452,22 +4075,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4477,11 +4099,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4496,14 +4114,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4531,7 +4141,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4540,17 +4150,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4572,7 +4178,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4591,23 +4196,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4619,13 +4223,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4639,7 +4239,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4658,22 +4257,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4685,13 +4283,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4730,7 +4324,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4749,7 +4342,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4757,16 +4350,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4778,13 +4370,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4816,7 +4404,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4835,7 +4422,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4845,16 +4432,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4866,13 +4452,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4884,7 +4466,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4903,21 +4484,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4929,13 +4509,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -4943,7 +4519,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4962,20 +4537,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4987,13 +4561,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5007,7 +4577,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5026,22 +4595,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5053,13 +4621,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -5100,7 +4664,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5119,7 +4682,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5128,16 +4691,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5149,13 +4711,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -5192,7 +4750,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5211,7 +4768,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5222,16 +4779,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5243,13 +4799,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -5264,7 +4816,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5283,22 +4834,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5310,13 +4860,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -5328,7 +4874,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5347,21 +4892,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5373,13 +4917,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5396,7 +4936,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5415,23 +4954,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5443,13 +4981,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5479,7 +5013,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5501,16 +5034,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5521,14 +5053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5538,11 +5070,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5554,8 +5086,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -5584,7 +5114,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -5607,16 +5136,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5627,14 +5155,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5644,11 +5172,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5660,15 +5188,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -5694,7 +5218,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5702,16 +5226,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5725,13 +5248,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -5755,14 +5274,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5775,7 +5286,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5788,18 +5299,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5812,7 +5315,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5823,19 +5326,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5868,7 +5361,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5880,20 +5373,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -5918,7 +5413,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5926,16 +5421,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5955,10 +5449,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -5981,14 +5471,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6001,7 +5483,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6014,18 +5496,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6035,7 +5509,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -6045,19 +5523,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6090,7 +5557,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6100,20 +5567,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6138,7 +5607,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6146,16 +5615,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6167,13 +5635,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6196,14 +5660,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6216,7 +5672,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6228,18 +5684,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6249,7 +5697,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -6258,19 +5710,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6303,7 +5744,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6313,17 +5754,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6342,22 +5781,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6367,11 +5805,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6386,14 +5820,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6421,7 +5847,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6430,17 +5856,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6463,7 +5885,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6482,22 +5903,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6509,13 +5929,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -6527,7 +5943,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6546,21 +5961,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6572,13 +5986,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6597,21 +6007,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6621,11 +6030,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -6638,14 +6043,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6672,7 +6069,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6681,17 +6078,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -6710,22 +6103,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6735,11 +6127,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6755,14 +6143,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6790,7 +6170,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6799,17 +6179,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -6826,7 +6202,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6845,21 +6220,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6869,11 +6243,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: @@ -6882,7 +6252,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -6905,16 +6274,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6925,14 +6293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -6942,11 +6310,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6958,10 +6326,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> AsyncIterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -6970,7 +6334,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -6993,16 +6356,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Async def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7013,14 +6375,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7030,11 +6392,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7046,8 +6408,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7059,7 +6419,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] @@ -7082,17 +6441,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7103,14 +6461,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7120,11 +6478,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7136,10 +6494,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -7157,22 +6511,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7182,11 +6535,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7203,14 +6552,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7238,7 +6579,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7247,17 +6588,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7276,21 +6613,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7300,11 +6636,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7314,14 +6646,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7348,7 +6672,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7357,17 +6681,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7386,21 +6706,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7410,11 +6729,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7424,14 +6739,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7458,7 +6765,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7467,17 +6774,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7496,21 +6799,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7520,11 +6822,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7534,14 +6832,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7568,7 +6858,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7577,17 +6867,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7606,21 +6892,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7630,11 +6915,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7644,14 +6925,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7678,7 +6951,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7687,23 +6960,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7732,7 +7001,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7740,16 +7009,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7759,11 +7027,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -7787,14 +7051,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7805,7 +7061,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7818,18 +7074,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7840,7 +7088,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -7850,19 +7098,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7892,7 +7130,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7901,17 +7139,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7933,7 +7167,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7952,22 +7185,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7979,13 +7211,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8004,21 +7232,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8028,11 +7255,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8044,14 +7267,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8078,7 +7293,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8087,17 +7302,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8109,7 +7320,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8128,21 +7338,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8152,11 +7361,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8175,21 +7380,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8203,13 +7407,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -8221,14 +7421,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -8258,7 +7450,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8270,23 +7462,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -8312,7 +7502,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8320,16 +7510,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8343,13 +7532,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -8374,14 +7559,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8394,7 +7571,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8407,18 +7584,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8431,7 +7600,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -8441,20 +7610,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8487,7 +7645,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8499,20 +7657,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -8537,7 +7697,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8545,16 +7705,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8568,13 +7727,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8597,14 +7752,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8617,7 +7764,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8629,18 +7776,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8650,7 +7789,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -8659,19 +7802,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8704,7 +7836,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8716,17 +7848,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -8756,7 +7886,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8780,16 +7909,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8800,14 +7928,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -8817,11 +7945,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8833,15 +7961,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8867,7 +7991,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8875,16 +7999,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8904,10 +8027,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -8929,14 +8048,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8949,7 +8060,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8961,18 +8072,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8985,7 +8088,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -8994,20 +8097,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9040,7 +8132,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9050,23 +8142,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9092,7 +8182,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9100,16 +8190,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9121,13 +8210,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9150,14 +8235,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9170,7 +8247,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9182,18 +8259,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9206,7 +8275,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9215,20 +8284,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9261,7 +8320,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9271,17 +8330,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9300,22 +8357,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9325,11 +8381,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9344,14 +8396,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9379,7 +8423,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9388,17 +8432,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9419,7 +8459,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9438,22 +8477,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9465,19 +8503,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9506,7 +8540,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9514,16 +8548,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9533,11 +8566,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -9564,14 +8593,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9582,7 +8603,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9597,18 +8618,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9619,7 +8632,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9631,20 +8644,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9674,7 +8676,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9683,23 +8685,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9727,7 +8725,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9736,16 +8734,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9755,11 +8752,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -9788,14 +8781,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9806,7 +8791,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -9820,21 +8805,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9845,7 +8822,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9857,23 +8834,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9904,7 +8870,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9913,17 +8879,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -9935,7 +8897,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -9954,21 +8915,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9980,13 +8940,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -9994,7 +8950,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10018,16 +8973,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10038,14 +8992,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10055,11 +9009,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10071,17 +9025,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10105,15 +9054,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10124,14 +9072,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10141,11 +9089,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10157,8 +9105,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10170,7 +9116,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10194,17 +9139,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10215,14 +9159,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10232,11 +9176,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10248,10 +9192,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10262,7 +9202,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10286,17 +9225,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10307,14 +9245,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10324,11 +9262,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10340,15 +9278,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10379,7 +9313,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10388,16 +9322,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10407,11 +9340,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -10443,14 +9372,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10461,7 +9382,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -10477,7 +9398,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -10485,14 +9406,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10503,7 +9416,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10517,24 +9430,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10565,7 +9467,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10574,23 +9476,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10619,7 +9517,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10627,16 +9525,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10646,11 +9543,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -10675,14 +9568,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10693,7 +9578,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10706,18 +9591,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10728,7 +9605,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -10738,20 +9615,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10781,7 +9647,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10790,23 +9656,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10835,7 +9697,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10843,16 +9705,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10862,11 +9723,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10891,14 +9748,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10909,7 +9758,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10922,18 +9771,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10944,7 +9785,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -10954,20 +9795,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10997,7 +9827,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11006,23 +9836,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11051,7 +9877,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11059,16 +9885,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11078,11 +9903,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -11108,14 +9929,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11126,7 +9939,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11140,18 +9953,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11162,7 +9967,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11173,20 +9978,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11216,7 +10010,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11225,23 +10019,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11270,7 +10060,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11278,16 +10068,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11297,11 +10086,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -11329,14 +10114,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11347,7 +10124,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11363,18 +10140,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11385,7 +10154,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11398,20 +10167,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11441,7 +10199,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11450,23 +10208,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11492,7 +10246,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11500,16 +10254,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11519,11 +10272,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -11548,14 +10297,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11566,7 +10307,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11579,18 +10320,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11601,7 +10334,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11611,20 +10344,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11654,7 +10376,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11663,23 +10385,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11708,7 +10426,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11716,16 +10434,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11735,11 +10452,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11765,14 +10478,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11783,7 +10488,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11797,18 +10502,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11819,7 +10516,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11831,20 +10528,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11874,7 +10560,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11883,23 +10569,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11928,7 +10610,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11936,16 +10618,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11955,11 +10636,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -11985,14 +10662,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12003,7 +10672,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12017,18 +10686,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12039,7 +10700,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12050,20 +10711,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12093,7 +10743,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12102,20 +10752,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -12128,7 +10774,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12147,22 +10792,21 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12174,13 +10818,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -12205,7 +10845,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12216,7 +10855,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12229,11 +10868,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12244,7 +10882,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -12255,13 +10893,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12289,7 +10924,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12297,16 +10932,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12316,17 +10950,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12352,7 +10982,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12360,16 +10990,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12379,11 +11008,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -12407,14 +11032,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12425,7 +11042,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12438,18 +11055,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12460,7 +11069,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -12471,19 +11080,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12513,7 +11112,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12522,17 +11121,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -12562,7 +11157,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] @@ -12585,16 +11179,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12605,14 +11198,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12622,11 +11215,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12638,8 +11231,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12662,7 +11253,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12687,7 +11278,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12695,16 +11286,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12724,10 +11314,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -12749,14 +11335,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12767,7 +11345,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12779,18 +11357,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12798,7 +11368,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -12807,19 +11377,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12850,7 +11409,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12860,20 +11419,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12898,7 +11459,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12906,16 +11467,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12935,10 +11495,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -12960,14 +11516,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12978,7 +11526,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12990,18 +11538,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13009,7 +11549,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -13018,19 +11562,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13061,7 +11594,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13071,17 +11604,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -13100,21 +11631,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13124,11 +11654,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -13138,14 +11664,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13172,7 +11690,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13181,17 +11699,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13205,7 +11719,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -13224,22 +11737,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13251,13 +11763,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -13265,7 +11773,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13287,16 +11794,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13307,14 +11813,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13324,11 +11830,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13340,16 +11846,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13371,15 +11872,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13390,14 +11890,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13407,11 +11907,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13423,8 +11923,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -13470,7 +11968,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13481,7 +11978,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13496,11 +11993,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13511,7 +12007,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -13523,12 +12019,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13556,7 +12048,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13564,16 +12056,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13593,10 +12084,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -13619,7 +12106,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13630,7 +12116,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13643,11 +12129,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13658,7 +12143,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -13668,12 +12153,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13701,7 +12183,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13709,16 +12191,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13730,13 +12211,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -13755,21 +12232,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13779,11 +12255,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13796,14 +12268,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13830,7 +12294,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13839,17 +12303,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13871,7 +12331,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13890,22 +12349,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13917,13 +12375,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -13931,7 +12385,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -13955,16 +12408,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13975,14 +12427,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -13992,11 +12444,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14008,17 +12460,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -14042,15 +12489,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14061,14 +12507,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -14078,11 +12524,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14094,8 +12540,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -14122,7 +12566,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -14148,7 +12592,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -14157,16 +12601,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14178,13 +12621,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -14211,14 +12650,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -14231,7 +12662,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14246,18 +12677,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -14270,7 +12693,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -14283,19 +12706,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -14328,7 +12740,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14338,17 +12750,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -14367,22 +12777,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14392,11 +12801,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14410,14 +12815,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14445,7 +12842,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14454,17 +12851,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14478,7 +12871,6 @@ async def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -14497,22 +12889,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14524,13 +12915,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -14576,7 +12963,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14587,7 +12973,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14601,11 +12987,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14616,7 +13001,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -14628,12 +13013,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14661,7 +13042,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14669,16 +13050,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14698,10 +13078,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -14725,7 +13101,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14736,7 +13111,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14750,11 +13125,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14765,7 +13139,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -14776,12 +13150,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14809,7 +13180,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14817,16 +13188,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14838,13 +13208,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14859,7 +13225,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14878,21 +13243,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14902,11 +13266,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14930,7 +13290,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14949,22 +13308,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14976,13 +13334,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15003,7 +13357,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15027,17 +13380,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15048,14 +13400,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15065,11 +13417,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15081,10 +13433,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -15098,7 +13446,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15122,16 +13469,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15142,14 +13488,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15159,11 +13505,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15175,10 +13521,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -15205,7 +13547,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -15231,7 +13573,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15240,16 +13582,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15269,10 +13610,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -15299,14 +13636,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15320,7 +13649,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15336,18 +13665,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15361,7 +13682,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -15375,19 +13696,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15421,7 +13731,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15431,24 +13741,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -15474,7 +13782,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15483,16 +13791,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15506,13 +13813,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15538,14 +13841,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15559,7 +13854,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15573,18 +13868,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15598,7 +13885,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -15610,19 +13897,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15656,7 +13932,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15666,17 +13942,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -15699,22 +13973,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15724,11 +13997,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15748,14 +14017,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15783,7 +14044,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15792,17 +14053,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15827,7 +14084,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -15846,23 +14102,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15874,13 +14129,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -15893,7 +14144,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.CapacityReservation] @@ -15916,17 +14166,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15937,14 +14186,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -15954,11 +14203,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15970,12 +14219,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16000,7 +14245,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16026,7 +14271,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16035,16 +14280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16064,10 +14308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -16095,14 +14335,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16116,7 +14348,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16132,18 +14364,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16157,7 +14381,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -16170,20 +14394,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16217,7 +14430,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16227,24 +14440,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16270,7 +14481,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16279,16 +14490,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16308,10 +14518,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -16339,14 +14545,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16360,7 +14558,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16376,18 +14574,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16401,7 +14591,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -16414,20 +14604,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16461,7 +14641,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16471,17 +14651,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -16500,22 +14678,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16525,11 +14702,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16544,14 +14717,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16579,7 +14744,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16588,17 +14753,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16620,7 +14781,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -16639,23 +14799,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16667,13 +14826,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -16685,7 +14840,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16709,17 +14863,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16730,14 +14883,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -16747,11 +14900,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16763,12 +14916,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16805,21 +14954,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16829,11 +14977,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -16845,14 +14989,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16879,7 +15015,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16888,17 +15024,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16917,21 +15049,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16941,11 +15072,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -16959,14 +15086,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16993,7 +15112,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17002,17 +15121,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17031,21 +15146,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17055,11 +15169,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -17073,14 +15183,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17107,7 +15209,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17116,17 +15218,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -17138,7 +15236,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -17157,21 +15254,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17183,16 +15279,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17218,7 +15310,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17244,7 +15336,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17254,16 +15346,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17283,10 +15374,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -17316,14 +15403,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17338,7 +15417,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17355,18 +15434,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17381,7 +15452,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -17395,20 +15466,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17443,7 +15503,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17453,17 +15513,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -17471,7 +15529,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17497,7 +15555,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17507,16 +15565,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17528,13 +15585,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -17565,14 +15618,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17587,7 +15632,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17604,18 +15649,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17630,7 +15667,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -17644,20 +15681,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17692,7 +15719,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17702,17 +15729,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -17731,23 +15756,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17757,11 +15781,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17777,14 +15797,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17813,7 +15825,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17822,17 +15834,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17857,7 +15865,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -17876,7 +15883,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17884,16 +15891,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17905,13 +15911,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -17933,7 +15935,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17952,23 +15953,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17980,13 +15980,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -18014,7 +16010,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -18043,7 +16041,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18052,16 +16050,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18071,11 +16068,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -18103,14 +16096,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18122,7 +16107,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18137,18 +16122,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18160,7 +16137,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -18172,20 +16151,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18216,7 +16185,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18225,17 +16194,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18254,22 +16219,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18279,11 +16243,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -18298,14 +16258,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18333,7 +16285,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18342,17 +16294,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18371,22 +16319,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18396,11 +16343,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -18416,14 +16359,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18451,7 +16386,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18460,24 +16395,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -18503,7 +16434,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18512,16 +16443,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18541,10 +16471,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload async def begin_update( self, @@ -18571,14 +16497,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18592,7 +16510,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18608,18 +16526,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18633,7 +16543,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -18646,19 +16556,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18692,7 +16591,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18702,17 +16601,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -18737,23 +16634,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18763,11 +16659,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18790,14 +16682,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18826,7 +16710,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18835,17 +16719,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18869,7 +16749,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -18888,23 +16767,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18916,13 +16794,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -18936,7 +16810,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -18955,22 +16828,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18982,13 +16854,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19017,7 +16885,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19041,7 +16908,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -19049,12 +16916,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19065,14 +16931,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -19082,11 +16948,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19098,10 +16964,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -19125,23 +16987,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19151,11 +17012,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -19181,14 +17038,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19217,7 +17066,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19226,17 +17075,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19255,22 +17100,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19280,11 +17124,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -19298,14 +17138,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19333,7 +17165,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19342,17 +17174,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19371,22 +17199,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19396,11 +17223,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -19414,14 +17237,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19449,7 +17264,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19458,17 +17273,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19487,22 +17298,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19512,11 +17322,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -19531,14 +17337,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19566,7 +17364,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19575,17 +17373,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -19611,7 +17405,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -19630,23 +17423,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19658,13 +17450,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19683,22 +17471,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19708,11 +17495,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -19726,14 +17509,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19761,7 +17536,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19770,17 +17545,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -19794,7 +17565,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19813,22 +17583,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19838,18 +17607,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -19875,7 +17640,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -19884,16 +17649,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19907,13 +17671,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -19939,14 +17699,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -19960,7 +17712,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19974,18 +17726,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -19999,7 +17743,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -20011,19 +17755,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -20057,7 +17790,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20069,17 +17802,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -20103,7 +17834,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20128,23 +17859,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20158,13 +17888,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20186,14 +17912,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20203,7 +17921,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20212,18 +17930,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20233,7 +17943,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20241,19 +17951,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20285,7 +17985,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20298,20 +17998,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20336,23 +18034,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20366,13 +18063,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20393,14 +18086,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20410,7 +18095,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20418,18 +18103,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20439,7 +18116,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20447,19 +18124,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20491,7 +18157,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20504,17 +18170,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20543,7 +18207,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20567,16 +18230,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20587,14 +18249,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20604,11 +18266,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20620,10 +18282,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20632,7 +18290,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20651,21 +18308,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20677,20 +18333,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20716,7 +18368,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20725,16 +18377,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20754,10 +18405,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20784,14 +18431,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20805,7 +18444,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20821,18 +18460,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20846,7 +18477,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -20859,19 +18490,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20905,7 +18525,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20915,24 +18535,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20958,7 +18576,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20967,16 +18585,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20988,13 +18605,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21022,14 +18635,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21043,7 +18648,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21059,18 +18664,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21084,7 +18681,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21097,19 +18694,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21143,7 +18730,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21153,17 +18740,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21182,22 +18767,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21207,11 +18791,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21226,14 +18806,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21261,7 +18833,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21270,17 +18842,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21302,7 +18870,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21321,23 +18888,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21349,13 +18915,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21369,7 +18931,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21393,18 +18954,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21415,14 +18975,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21432,11 +18992,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21448,12 +19008,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21479,7 +19035,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21505,7 +19061,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21515,16 +19071,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21544,10 +19099,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21576,14 +19127,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21598,7 +19141,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21615,18 +19158,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21641,7 +19176,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21655,19 +19190,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21702,7 +19226,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21712,17 +19236,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21730,7 +19252,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21756,7 +19278,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21766,16 +19288,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21787,13 +19308,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21823,14 +19340,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21845,7 +19354,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21862,18 +19371,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21888,7 +19389,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -21902,19 +19403,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21949,7 +19440,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21959,17 +19450,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -21988,23 +19477,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22014,11 +19502,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22034,14 +19518,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22070,7 +19546,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22079,17 +19555,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22114,7 +19586,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22133,7 +19604,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22141,16 +19612,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22162,13 +19632,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22190,7 +19656,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22214,19 +19679,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22237,14 +19701,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22254,11 +19718,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22270,10 +19734,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } - class ResourceSkusOperations: """ @@ -22307,7 +19767,6 @@ def list( :keyword include_extended_locations: To Include Extended Locations information or not in the response. Default value is None. :paramtype include_extended_locations: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.ResourceSku] @@ -22330,17 +19789,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, filter=filter, include_extended_locations=include_extended_locations, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22351,14 +19809,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -22368,11 +19826,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22384,8 +19842,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} - class GalleriesOperations: """ @@ -22408,7 +19864,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -22433,7 +19889,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -22441,16 +19897,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22473,10 +19928,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -22501,14 +19952,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22520,7 +19963,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22535,18 +19978,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22555,7 +19990,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -22566,19 +20001,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_07_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_07_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22610,7 +20034,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22620,20 +20044,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -22658,7 +20084,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -22666,16 +20092,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22687,13 +20112,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22718,14 +20139,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22737,7 +20150,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22751,18 +20164,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22771,7 +20176,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -22782,19 +20191,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_07_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_07_01.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -22826,7 +20224,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22836,17 +20234,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -22866,7 +20262,6 @@ async def get( :keyword select: The select expression to apply on the operation. "Permissions" Default value is None. :paramtype select: str or ~azure.mgmt.compute.v2021_07_01.models.SelectPermissions - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -22885,22 +20280,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22912,13 +20306,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -22937,21 +20327,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22961,11 +20350,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -22975,14 +20360,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23009,7 +20386,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23018,17 +20395,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -23036,7 +20409,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -23058,16 +20430,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23078,14 +20449,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -23095,11 +20466,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23111,15 +20482,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -23141,15 +20507,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23160,14 +20525,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -23177,11 +20542,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23193,8 +20558,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -23221,7 +20584,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -23247,7 +20610,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -23256,16 +20619,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23288,10 +20650,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -23320,14 +20678,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23340,7 +20690,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23358,18 +20708,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23382,7 +20724,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -23397,19 +20739,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23442,7 +20773,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23452,24 +20783,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -23495,7 +20824,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -23504,16 +20833,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23525,13 +20853,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -23560,14 +20884,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23580,7 +20896,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23597,18 +20913,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23621,7 +20929,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -23636,19 +20944,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23681,7 +20978,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23691,17 +20988,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -23716,7 +21011,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -23735,22 +21029,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23762,13 +21055,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -23787,22 +21076,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23812,11 +21100,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -23831,14 +21115,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23866,7 +21142,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23875,17 +21151,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -23898,7 +21170,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -23921,17 +21192,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23942,14 +21212,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -23959,11 +21229,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23975,10 +21245,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -24006,7 +21272,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -24032,7 +21298,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -24042,16 +21308,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24074,10 +21339,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -24111,14 +21372,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24133,7 +21386,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24155,18 +21408,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24181,7 +21426,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -24200,19 +21445,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24247,7 +21482,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24257,17 +21492,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -24275,7 +21508,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -24301,7 +21534,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -24311,16 +21544,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24332,13 +21564,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -24373,14 +21601,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24395,7 +21615,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24417,18 +21637,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24443,7 +21655,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -24462,20 +21674,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -24510,7 +21711,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24520,17 +21721,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -24559,7 +21758,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -24578,7 +21776,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -24586,16 +21784,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24607,13 +21804,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -24637,23 +21830,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24663,11 +21855,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -24691,14 +21879,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24727,7 +21907,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -24736,17 +21916,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -24762,7 +21938,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion] @@ -24785,18 +21960,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24807,14 +21981,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -24824,11 +21998,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24840,10 +22014,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -24870,7 +22040,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -24896,7 +22066,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -24905,16 +22075,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24937,10 +22106,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -24969,14 +22134,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -24990,7 +22147,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25008,18 +22165,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -25033,7 +22182,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -25048,19 +22197,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -25094,7 +22233,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25104,24 +22243,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -25147,7 +22284,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25156,16 +22293,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25177,13 +22313,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -25213,14 +22345,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -25234,7 +22358,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25252,18 +22376,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -25277,7 +22393,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -25292,20 +22408,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -25339,7 +22444,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25349,17 +22454,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -25375,7 +22478,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -25394,22 +22496,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25421,13 +22522,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -25446,22 +22543,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25471,11 +22567,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -25491,14 +22583,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25526,7 +22610,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -25535,17 +22619,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -25558,7 +22638,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryApplication] @@ -25581,17 +22660,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25602,14 +22680,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -25619,11 +22697,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25635,10 +22713,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -25666,7 +22740,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -25692,7 +22766,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25702,16 +22776,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25734,10 +22807,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -25772,14 +22841,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25794,7 +22855,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25816,18 +22877,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25842,7 +22895,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -25861,21 +22914,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -25910,7 +22952,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25920,17 +22962,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -25938,7 +22978,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -25964,7 +23004,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -25974,16 +23014,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25995,13 +23034,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -26037,14 +23072,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -26059,7 +23086,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26081,18 +23108,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -26107,7 +23126,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -26126,20 +23145,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -26174,7 +23183,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26184,17 +23193,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -26223,7 +23230,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -26242,7 +23248,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -26250,16 +23256,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26271,13 +23276,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -26301,23 +23302,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26327,11 +23327,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -26355,14 +23351,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26391,7 +23379,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -26400,17 +23388,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -26426,7 +23410,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -26450,18 +23433,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26472,14 +23454,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -26489,11 +23471,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26505,10 +23487,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -26534,7 +23512,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -26560,7 +23538,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -26568,16 +23546,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26597,10 +23574,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -26622,14 +23595,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -26642,7 +23607,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26654,18 +23619,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -26678,7 +23635,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -26688,19 +23645,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2021_07_01.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2021_07_01.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -26733,7 +23679,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26743,17 +23689,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -26787,7 +23731,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.SharedGallery] @@ -26810,17 +23753,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26831,14 +23773,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -26848,11 +23790,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26864,10 +23806,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -26876,7 +23814,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -26895,21 +23832,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26921,13 +23857,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -26968,7 +23900,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImage] @@ -26991,18 +23922,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27013,14 +23943,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -27030,11 +23960,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27046,10 +23976,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -27063,7 +23989,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -27082,22 +24007,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27109,13 +24033,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -27160,7 +24080,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -27184,19 +24103,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27207,14 +24125,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -27224,11 +24142,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27240,10 +24158,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -27267,7 +24181,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -27286,23 +24199,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27314,13 +24226,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -27351,7 +24259,6 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -27370,21 +24277,20 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27396,13 +24302,9 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -27437,7 +24339,6 @@ async def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -27456,22 +24357,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27483,13 +24383,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore class CommunityGalleryImageVersionsOperations: @@ -27534,7 +24430,6 @@ async def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -27553,23 +24448,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27581,10 +24475,6 @@ async def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_compute_management_client_enums.py index 47b4b6aa66d3..9bea44b200de 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_compute_management_client_enums.py @@ -451,7 +451,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_models_py3.py index c5742a555e56..e4df9d4f64b6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -265,7 +261,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -290,7 +286,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -305,7 +301,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -371,7 +367,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -428,7 +424,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -527,7 +523,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet] @@ -840,7 +836,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -952,7 +948,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1054,7 +1050,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup] @@ -1214,7 +1210,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.CapacityReservation] @@ -1422,26 +1418,6 @@ class CommunityGallery(PirCommunityGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this community gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class CommunityGalleryImage(PirCommunityGalleryResource): # pylint: disable=too-many-instance-attributes """Specifies information about the gallery image definition that you want to create or update. @@ -1700,7 +1676,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1931,7 +1907,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -1970,7 +1946,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2147,7 +2123,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2263,7 +2239,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup] @@ -2467,7 +2443,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -2739,17 +2715,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2906,7 +2871,7 @@ class Gallery(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2995,7 +2960,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3096,7 +3061,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.GalleryApplication] @@ -3269,7 +3234,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3341,7 +3306,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion] @@ -3461,12 +3426,12 @@ def __init__( self.replication_mode = replication_mode -class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): +class GalleryApplicationVersionPublishingProfile(GalleryArtifactPublishingProfileBase): # pylint: disable=name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -3642,7 +3607,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2021_07_01.models.ManagedArtifact @@ -3746,7 +3711,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -3825,7 +3790,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4017,7 +3982,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -4057,7 +4022,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -4252,7 +4217,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4330,7 +4295,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion] @@ -4392,64 +4357,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) :vartype replication_mode: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationMode """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2021_07_01.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2021_07_01.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationMode - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -4564,7 +4471,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.Gallery] @@ -4609,32 +4516,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2021_07_01.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2021_07_01.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2021_07_01.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryUpdate(UpdateResourceDefinition): """Specifies information about the Shared Image Gallery that you want to update. @@ -4852,7 +4733,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5032,7 +4913,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_07_01.models.SubResource @@ -5140,7 +5021,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.Image] @@ -5174,7 +5055,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_07_01.models.SubResource @@ -5617,7 +5498,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -5650,7 +5531,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -5936,7 +5817,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.Usage] @@ -5970,7 +5851,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6173,7 +6054,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -6343,7 +6224,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2021_07_01.models.OrchestrationServiceNames @@ -6416,7 +6297,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -6578,7 +6459,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -6611,18 +6492,6 @@ class OSDiskImageEncryption(DiskImageEncryption): :vartype disk_encryption_set_id: str """ - _attribute_map = { - "disk_encryption_set_id": {"key": "diskEncryptionSetId", "type": "str"}, - } - - def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword disk_encryption_set_id: A relative URI containing the resource ID of the disk - encryption set. - :paramtype disk_encryption_set_id: str - """ - super().__init__(disk_encryption_set_id=disk_encryption_set_id, **kwargs) - class OSProfile(_serialization.Model): """Specifies the operating system settings for the virtual machine. Some of the settings cannot be @@ -7047,7 +6916,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7141,7 +7010,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup] @@ -7179,17 +7048,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -7230,7 +7088,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2021_07_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2021_07_01.models.PublicIPAddressSkuTier """ @@ -7249,8 +7107,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2021_07_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2021_07_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -7261,7 +7118,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -7434,7 +7291,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7861,7 +7718,7 @@ def __init__(self, **kwargs: Any) -> None: class ResourceSkusResult(_serialization.Model): """The List Resource Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the subscription. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.ResourceSku] @@ -7995,7 +7852,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8670,7 +8527,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8733,7 +8590,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -8796,7 +8653,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -8870,7 +8727,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -8917,7 +8774,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -8950,7 +8807,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.RunCommandDocumentBase] @@ -8986,7 +8843,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -9209,24 +9066,6 @@ class SharedGallery(PirSharedGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this shared gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class SharedGalleryImage(PirSharedGalleryResource): # pylint: disable=too-many-instance-attributes """Specifies information about the gallery image definition that you want to create or update. @@ -9349,7 +9188,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImage] @@ -9440,7 +9279,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImageVersion] @@ -9476,7 +9315,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.SharedGallery] @@ -9584,7 +9423,7 @@ def __init__( class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. :code:`
    `:code:`
    ` Possible values are: :code:`
    `:code:`
    ` **Add** @@ -9783,7 +9622,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -9793,7 +9632,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -9822,7 +9661,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -9837,7 +9676,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9893,7 +9732,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource] @@ -10075,7 +9914,7 @@ def __init__( class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -10170,7 +10009,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -10191,67 +10030,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -10320,7 +10098,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -10463,7 +10241,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -10536,7 +10314,7 @@ def __init__(self, *, value: Optional[str] = None, localized_value: Optional[str class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -10583,7 +10361,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. @@ -10764,7 +10542,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11250,7 +11028,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -11340,7 +11118,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11478,7 +11256,7 @@ def __init__( self.suppress_failures = suppress_failures -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -11522,7 +11300,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11823,7 +11601,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_07_01.models.UserAssignedIdentitiesValue] """ @@ -11855,7 +11633,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2021_07_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_07_01.models.UserAssignedIdentitiesValue] """ @@ -11869,7 +11647,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11932,7 +11710,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -12079,7 +11857,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -12404,7 +12182,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12437,10 +12215,10 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -12547,7 +12325,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12567,10 +12345,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -12736,10 +12514,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12838,10 +12616,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -12896,7 +12674,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13143,7 +12921,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand] @@ -13292,7 +13070,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13537,7 +13315,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -13782,10 +13560,10 @@ def __init__( self.suppress_failures = suppress_failures -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension] @@ -14007,7 +13785,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -14046,7 +13824,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -14057,7 +13835,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -14128,7 +13906,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -14155,7 +13933,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14310,10 +14088,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -14355,7 +14133,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet] @@ -14391,7 +14169,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetSku] @@ -14427,7 +14205,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet] @@ -14460,7 +14238,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -14504,10 +14282,10 @@ def __init__( self.disk_encryption_set = disk_encryption_set -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14610,7 +14388,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -14635,7 +14413,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -14666,7 +14444,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -14685,7 +14463,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -14946,10 +14724,10 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -15037,10 +14815,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -15067,7 +14847,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -15075,18 +14855,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -15395,7 +15163,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -15513,7 +15281,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -15614,12 +15382,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -15652,7 +15420,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -15787,7 +15555,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -15840,7 +15608,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -15988,7 +15756,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -16366,7 +16134,7 @@ def __init__( self.suppress_failures = suppress_failures -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -16389,7 +16157,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -16558,10 +16326,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -16705,7 +16473,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM] @@ -16738,7 +16506,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -17525,14 +17293,14 @@ def __init__( # pylint: disable=too-many-locals class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -17567,7 +17335,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -17581,7 +17349,7 @@ def __init__( self.configuration_reference = configuration_reference -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/operations/_operations.py index 5f200afd080f..a4a0d12f2d32 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_07_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -58,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -182,7 +182,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -238,7 +238,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -269,7 +269,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -305,7 +305,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -341,7 +341,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -368,7 +368,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -410,7 +410,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -436,7 +438,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -466,7 +468,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -500,7 +502,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -534,7 +536,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -597,7 +599,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +629,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -651,7 +655,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -787,7 +791,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -818,7 +822,9 @@ def build_dedicated_hosts_list_by_host_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -842,7 +848,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -996,7 +1002,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1027,7 +1033,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1060,7 +1066,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1091,7 +1097,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -1137,7 +1143,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1172,7 +1178,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1207,7 +1213,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1233,7 +1239,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1273,7 +1279,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1340,7 +1346,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1388,7 +1394,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1419,7 +1425,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1448,7 +1454,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1480,7 +1486,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1522,7 +1528,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1572,7 +1578,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1604,7 +1610,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1635,7 +1641,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1695,7 +1701,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1757,7 +1765,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1895,7 +1903,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1926,7 +1934,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1951,7 +1959,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1978,7 +1986,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2059,7 +2067,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2254,7 +2262,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2294,7 +2302,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2319,7 +2327,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2344,7 +2352,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2375,7 +2383,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2409,7 +2417,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2443,7 +2451,7 @@ def build_virtual_machines_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2473,7 +2481,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2507,7 +2515,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2541,7 +2549,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2573,7 +2581,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2611,7 +2619,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2642,7 +2650,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2680,7 +2688,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2711,7 +2719,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2741,7 +2749,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2767,7 +2777,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2798,7 +2808,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2829,7 +2839,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2867,7 +2877,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2898,7 +2908,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2929,7 +2939,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2960,7 +2970,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2991,7 +3001,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3022,7 +3032,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3053,7 +3063,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3084,7 +3094,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3121,7 +3131,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3152,7 +3162,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3336,7 +3346,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3390,7 +3400,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3426,7 +3436,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3462,7 +3472,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3495,7 +3505,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -3535,7 +3545,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3565,7 +3575,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3708,7 +3720,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3744,7 +3756,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3780,7 +3792,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3813,7 +3825,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -3853,7 +3865,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -3889,7 +3901,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -3922,7 +3934,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -3963,7 +3975,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4004,7 +4016,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4084,7 +4096,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4117,7 +4129,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4152,7 +4164,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4187,7 +4199,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4213,7 +4225,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -4253,7 +4265,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4284,7 +4296,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4309,7 +4321,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4334,7 +4346,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4359,7 +4371,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4390,7 +4402,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4431,7 +4443,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4472,7 +4484,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4510,7 +4522,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4552,7 +4564,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4592,7 +4604,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4624,7 +4636,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4650,7 +4662,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4676,7 +4688,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4711,7 +4723,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4745,7 +4757,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4785,7 +4797,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4817,7 +4829,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -4863,7 +4875,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4897,7 +4909,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4923,7 +4935,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4949,7 +4961,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4975,7 +4987,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5017,7 +5029,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5043,7 +5055,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5069,7 +5081,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5104,7 +5116,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5137,7 +5149,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5170,7 +5182,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5197,7 +5211,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5228,7 +5242,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5263,7 +5277,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5298,7 +5312,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5330,7 +5344,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5370,7 +5384,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5403,7 +5417,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5444,7 +5458,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5485,7 +5499,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5523,7 +5537,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5565,7 +5579,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5778,7 +5792,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5832,7 +5846,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5966,7 +5980,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5997,7 +6011,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -6038,7 +6052,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -6121,7 +6135,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -6159,7 +6173,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6191,7 +6205,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6226,7 +6240,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6293,7 +6307,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6325,7 +6339,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6356,7 +6370,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -6399,7 +6413,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -6442,7 +6456,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -6486,7 +6500,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -6526,7 +6540,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6558,7 +6572,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6729,7 +6743,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -6769,7 +6783,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -6838,7 +6852,7 @@ def build_community_galleries_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_get_request( +def build_community_gallery_images_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6870,7 +6884,7 @@ def build_community_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_get_request( +def build_community_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, @@ -6932,7 +6946,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -6956,14 +6969,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6974,14 +6986,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -6991,11 +7003,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7007,8 +7019,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -7051,7 +7061,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7062,7 +7071,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7074,11 +7083,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7089,7 +7097,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -7099,12 +7107,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7132,7 +7136,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -7140,16 +7144,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7161,13 +7164,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -7190,7 +7189,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7201,7 +7199,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7213,11 +7211,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7228,7 +7225,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -7238,12 +7235,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7271,7 +7264,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -7279,16 +7272,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7300,13 +7292,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7318,7 +7306,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7337,21 +7324,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7361,11 +7347,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -7375,7 +7357,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -7394,21 +7375,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7420,13 +7400,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -7437,7 +7413,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -7459,16 +7434,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7479,14 +7453,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -7496,11 +7470,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7512,17 +7486,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -7544,16 +7513,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7564,14 +7532,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -7581,11 +7549,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7597,10 +7565,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -7612,7 +7576,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -7634,17 +7597,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7655,14 +7617,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7672,11 +7634,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7688,10 +7650,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -7735,7 +7693,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7746,7 +7703,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7759,11 +7716,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7774,7 +7730,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -7784,12 +7740,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7817,7 +7769,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -7825,16 +7777,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7854,10 +7805,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -7880,7 +7827,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7891,7 +7837,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7904,11 +7850,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7919,7 +7864,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -7929,12 +7874,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7962,7 +7904,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -7970,16 +7912,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7991,13 +7932,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -8009,7 +7946,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8028,21 +7964,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8052,11 +7987,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -8076,7 +8007,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8095,22 +8025,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8122,19 +8051,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -8158,15 +8082,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8177,14 +8100,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -8194,11 +8117,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8210,10 +8133,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -8222,7 +8141,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -8246,16 +8164,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8266,14 +8183,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -8283,11 +8200,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8299,10 +8216,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -8347,7 +8260,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8358,7 +8270,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8372,11 +8284,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8387,7 +8298,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -8399,12 +8310,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8432,7 +8339,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -8440,16 +8347,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8469,10 +8375,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -8494,7 +8396,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8505,7 +8406,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8517,11 +8418,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8532,7 +8432,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -8542,12 +8442,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8575,7 +8471,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -8583,16 +8479,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8604,13 +8499,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -8622,7 +8513,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8641,21 +8531,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8665,11 +8554,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -8691,7 +8576,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -8710,22 +8594,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8737,13 +8620,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -8752,7 +8631,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -8774,16 +8652,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8794,14 +8671,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -8811,11 +8688,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8827,16 +8704,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -8858,15 +8730,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8877,14 +8748,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -8894,11 +8765,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8910,8 +8781,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -8938,7 +8807,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -8964,7 +8833,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -8973,16 +8842,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9002,10 +8870,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -9030,14 +8894,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9050,7 +8906,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9064,18 +8920,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9088,7 +8936,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -9100,19 +8948,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9145,7 +8982,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9155,24 +8992,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -9198,7 +9033,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -9207,16 +9042,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9228,13 +9062,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -9260,14 +9090,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9280,7 +9102,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9294,18 +9116,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9318,7 +9132,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -9330,19 +9144,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] @@ -9375,7 +9178,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9385,17 +9188,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -9414,22 +9215,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9439,11 +9239,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -9457,14 +9253,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9492,7 +9280,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9501,17 +9289,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -9535,7 +9319,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -9554,23 +9337,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9582,13 +9364,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -9601,7 +9379,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -9623,17 +9400,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9644,14 +9420,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -9661,11 +9437,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9677,10 +9453,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - class SshPublicKeysOperations: """ @@ -9707,7 +9479,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -9731,15 +9502,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9750,14 +9520,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -9767,11 +9537,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9783,8 +9553,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -9794,7 +9562,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -9818,16 +9585,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9838,14 +9604,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -9855,11 +9621,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9871,10 +9637,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -9896,7 +9658,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9907,7 +9668,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9919,11 +9680,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9934,7 +9694,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -9944,12 +9704,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9977,7 +9733,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -9985,16 +9741,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10014,10 +9769,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -10039,7 +9790,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10050,7 +9800,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10062,11 +9812,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10077,7 +9826,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -10087,12 +9836,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10120,7 +9866,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -10128,16 +9874,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10149,13 +9894,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -10167,7 +9908,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10186,21 +9926,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10210,11 +9949,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -10224,7 +9959,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -10243,21 +9977,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10269,13 +10002,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -10289,7 +10018,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -10308,21 +10036,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10334,13 +10061,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -10377,7 +10100,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -10396,23 +10118,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10424,13 +10145,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -10442,7 +10159,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -10461,21 +10177,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10487,13 +10202,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -10521,7 +10232,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -10540,7 +10250,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -10549,16 +10259,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10570,13 +10279,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -10604,7 +10309,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -10630,7 +10335,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -10639,16 +10344,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10668,10 +10372,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10698,14 +10398,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10719,7 +10411,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10735,18 +10427,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10760,7 +10444,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -10773,20 +10457,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10820,7 +10493,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10830,24 +10503,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -10873,7 +10544,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -10882,16 +10553,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10903,13 +10573,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -10938,14 +10604,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10959,7 +10617,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10975,18 +10633,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -11000,7 +10650,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -11013,20 +10663,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -11060,7 +10699,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11070,17 +10709,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -11099,22 +10736,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11124,11 +10760,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11143,14 +10775,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11178,7 +10802,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11187,17 +10811,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11219,7 +10839,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11238,23 +10857,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11266,13 +10884,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11286,7 +10900,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11305,22 +10918,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11332,13 +10944,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -11377,7 +10985,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -11396,7 +11003,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -11404,16 +11011,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11425,13 +11031,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11463,7 +11065,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11482,7 +11083,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -11492,16 +11093,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11513,13 +11113,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -11531,7 +11127,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11550,21 +11145,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11576,13 +11170,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -11590,7 +11180,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11609,20 +11198,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11634,13 +11222,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -11654,7 +11238,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11673,22 +11256,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11700,13 +11282,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -11747,7 +11325,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -11766,7 +11343,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -11775,16 +11352,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11796,13 +11372,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11839,7 +11411,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11858,7 +11429,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -11869,16 +11440,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11890,13 +11460,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -11911,7 +11477,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11930,22 +11495,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11957,13 +11521,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -11975,7 +11535,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11994,21 +11553,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12020,13 +11578,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -12043,7 +11597,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12062,23 +11615,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12090,13 +11642,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -12126,7 +11674,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -12148,16 +11695,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12168,14 +11714,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -12185,11 +11731,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12201,8 +11747,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -12231,7 +11775,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -12253,16 +11796,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12273,14 +11815,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12290,11 +11832,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12306,15 +11848,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -12340,7 +11878,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12348,16 +11886,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12371,13 +11908,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -12401,14 +11934,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12421,7 +11946,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12434,18 +11959,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12458,7 +11975,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -12469,19 +11986,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -12514,7 +12021,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12526,20 +12033,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -12564,7 +12073,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12572,16 +12081,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12601,10 +12109,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -12627,14 +12131,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12646,7 +12142,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12659,18 +12155,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12679,7 +12167,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -12689,19 +12181,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12733,7 +12214,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12743,20 +12224,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -12781,7 +12264,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12789,16 +12272,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12810,13 +12292,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -12839,14 +12317,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12858,7 +12328,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12870,18 +12340,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12890,7 +12352,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -12899,19 +12365,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] @@ -12943,7 +12398,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12953,17 +12408,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -12982,22 +12435,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13007,11 +12459,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13026,14 +12474,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13061,7 +12501,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13070,17 +12510,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13103,7 +12539,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -13122,22 +12557,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13149,13 +12583,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -13167,7 +12597,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13186,21 +12615,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13212,13 +12640,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13237,21 +12661,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13261,11 +12684,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13276,14 +12695,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13310,7 +12721,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13319,17 +12730,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -13348,22 +12755,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13373,11 +12779,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -13393,14 +12795,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13428,7 +12822,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13437,17 +12831,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -13464,7 +12854,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13483,21 +12872,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13507,11 +12895,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: @@ -13520,7 +12904,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -13542,16 +12925,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13562,14 +12944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -13579,11 +12961,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13595,10 +12977,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Iterable["_models.VirtualMachine"]: """Lists all of the virtual machines in the specified subscription. Use the nextLink property in @@ -13607,7 +12985,6 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera :keyword status_only: statusOnly=true enables fetching run time status of all Virtual Machines in the subscription. Default value is None. :paramtype status_only: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -13629,16 +13006,15 @@ def list_all(self, *, status_only: Optional[str] = None, **kwargs: Any) -> Itera def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13649,14 +13025,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -13666,11 +13042,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13682,8 +13058,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13695,7 +13069,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -13717,17 +13090,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13738,14 +13110,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -13755,11 +13127,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13771,10 +13143,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -13792,22 +13160,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13817,11 +13184,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13838,14 +13201,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13873,7 +13228,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13882,17 +13237,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13911,21 +13262,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13935,11 +13285,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13949,14 +13295,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13983,7 +13321,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13992,17 +13330,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14021,21 +13355,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14045,11 +13378,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14059,14 +13388,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14093,7 +13414,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14102,17 +13423,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14131,21 +13448,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14155,11 +13471,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14169,14 +13481,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14203,7 +13507,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14212,17 +13516,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14241,21 +13541,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14265,11 +13564,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14279,14 +13574,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14313,7 +13600,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14322,23 +13609,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -14367,7 +13650,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14375,16 +13658,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14394,11 +13676,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -14422,14 +13700,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14440,7 +13710,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -14453,18 +13723,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14475,7 +13737,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -14485,19 +13747,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -14527,7 +13779,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14536,17 +13788,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -14568,7 +13816,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14587,22 +13834,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14614,13 +13860,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14639,21 +13881,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14663,11 +13904,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -14677,14 +13914,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14711,7 +13940,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14720,17 +13949,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -14742,7 +13967,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14761,21 +13985,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14785,11 +14008,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -14808,21 +14027,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14836,13 +14054,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -14854,14 +14068,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -14891,7 +14097,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14903,23 +14109,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -14945,7 +14149,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14953,16 +14157,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14976,13 +14179,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -15007,14 +14206,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -15027,7 +14218,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15040,18 +14231,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -15064,7 +14247,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -15074,20 +14257,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -15120,7 +14292,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15132,20 +14304,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -15170,7 +14344,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15178,16 +14352,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15201,13 +14374,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -15230,14 +14399,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -15249,7 +14410,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15261,18 +14422,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -15281,7 +14434,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -15290,19 +14447,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -15334,7 +14480,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15346,17 +14492,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -15386,7 +14530,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15410,16 +14553,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15430,14 +14572,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -15447,11 +14589,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15463,15 +14605,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -15497,7 +14635,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15505,16 +14643,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15534,10 +14671,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -15559,14 +14692,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15579,7 +14704,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15591,18 +14716,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15615,7 +14732,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -15624,20 +14741,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15670,7 +14776,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15680,23 +14786,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -15722,7 +14826,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15730,16 +14834,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15751,13 +14854,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15780,14 +14879,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15800,7 +14891,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15812,18 +14903,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15836,7 +14919,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -15845,20 +14928,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15891,7 +14964,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15901,17 +14974,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15930,22 +15001,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15955,11 +15025,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15974,14 +15040,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16009,7 +15067,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16018,17 +15076,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -16049,7 +15103,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -16068,22 +15121,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16095,19 +15147,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16136,7 +15184,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16144,16 +15192,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16163,11 +15210,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -16194,14 +15237,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16212,7 +15247,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16227,18 +15262,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16249,7 +15276,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -16261,20 +15288,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16304,7 +15320,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16313,23 +15329,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -16357,7 +15369,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16366,16 +15378,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16385,11 +15396,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -16418,14 +15425,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16436,7 +15435,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -16450,21 +15449,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16475,7 +15466,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -16487,23 +15478,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16534,7 +15514,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16543,17 +15523,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -16565,7 +15541,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -16584,21 +15559,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16610,13 +15584,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -16624,7 +15594,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -16648,16 +15617,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16668,14 +15636,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -16685,11 +15653,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16701,17 +15669,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -16735,15 +15698,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16754,14 +15716,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -16771,11 +15733,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16787,8 +15749,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16800,7 +15760,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -16824,17 +15783,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16845,14 +15803,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -16862,11 +15820,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16878,10 +15836,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16892,7 +15846,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -16916,17 +15869,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16937,14 +15889,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -16954,11 +15906,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16970,15 +15922,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -17009,7 +15957,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17018,16 +15966,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17037,11 +15984,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -17073,14 +16016,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17091,7 +16026,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -17107,7 +16042,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -17115,14 +16050,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17133,7 +16060,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -17147,24 +16074,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17195,7 +16111,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17204,23 +16120,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17249,7 +16161,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17257,16 +16169,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17276,11 +16187,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -17305,14 +16212,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17323,7 +16222,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17336,18 +16235,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17358,7 +16249,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -17368,20 +16259,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17411,7 +16291,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17420,23 +16300,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17465,7 +16341,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17473,16 +16349,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17492,11 +16367,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -17521,14 +16392,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17539,7 +16402,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17552,18 +16415,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17574,7 +16429,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -17584,20 +16439,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17627,7 +16471,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17636,23 +16480,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17681,7 +16521,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17689,16 +16529,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17708,11 +16547,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -17738,14 +16573,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17756,7 +16583,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17770,18 +16597,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17792,7 +16611,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -17803,20 +16622,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17846,7 +16654,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17855,23 +16663,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17900,7 +16704,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17908,16 +16712,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17927,11 +16730,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -17959,14 +16758,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17977,7 +16768,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17993,18 +16784,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18015,7 +16798,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -18028,20 +16811,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18071,7 +16843,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18080,23 +16852,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -18122,7 +16890,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18130,16 +16898,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18149,11 +16916,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -18178,14 +16941,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18196,7 +16951,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18209,18 +16964,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18231,7 +16978,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -18241,20 +16988,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18284,7 +17020,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18293,23 +17029,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18338,7 +17070,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18346,16 +17078,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18365,11 +17096,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -18395,14 +17122,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18413,7 +17132,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18427,18 +17146,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18449,7 +17160,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -18461,20 +17172,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18504,7 +17204,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18513,23 +17213,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18558,7 +17254,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18566,16 +17262,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18585,11 +17280,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -18615,14 +17306,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18633,7 +17316,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18647,18 +17330,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18669,7 +17344,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -18680,20 +17355,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18723,7 +17387,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18732,20 +17396,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, *, platform_update_domain: int, **kwargs: Any ) -> _models.RecoveryWalkResponse: """Manual platform update domain walk to update virtual machines in a service fabric virtual @@ -18758,7 +17418,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword platform_update_domain: The platform update domain for which a manual recovery walk is requested. Required. :paramtype platform_update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -18777,22 +17436,21 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, platform_update_domain=platform_update_domain, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18804,13 +17462,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -18835,7 +17489,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18846,7 +17499,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18859,11 +17512,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18874,7 +17526,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -18885,13 +17537,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18919,7 +17568,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18927,16 +17576,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18946,17 +17594,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -18982,7 +17626,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18990,16 +17634,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19009,11 +17652,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -19037,14 +17676,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19055,7 +17686,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19068,18 +17699,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19090,7 +17713,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -19101,19 +17724,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -19143,7 +17756,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19152,17 +17765,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -19192,7 +17801,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -19214,16 +17822,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19234,14 +17841,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -19251,11 +17858,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19267,8 +17874,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -19291,7 +17896,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -19316,7 +17921,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -19324,16 +17929,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19353,10 +17957,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -19378,14 +17978,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19396,7 +17988,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19408,18 +18000,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19427,7 +18011,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -19436,19 +18020,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19479,7 +18052,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19489,20 +18062,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -19527,7 +18102,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -19535,16 +18110,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19564,10 +18138,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -19589,14 +18159,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19607,7 +18169,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19619,18 +18181,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19638,7 +18192,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -19647,19 +18205,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19690,7 +18237,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19700,17 +18247,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -19729,21 +18274,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19753,11 +18297,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -19767,14 +18307,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19801,7 +18333,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19810,17 +18342,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19834,7 +18362,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -19853,22 +18380,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19880,13 +18406,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -19894,7 +18416,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19916,16 +18437,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19936,14 +18456,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19953,11 +18473,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19969,16 +18489,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -20000,15 +18515,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20019,14 +18533,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -20036,11 +18550,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20052,8 +18566,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -20099,7 +18611,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20110,7 +18621,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20125,11 +18636,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20140,7 +18650,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -20152,12 +18662,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20185,7 +18691,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -20193,16 +18699,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20222,10 +18727,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -20248,7 +18749,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20259,7 +18759,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20272,11 +18772,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20287,7 +18786,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -20297,12 +18796,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20330,7 +18826,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -20338,16 +18834,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20359,13 +18854,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -20384,21 +18875,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20408,11 +18898,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20425,14 +18911,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20459,7 +18937,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20468,17 +18946,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -20500,7 +18974,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -20519,22 +18992,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20546,13 +19018,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -20560,7 +19028,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -20584,16 +19051,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20604,14 +19070,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -20621,11 +19087,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20637,17 +19103,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -20671,15 +19132,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20690,14 +19150,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -20707,11 +19167,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20723,8 +19183,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -20751,7 +19209,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -20777,7 +19235,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -20786,16 +19244,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20807,13 +19264,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -20840,14 +19293,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -20860,7 +19305,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20875,18 +19320,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -20899,7 +19336,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -20912,19 +19349,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RestorePoint] @@ -20957,7 +19383,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20967,17 +19393,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -20996,22 +19420,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21021,11 +19444,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21039,14 +19458,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21074,7 +19485,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21083,17 +19494,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21107,7 +19514,6 @@ def get( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -21126,22 +19532,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21153,13 +19558,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -21205,7 +19606,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21216,7 +19616,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21230,11 +19630,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21245,7 +19644,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -21257,12 +19656,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21290,7 +19685,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -21298,16 +19693,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21327,10 +19721,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -21354,7 +19744,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21365,7 +19754,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21379,11 +19768,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21394,7 +19782,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -21405,12 +19793,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21438,7 +19823,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -21446,16 +19831,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21467,13 +19851,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -21488,7 +19868,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21507,21 +19886,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21531,11 +19909,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -21559,7 +19933,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21578,22 +19951,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21605,13 +19977,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -21632,7 +20000,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -21656,17 +20023,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21677,14 +20043,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -21694,11 +20060,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21710,10 +20076,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -21727,7 +20089,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -21751,16 +20112,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21771,14 +20131,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -21788,11 +20148,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21804,10 +20164,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -21834,7 +20190,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -21860,7 +20216,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21869,16 +20225,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21898,10 +20253,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -21928,14 +20279,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21949,7 +20292,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21965,18 +20308,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21990,7 +20325,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -22004,19 +20339,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22050,7 +20374,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22060,24 +20384,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -22103,7 +20425,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -22112,16 +20434,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22135,13 +20456,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22167,14 +20484,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22188,7 +20497,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22202,18 +20511,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22227,7 +20528,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -22239,19 +20540,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -22285,7 +20575,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22295,17 +20585,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -22328,22 +20616,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22353,11 +20640,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -22377,14 +20660,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22412,7 +20687,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22421,17 +20696,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -22456,7 +20727,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -22475,23 +20745,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22503,13 +20772,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -22522,7 +20787,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.CapacityReservation] @@ -22545,17 +20809,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22566,14 +20829,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -22583,11 +20846,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22599,12 +20862,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -22629,7 +20888,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -22655,7 +20914,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -22664,16 +20923,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22693,10 +20951,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -22724,14 +20978,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22745,7 +20991,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22761,18 +21007,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22786,7 +21024,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -22799,20 +21037,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22846,7 +21073,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22856,24 +21083,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -22899,7 +21124,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -22908,16 +21133,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22937,10 +21161,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -22968,14 +21188,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22989,7 +21201,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23005,18 +21217,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -23030,7 +21234,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -23043,20 +21247,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -23090,7 +21284,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23100,17 +21294,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -23129,22 +21321,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23154,11 +21345,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23173,14 +21360,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23208,7 +21387,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23217,17 +21396,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23249,7 +21424,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -23268,23 +21442,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23296,13 +21469,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -23314,7 +21483,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -23338,17 +21506,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23359,14 +21526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -23376,11 +21543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23392,12 +21559,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23434,21 +21597,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23458,11 +21620,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23472,14 +21630,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23506,7 +21656,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23515,17 +21665,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -23544,21 +21690,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23568,11 +21713,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -23586,14 +21727,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23620,7 +21753,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23629,17 +21762,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -23658,21 +21787,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23682,11 +21810,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -23700,14 +21824,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23734,7 +21850,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23743,17 +21859,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -23765,7 +21877,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -23784,21 +21895,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23810,16 +21920,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23845,7 +21951,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -23871,7 +21977,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23881,16 +21987,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23910,10 +22015,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -23943,14 +22044,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23965,7 +22058,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23982,18 +22075,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24008,7 +22093,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -24022,20 +22107,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24070,7 +22144,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24080,17 +22154,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -24098,7 +22170,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -24124,7 +22196,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24134,16 +22206,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24155,13 +22226,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -24192,14 +22259,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24214,7 +22273,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24231,18 +22290,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24257,7 +22308,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -24271,20 +22322,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -24319,7 +22360,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24329,17 +22370,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -24358,23 +22397,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24384,11 +22422,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24404,14 +22438,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24440,7 +22466,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24449,17 +22475,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24484,7 +22506,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -24503,7 +22524,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24511,16 +22532,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24532,13 +22552,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24560,7 +22576,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -24579,23 +22594,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24607,13 +22621,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -24641,7 +22651,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -24670,7 +22682,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24679,16 +22691,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24698,11 +22709,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -24730,14 +22737,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24749,7 +22748,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -24764,18 +22763,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24787,7 +22778,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -24799,20 +22792,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -24843,7 +22826,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24852,17 +22835,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24881,22 +22860,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24906,11 +22884,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -24925,14 +22899,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24960,7 +22926,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24969,17 +22935,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24998,22 +22960,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25023,11 +22984,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -25043,14 +23000,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25078,7 +23027,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25087,24 +23036,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -25130,7 +23075,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25139,16 +23084,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25168,10 +23112,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } - @overload def begin_update( self, @@ -25198,14 +23138,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25219,7 +23151,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25235,18 +23167,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25260,7 +23184,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -25273,19 +23197,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25319,7 +23232,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25329,17 +23242,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25364,23 +23275,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25390,11 +23300,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25417,14 +23323,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25453,7 +23351,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25462,17 +23360,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25496,7 +23390,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -25515,23 +23408,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25543,13 +23435,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -25563,7 +23451,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -25582,22 +23469,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25609,13 +23495,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -25644,7 +23526,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -25668,7 +23549,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -25676,12 +23557,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25692,14 +23572,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -25709,11 +23589,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25725,10 +23605,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -25752,23 +23628,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25778,11 +23653,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -25808,14 +23679,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25844,7 +23707,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25853,17 +23716,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25882,22 +23741,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25907,11 +23765,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -25925,14 +23779,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25960,7 +23806,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25969,17 +23815,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25998,22 +23840,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26023,11 +23864,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -26041,14 +23878,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26076,7 +23905,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26085,17 +23914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26114,22 +23939,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26139,11 +23963,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -26158,14 +23978,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26193,7 +24005,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26202,17 +24014,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -26238,7 +24046,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -26257,23 +24064,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26285,13 +24091,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -26310,22 +24112,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26335,11 +24136,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -26353,14 +24150,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26388,7 +24177,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26397,17 +24186,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -26421,7 +24206,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -26440,22 +24224,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26465,18 +24248,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -26502,7 +24281,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -26511,16 +24290,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26534,13 +24312,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -26566,14 +24340,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -26586,7 +24352,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26600,18 +24366,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -26624,7 +24382,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -26636,19 +24394,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.RunCommandResult] @@ -26681,7 +24428,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26693,17 +24440,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -26727,7 +24472,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26752,23 +24497,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26782,13 +24526,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26810,14 +24550,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26827,7 +24559,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26836,18 +24568,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26857,7 +24581,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26865,19 +24589,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26909,7 +24623,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26921,20 +24635,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26959,23 +24671,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26989,13 +24700,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -27016,14 +24723,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27033,7 +24732,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -27041,18 +24740,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27062,7 +24753,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -27070,19 +24761,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27114,7 +24794,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27126,17 +24806,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -27165,7 +24843,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -27189,16 +24866,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27209,14 +24885,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -27226,11 +24902,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27242,10 +24918,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -27254,7 +24926,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -27273,21 +24944,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27299,20 +24969,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27338,7 +25004,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27347,16 +25013,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27376,10 +25041,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27406,14 +25067,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27427,7 +25080,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27443,18 +25096,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27468,7 +25113,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -27481,19 +25126,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27527,7 +25161,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27537,24 +25171,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27580,7 +25212,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27589,16 +25221,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27610,13 +25241,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27644,14 +25271,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27665,7 +25284,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27681,18 +25300,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27706,7 +25317,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -27719,19 +25330,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27765,7 +25366,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27775,17 +25376,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27804,22 +25403,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27829,11 +25427,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27848,14 +25442,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27883,7 +25469,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27892,17 +25478,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27924,7 +25506,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27943,23 +25524,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27971,13 +25551,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27991,7 +25567,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28015,18 +25590,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28037,14 +25611,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28054,11 +25628,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28070,12 +25644,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -28101,7 +25671,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28127,7 +25697,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28137,16 +25707,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28166,10 +25735,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -28198,14 +25763,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28220,7 +25777,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28237,18 +25794,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28263,7 +25812,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -28277,19 +25826,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28324,7 +25862,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28334,17 +25872,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -28352,7 +25888,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28378,7 +25914,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28388,16 +25924,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28409,13 +25944,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -28445,14 +25976,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28467,7 +25990,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28484,18 +26007,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28510,7 +26025,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -28524,19 +26039,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28571,7 +26076,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28581,17 +26086,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -28610,23 +26113,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28636,11 +26138,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28656,14 +26154,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28692,7 +26182,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28701,17 +26191,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -28736,7 +26222,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -28755,7 +26240,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28763,16 +26248,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28784,13 +26268,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28812,7 +26292,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28836,19 +26315,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28859,14 +26337,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28876,11 +26354,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28892,10 +26370,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } - class ResourceSkusOperations: """ @@ -28929,7 +26403,6 @@ def list( :keyword include_extended_locations: To Include Extended Locations information or not in the response. Default value is None. :paramtype include_extended_locations: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ResourceSku or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.ResourceSku] :raises ~azure.core.exceptions.HttpResponseError: @@ -28951,17 +26424,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_resource_skus_list_request( + _request = build_resource_skus_list_request( subscription_id=self._config.subscription_id, filter=filter, include_extended_locations=include_extended_locations, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28972,14 +26444,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceSkusResult", pipeline_response) @@ -28989,11 +26461,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29005,8 +26477,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/skus"} - class GalleriesOperations: """ @@ -29029,7 +26499,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -29054,7 +26524,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -29062,16 +26532,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29094,10 +26563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -29122,14 +26587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29140,7 +26597,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29155,18 +26612,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29174,7 +26623,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -29185,19 +26634,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_07_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_07_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29228,7 +26666,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29238,20 +26676,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -29276,7 +26716,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -29284,16 +26724,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29305,13 +26744,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -29336,14 +26771,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29354,7 +26781,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29368,18 +26795,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29387,7 +26806,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -29398,19 +26821,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_07_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_07_01.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29441,7 +26853,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29451,17 +26863,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -29481,7 +26891,6 @@ def get( :keyword select: The select expression to apply on the operation. "Permissions" Default value is None. :paramtype select: str or ~azure.mgmt.compute.v2021_07_01.models.SelectPermissions - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -29500,22 +26909,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29527,13 +26935,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -29552,21 +26956,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29576,11 +26979,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -29590,14 +26989,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29624,7 +27015,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -29633,17 +27024,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -29651,7 +27038,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29673,16 +27059,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29693,14 +27078,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -29710,11 +27095,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29726,15 +27111,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -29756,15 +27136,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29775,14 +27154,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -29792,11 +27171,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29808,8 +27187,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -29836,7 +27213,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -29862,7 +27239,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -29871,16 +27248,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29903,10 +27279,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -29935,14 +27307,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -29955,7 +27319,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29973,18 +27337,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -29997,7 +27353,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -30012,19 +27368,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -30057,7 +27402,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30067,24 +27412,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -30110,7 +27453,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -30119,16 +27462,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30140,13 +27482,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -30175,14 +27513,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -30195,7 +27525,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30212,18 +27542,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -30236,7 +27558,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -30251,19 +27573,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] @@ -30296,7 +27607,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30306,17 +27617,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -30331,7 +27640,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -30350,22 +27658,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30377,13 +27684,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -30402,22 +27705,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30427,11 +27729,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -30446,14 +27744,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -30481,7 +27771,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -30490,17 +27780,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -30513,7 +27799,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -30535,17 +27820,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -30556,14 +27840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -30573,11 +27857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30589,10 +27873,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -30620,7 +27900,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -30646,7 +27926,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -30656,16 +27936,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30688,10 +27967,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -30725,14 +28000,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -30747,7 +28014,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -30769,18 +28036,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -30795,7 +28054,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -30814,19 +28073,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -30861,7 +28110,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -30871,17 +28120,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -30889,7 +28136,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -30915,7 +28162,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -30925,16 +28172,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -30946,13 +28192,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -30987,14 +28229,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -31009,7 +28243,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -31031,18 +28265,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -31057,7 +28283,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -31076,20 +28302,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -31124,7 +28339,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -31134,17 +28349,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -31173,7 +28386,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -31192,7 +28404,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -31200,16 +28412,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31221,13 +28432,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -31251,23 +28458,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31277,11 +28483,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -31305,14 +28507,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -31341,7 +28535,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -31350,17 +28544,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -31376,7 +28566,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryImageVersion] @@ -31399,18 +28588,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -31421,14 +28609,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -31438,11 +28626,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31454,10 +28642,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -31484,7 +28668,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -31510,7 +28694,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -31519,16 +28703,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31551,10 +28734,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -31583,14 +28762,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31604,7 +28775,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -31622,18 +28793,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31647,7 +28810,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -31662,19 +28825,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31708,7 +28861,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -31718,24 +28871,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -31761,7 +28912,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -31770,16 +28921,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -31791,13 +28941,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -31827,14 +28973,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31848,7 +28986,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -31866,18 +29004,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31891,7 +29021,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -31906,20 +29036,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -31953,7 +29072,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -31963,17 +29082,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -31989,7 +29106,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -32008,22 +29124,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32035,13 +29150,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -32060,22 +29171,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32085,11 +29195,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -32105,14 +29211,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -32140,7 +29238,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -32149,17 +29247,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -32172,7 +29266,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -32194,17 +29287,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -32215,14 +29307,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -32232,11 +29324,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32248,10 +29340,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -32279,7 +29367,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -32305,7 +29393,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -32315,16 +29403,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32347,10 +29434,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -32385,14 +29468,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32407,7 +29482,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -32429,18 +29504,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32455,7 +29522,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -32474,21 +29541,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32523,7 +29579,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -32533,17 +29589,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -32551,7 +29605,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -32577,7 +29631,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -32587,16 +29641,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32608,13 +29661,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -32650,14 +29699,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32672,7 +29713,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -32694,18 +29735,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32720,7 +29753,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -32739,20 +29772,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -32787,7 +29810,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -32797,17 +29820,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -32836,7 +29857,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_07_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -32855,7 +29875,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -32863,16 +29883,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32884,13 +29903,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -32914,23 +29929,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -32940,11 +29954,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -32968,14 +29978,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -33004,7 +30006,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -33013,17 +30015,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -33039,7 +30037,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -33063,18 +30060,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33085,14 +30081,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -33102,11 +30098,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33118,10 +30114,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -33147,7 +30139,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -33173,7 +30165,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -33181,16 +30173,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33210,10 +30201,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -33235,14 +30222,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.SharingUpdate] @@ -33254,7 +30233,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -33266,18 +30245,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.SharingUpdate] @@ -33289,7 +30260,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -33299,19 +30270,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2021_07_01.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2021_07_01.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_07_01.models.SharingUpdate] @@ -33343,7 +30303,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -33353,17 +30313,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -33397,7 +30355,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -33419,17 +30376,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33440,14 +30396,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -33457,11 +30413,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33473,10 +30429,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -33485,7 +30437,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -33504,21 +30455,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33530,13 +30480,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -33577,7 +30523,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -33599,18 +30544,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33621,14 +30565,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -33638,11 +30582,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33654,10 +30598,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -33671,7 +30611,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -33690,22 +30629,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33717,13 +30655,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -33768,7 +30702,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2021_07_01.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -33792,19 +30725,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -33815,14 +30747,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -33832,11 +30764,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33848,10 +30780,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -33875,7 +30803,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -33894,23 +30821,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -33922,13 +30848,9 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -33959,7 +30881,6 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -33978,21 +30899,20 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -34004,13 +30924,9 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -34045,7 +30961,6 @@ def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -34064,22 +30979,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -34091,13 +31005,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore class CommunityGalleryImageVersionsOperations: @@ -34142,7 +31052,6 @@ def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_07_01.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -34161,23 +31070,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-07-01")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -34189,10 +31097,6 @@ def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_compute_management_client.py index e46d60224014..03742cacb924 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-08-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_configuration.py index 2c77dee6e35a..d2a8484d7de6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-08-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_metadata.json index d7b05a6b7ae1..8fd71ee7e344 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_compute_management_client.py index ebad0e33bbd6..4e222f7db4fc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-08-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_configuration.py index 033a35bd68a4..0194ceebf28d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-08-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/operations/_operations.py index a61cb76b8569..e94d68d70901 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_08_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_08_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_08_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_08_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1238,7 +1086,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1111,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1119,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1147,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1327,14 +1170,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1346,7 +1181,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1360,18 +1195,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1380,7 +1207,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1391,19 +1218,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1435,7 +1251,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1445,20 +1261,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1483,7 +1301,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1491,16 +1309,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1520,10 +1337,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1548,14 +1361,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1567,7 +1372,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1582,18 +1387,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1602,7 +1399,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1613,19 +1414,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1657,7 +1447,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1667,17 +1457,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1689,7 +1477,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1708,21 +1495,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1734,13 +1520,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1759,21 +1541,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1783,11 +1564,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1799,14 +1576,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1833,7 +1602,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1842,17 +1611,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1860,7 +1625,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1883,16 +1647,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1903,14 +1666,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1920,11 +1683,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1936,15 +1699,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.Snapshot] @@ -1967,15 +1725,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1986,14 +1743,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2003,11 +1760,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2019,13 +1776,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2051,7 +1806,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2059,16 +1814,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2082,13 +1836,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2114,14 +1864,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -2133,7 +1875,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2148,18 +1890,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -2171,7 +1905,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2183,19 +1917,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -2227,7 +1950,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2239,17 +1962,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2268,21 +1989,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2292,11 +2012,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2310,14 +2026,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2344,7 +2052,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2355,17 +2063,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2392,7 +2096,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2418,7 +2122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2426,16 +2130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2455,10 +2158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2483,14 +2182,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2503,7 +2194,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2518,18 +2209,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2542,7 +2225,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2554,19 +2237,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2599,7 +2272,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2609,23 +2282,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2651,7 +2322,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2330,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2688,10 +2358,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2716,14 +2382,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2736,7 +2394,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2751,18 +2409,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2775,7 +2425,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2787,19 +2437,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2832,7 +2473,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2866,7 +2505,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2885,21 +2523,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2911,13 +2548,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2936,21 +2569,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2960,11 +2592,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2978,14 +2606,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3012,7 +2632,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3021,17 +2641,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3041,7 +2657,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3064,16 +2679,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3084,14 +2698,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3101,11 +2715,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3117,15 +2731,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3148,15 +2757,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3167,14 +2775,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3184,11 +2792,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3200,8 +2808,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3214,7 +2820,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3236,17 +2841,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3257,14 +2861,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3274,11 +2878,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3290,10 +2894,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -3316,7 +2916,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3341,7 +2945,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3349,16 +2953,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3378,10 +2981,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3406,14 +3005,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3425,7 +3016,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3440,18 +3031,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3460,7 +3043,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3471,19 +3058,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3515,7 +3091,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3525,23 +3101,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3567,7 +3141,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3575,16 +3149,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3604,10 +3177,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3632,14 +3201,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3651,7 +3212,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3666,18 +3227,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3689,7 +3242,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3701,19 +3254,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3745,7 +3287,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3755,17 +3297,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3777,7 +3317,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3796,21 +3335,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3822,13 +3360,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3847,21 +3381,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3871,11 +3404,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3889,14 +3418,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3923,7 +3444,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3932,17 +3453,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -3950,7 +3467,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -3973,16 +3489,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3993,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4010,11 +3525,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4026,15 +3541,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -4057,15 +3567,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4076,14 +3585,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4093,11 +3602,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4109,8 +3618,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4123,7 +3630,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4142,21 +3648,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4168,20 +3673,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -4207,7 +3708,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -4216,16 +3717,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4245,12 +3745,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -4278,14 +3774,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4294,12 +3782,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4317,18 +3805,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4337,12 +3817,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -4358,20 +3838,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4405,7 +3874,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4415,17 +3884,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -4441,7 +3908,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -4460,22 +3926,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4487,15 +3952,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -4512,22 +3973,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4537,14 +3997,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -4557,14 +4013,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4592,7 +4040,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4601,17 +4049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -4625,7 +4069,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4649,17 +4092,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4670,14 +4112,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -4687,11 +4129,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4703,10 +4145,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -4749,7 +4187,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4768,23 +4205,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4796,13 +4232,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4818,7 +4250,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskRestorePoint] @@ -4841,18 +4272,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4863,14 +4293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4880,11 +4310,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4896,17 +4326,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4932,7 +4358,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4942,16 +4368,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4965,13 +4390,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5003,14 +4424,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -5024,7 +4437,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5043,18 +4456,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -5068,7 +4473,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5084,19 +4489,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] @@ -5130,7 +4524,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5142,17 +4536,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5176,23 +4568,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5202,11 +4593,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5229,14 +4616,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_compute_management_client_enums.py index 7bed3ea42986..41a390cafb28 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_compute_management_client_enums.py @@ -27,31 +27,28 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + disk and upload using write token in both disk and VM guest state""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -70,10 +67,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -83,17 +80,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -126,16 +123,16 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -143,13 +140,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -205,12 +202,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_models_py3.py index a10badd679db..aa7a05d18af6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/models/_models_py3.py @@ -136,11 +136,11 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", and "Import". + "ImportSecure", and "UploadPreparedSecure". :vartype create_option: str or ~azure.mgmt.compute.v2021_08_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -207,7 +207,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", and "Import". + "CopyStart", "ImportSecure", and "UploadPreparedSecure". :paramtype create_option: str or ~azure.mgmt.compute.v2021_08_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -253,7 +253,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -302,7 +302,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -633,7 +633,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -710,7 +710,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] @@ -766,7 +766,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -878,7 +878,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -980,7 +980,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2021_08_01.models.Disk] @@ -1198,7 +1198,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2021_08_01.models.DiskRestorePoint] @@ -1606,7 +1606,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1730,7 +1730,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2021_08_01.models.AccessLevel @@ -1778,7 +1778,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1844,7 +1844,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -1883,7 +1883,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_08_01.models.SourceVault @@ -1916,7 +1916,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_08_01.models.SourceVault @@ -2208,7 +2208,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2258,7 +2258,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2319,7 +2319,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2559,7 +2559,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2021_08_01.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/operations/_operations.py index 1f0b554459e4..fb8bc561ee3b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_08_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -697,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -728,7 +728,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -858,7 +858,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -943,7 +943,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -984,7 +984,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1022,7 +1022,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1060,7 +1060,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1131,7 +1131,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1169,7 +1169,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1212,7 +1212,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_08_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_08_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_08_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_08_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -2409,7 +2257,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2282,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2290,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2318,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -2498,14 +2341,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2516,7 +2351,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2530,18 +2365,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2549,7 +2376,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -2560,19 +2387,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2603,7 +2419,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2613,20 +2429,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2651,7 +2469,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2477,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2688,10 +2505,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -2716,14 +2529,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2734,7 +2539,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2749,18 +2554,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2768,7 +2565,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2779,19 +2580,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_08_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2822,7 +2612,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2832,17 +2622,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2854,7 +2642,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2873,21 +2660,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2899,13 +2685,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2924,21 +2706,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2948,11 +2729,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2964,14 +2741,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2998,7 +2767,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3007,17 +2776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -3025,7 +2790,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3047,16 +2811,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3067,14 +2830,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -3084,11 +2847,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3100,15 +2863,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3130,15 +2888,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3149,14 +2906,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -3166,11 +2923,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3182,13 +2939,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3214,7 +2969,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -3222,16 +2977,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3245,13 +2999,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -3277,14 +3027,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3295,7 +3037,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3310,18 +3052,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3332,7 +3066,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -3344,19 +3078,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3387,7 +3110,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3399,17 +3122,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -3428,21 +3149,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3452,11 +3172,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3468,14 +3184,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3502,7 +3210,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -3513,17 +3221,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -3550,7 +3254,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3576,7 +3280,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3584,16 +3288,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3613,10 +3316,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3641,14 +3340,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3660,7 +3351,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3675,18 +3366,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3698,7 +3381,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3710,19 +3393,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3754,7 +3427,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3764,23 +3437,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3806,7 +3477,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3814,16 +3485,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3843,10 +3513,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -3871,14 +3537,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3890,7 +3548,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3905,18 +3563,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3928,7 +3578,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3940,19 +3590,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] @@ -3984,7 +3625,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3994,17 +3635,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4016,7 +3655,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4035,21 +3673,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4061,13 +3698,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4086,21 +3719,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4110,11 +3742,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4126,14 +3754,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4160,7 +3780,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4169,17 +3789,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4187,7 +3803,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4209,16 +3824,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4229,14 +3843,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4246,11 +3860,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4262,15 +3876,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4292,15 +3901,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4311,14 +3919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4328,11 +3936,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4344,8 +3952,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4358,7 +3964,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4380,17 +3985,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4401,14 +4005,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4418,11 +4022,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4434,10 +4038,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -4460,7 +4060,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -4485,7 +4089,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4493,16 +4097,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4522,10 +4125,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -4550,14 +4149,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4568,7 +4159,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4583,18 +4174,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4602,7 +4185,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -4613,19 +4200,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4656,7 +4232,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4666,23 +4242,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -4708,7 +4282,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4716,16 +4290,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4745,10 +4318,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -4773,14 +4342,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4791,7 +4352,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4806,18 +4367,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4828,7 +4381,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -4840,19 +4393,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_08_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4883,7 +4425,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4893,17 +4435,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -4915,7 +4455,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -4934,21 +4473,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4498,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4985,21 +4519,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5009,11 +4542,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5025,14 +4554,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5059,7 +4580,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5068,17 +4589,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -5086,7 +4603,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5108,16 +4624,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5128,14 +4643,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -5145,11 +4660,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5161,15 +4676,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5191,15 +4701,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5210,14 +4719,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -5227,11 +4736,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5243,8 +4752,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -5257,7 +4764,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -5276,21 +4782,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5302,20 +4807,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -5341,7 +4842,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -5350,16 +4851,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5379,12 +4879,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -5412,14 +4908,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5428,12 +4916,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5451,18 +4939,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5471,12 +4951,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -5492,20 +4972,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5539,7 +5008,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5549,17 +5018,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -5575,7 +5042,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -5594,22 +5060,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5621,15 +5086,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -5646,22 +5107,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5671,14 +5131,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -5691,14 +5147,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5726,7 +5174,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5735,17 +5183,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -5759,7 +5203,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5783,17 +5226,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5804,14 +5246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -5821,11 +5263,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5837,10 +5279,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -5883,7 +5321,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_08_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -5902,23 +5339,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5930,13 +5366,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -5952,7 +5384,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_08_01.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -5974,18 +5405,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5996,14 +5426,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -6013,11 +5443,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6029,17 +5459,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6065,7 +5491,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -6075,16 +5501,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6098,13 +5523,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6136,14 +5557,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6156,7 +5569,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6175,18 +5588,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6199,7 +5604,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -6215,19 +5620,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_08_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_08_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6260,7 +5654,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6272,17 +5666,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -6306,23 +5698,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6332,11 +5723,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -6359,14 +5746,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_compute_management_client.py index 62d18b86aee8..c4ac686b10cd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -72,7 +74,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -97,7 +117,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-10-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -117,7 +137,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_configuration.py index edaf25514300..064b3cb4561d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-10-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_metadata.json index dcf50254a425..3d8ef7d03c18 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "galleries": "GalleriesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_compute_management_client.py index 3ab672e8edfd..65ad8b2a17c6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -72,7 +74,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -97,7 +117,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-10-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -117,7 +139,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_configuration.py index abb7a7ee6af2..37ef3ec71228 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-10-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/operations/_operations.py index b51011208667..44c052eeb87c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -86,7 +86,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -111,7 +111,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -119,16 +119,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -151,10 +150,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -179,14 +174,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -198,7 +185,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -213,18 +200,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -233,7 +212,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -244,19 +223,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_10_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_10_01.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -288,7 +256,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -298,20 +266,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -336,7 +306,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -344,16 +314,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -365,13 +334,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -396,14 +361,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -415,7 +372,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -429,18 +386,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -449,7 +398,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -460,19 +413,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_10_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_10_01.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -504,7 +446,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -514,17 +456,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -548,7 +488,6 @@ async def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -567,23 +506,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -595,13 +533,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -620,21 +554,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -644,11 +577,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -658,14 +587,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -692,7 +613,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -701,17 +622,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -719,7 +636,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -741,16 +657,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -761,14 +676,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -778,11 +693,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -794,15 +709,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -824,15 +734,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -843,14 +752,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -860,11 +769,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -876,8 +785,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -904,7 +811,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -930,7 +837,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -939,16 +846,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -971,10 +877,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -1003,14 +905,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1023,7 +917,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1041,18 +935,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1065,7 +951,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -1080,19 +966,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1125,7 +1000,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1135,24 +1010,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1178,7 +1051,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1187,16 +1060,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1208,13 +1080,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1243,14 +1111,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1263,7 +1123,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1280,18 +1140,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1304,7 +1156,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -1319,19 +1171,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1364,7 +1205,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1374,17 +1215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1399,7 +1238,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1418,22 +1256,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1445,13 +1282,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -1470,22 +1303,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1495,11 +1327,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1514,14 +1342,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1549,7 +1369,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1558,17 +1378,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -1581,7 +1397,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1604,17 +1419,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1625,14 +1439,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -1642,11 +1456,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1658,10 +1472,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -1689,7 +1499,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1715,7 +1525,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1725,16 +1535,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1757,10 +1566,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -1794,14 +1599,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1816,7 +1613,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1838,18 +1635,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1864,7 +1653,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -1883,19 +1672,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1930,7 +1709,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1940,17 +1719,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -1958,7 +1735,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1984,7 +1761,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1994,16 +1771,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2015,13 +1791,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2056,14 +1828,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2078,7 +1842,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2100,18 +1864,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2126,7 +1882,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -2145,20 +1901,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2193,7 +1938,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2203,17 +1948,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2242,7 +1985,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -2261,7 +2003,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2269,16 +2011,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2290,13 +2031,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -2320,23 +2057,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2346,11 +2082,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2374,14 +2106,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2410,7 +2134,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2419,17 +2143,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -2445,7 +2165,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion] @@ -2468,18 +2187,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2490,14 +2208,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -2507,11 +2225,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2523,10 +2241,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -2553,7 +2267,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2579,7 +2293,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2588,16 +2302,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2620,10 +2333,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -2652,14 +2361,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2673,7 +2374,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2691,18 +2392,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2716,7 +2409,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -2731,19 +2424,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2777,7 +2460,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2787,24 +2470,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2830,7 +2511,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2839,16 +2520,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2860,13 +2540,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2896,14 +2572,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2917,7 +2585,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2935,18 +2603,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2960,7 +2620,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -2975,20 +2635,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3022,7 +2671,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3032,17 +2681,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3058,7 +2705,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3077,22 +2723,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3104,13 +2749,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -3129,22 +2770,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3154,11 +2794,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3174,14 +2810,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3209,7 +2837,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3218,17 +2846,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -3241,7 +2865,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryApplication] @@ -3264,17 +2887,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3285,14 +2907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -3302,11 +2924,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3318,10 +2940,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -3349,7 +2967,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3375,7 +2993,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3385,16 +3003,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3417,10 +3034,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -3455,14 +3068,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3477,7 +3082,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3499,18 +3104,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3525,7 +3122,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -3544,21 +3141,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3593,7 +3179,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3603,17 +3189,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -3621,7 +3205,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3647,7 +3231,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3657,16 +3241,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3678,13 +3261,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3720,14 +3299,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3742,7 +3313,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3764,18 +3335,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3790,7 +3353,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -3809,20 +3372,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3857,7 +3410,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3867,17 +3420,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3906,7 +3457,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3925,7 +3475,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3933,16 +3483,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3954,13 +3503,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3984,23 +3529,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4010,11 +3554,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4038,14 +3578,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4074,7 +3606,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4083,17 +3615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -4109,7 +3637,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4133,18 +3660,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4155,14 +3681,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -4172,11 +3698,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4188,10 +3714,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -4217,7 +3739,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -4243,7 +3765,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4251,16 +3773,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4280,10 +3801,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -4305,14 +3822,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4325,7 +3834,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4337,18 +3846,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4361,7 +3862,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -4371,19 +3872,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2021_10_01.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2021_10_01.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4416,7 +3906,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4426,14 +3916,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/models/_models_py3.py index 4ad3224724cb..515e128be7ee 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/models/_models_py3.py @@ -191,7 +191,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -284,7 +284,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -333,7 +333,7 @@ class Gallery(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -427,7 +427,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -528,7 +528,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2021_10_01.models.GalleryApplication] @@ -701,7 +701,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -773,7 +773,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion] @@ -906,12 +906,12 @@ def __init__( class GalleryApplicationVersionPublishingProfile( GalleryArtifactPublishingProfileBase -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1098,7 +1098,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2021_10_01.models.ManagedArtifact @@ -1202,7 +1202,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -1313,7 +1313,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1514,7 +1514,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -1554,7 +1554,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1758,7 +1758,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1836,7 +1836,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion] @@ -1902,71 +1902,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) list[~azure.mgmt.compute.v2021_10_01.models.GalleryTargetExtendedLocation] """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - "target_extended_locations": {"key": "targetExtendedLocations", "type": "[GalleryTargetExtendedLocation]"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - target_extended_locations: Optional[List["_models.GalleryTargetExtendedLocation"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2021_10_01.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", - "Premium_LRS", and "StandardSSD_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2021_10_01.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2021_10_01.models.ReplicationMode - :keyword target_extended_locations: The target extended locations where the Image Version is - going to be replicated to. This property is updatable. - :paramtype target_extended_locations: - list[~azure.mgmt.compute.v2021_10_01.models.GalleryTargetExtendedLocation] - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - target_extended_locations=target_extended_locations, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -2081,7 +2016,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2021_10_01.models.Gallery] @@ -2126,32 +2061,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2021_10_01.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2021_10_01.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2021_10_01.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryTargetExtendedLocation(_serialization.Model): """GalleryTargetExtendedLocation. @@ -2369,7 +2278,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -2754,7 +2663,7 @@ def __init__(self, *, summary: Optional[List["_models.RegionalSharingStatus"]] = class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. :code:`
    `:code:`
    ` Possible values are: :code:`
    `:code:`
    ` **Add** @@ -2822,7 +2731,7 @@ def __init__(self, *, is_soft_delete_enabled: Optional[bool] = None, **kwargs: A class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -2883,7 +2792,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -2930,7 +2839,7 @@ def __init__(self, *, install: str, remove: str, update: Optional[str] = None, * class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/operations/_operations.py index 67effa96694b..9872ca6a2a52 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_10_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -179,7 +179,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -233,7 +233,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -367,7 +367,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -398,7 +398,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -439,7 +439,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -522,7 +522,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -560,7 +560,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +592,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +627,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -694,7 +694,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -726,7 +726,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -757,7 +757,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -800,7 +800,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -843,7 +843,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -887,7 +887,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -927,7 +927,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -959,7 +959,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1014,7 +1014,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1039,7 +1039,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1047,16 +1047,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1079,10 +1078,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -1107,14 +1102,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1125,7 +1112,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1140,18 +1127,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1159,7 +1138,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -1170,19 +1149,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_10_01.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_10_01.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1213,7 +1181,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1223,20 +1191,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1261,7 +1231,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1269,16 +1239,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1290,13 +1259,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -1321,14 +1286,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1339,7 +1296,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1353,18 +1310,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1372,7 +1321,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -1383,19 +1336,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2021_10_01.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2021_10_01.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1426,7 +1368,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1436,17 +1378,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -1470,7 +1410,6 @@ def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -1489,23 +1428,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1517,13 +1455,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -1542,21 +1476,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1566,11 +1499,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1580,14 +1509,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1614,7 +1535,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1623,17 +1544,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -1641,7 +1558,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1663,16 +1579,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1683,14 +1598,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -1700,11 +1615,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1716,15 +1631,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_10_01.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1746,15 +1656,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1765,14 +1674,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -1782,11 +1691,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1798,8 +1707,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -1826,7 +1733,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1852,7 +1759,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1861,16 +1768,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1893,10 +1799,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -1925,14 +1827,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1945,7 +1839,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1963,18 +1857,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -1987,7 +1873,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -2002,19 +1888,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -2047,7 +1922,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2057,24 +1932,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2100,7 +1973,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2109,16 +1982,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2130,13 +2002,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2165,14 +2033,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -2185,7 +2045,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2202,18 +2062,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -2226,7 +2078,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -2241,19 +2093,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] @@ -2286,7 +2127,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2296,17 +2137,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -2321,7 +2160,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2340,22 +2178,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2367,13 +2204,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -2392,22 +2225,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2417,11 +2249,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -2436,14 +2264,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2471,7 +2291,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2480,17 +2300,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -2503,7 +2319,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2525,17 +2340,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2546,14 +2360,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -2563,11 +2377,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2579,10 +2393,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -2610,7 +2420,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -2636,7 +2446,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2646,16 +2456,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2678,10 +2487,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -2715,14 +2520,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2737,7 +2534,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2759,18 +2556,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2785,7 +2574,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -2804,19 +2593,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2851,7 +2630,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2861,17 +2640,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -2879,7 +2656,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -2905,7 +2682,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2915,16 +2692,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2936,13 +2712,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2977,14 +2749,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2999,7 +2763,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3021,18 +2785,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3047,7 +2803,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -3066,20 +2822,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3114,7 +2859,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3124,17 +2869,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3163,7 +2906,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3182,7 +2924,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3190,16 +2932,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3211,13 +2952,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3241,23 +2978,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3267,11 +3003,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -3295,14 +3027,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3331,7 +3055,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3340,17 +3064,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -3366,7 +3086,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryImageVersion] @@ -3389,18 +3108,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3411,14 +3129,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -3428,11 +3146,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3444,10 +3162,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -3474,7 +3188,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3500,7 +3214,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3509,16 +3223,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3541,10 +3254,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -3573,14 +3282,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3594,7 +3295,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3612,18 +3313,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3637,7 +3330,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -3652,19 +3345,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3698,7 +3381,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3708,24 +3391,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3751,7 +3432,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3760,16 +3441,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3781,13 +3461,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3817,14 +3493,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3838,7 +3506,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3856,18 +3524,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3881,7 +3541,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -3896,20 +3556,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3943,7 +3592,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3953,17 +3602,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3979,7 +3626,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3998,22 +3644,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4025,13 +3670,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -4050,22 +3691,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4075,11 +3715,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4095,14 +3731,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4130,7 +3758,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4139,17 +3767,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -4162,7 +3786,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_10_01.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -4184,17 +3807,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4205,14 +3827,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -4222,11 +3844,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4238,10 +3860,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -4269,7 +3887,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4295,7 +3913,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4305,16 +3923,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4337,10 +3954,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -4375,14 +3988,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4397,7 +4002,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4419,18 +4024,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4445,7 +4042,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -4464,21 +4061,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4513,7 +4099,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4523,17 +4109,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -4541,7 +4125,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4567,7 +4151,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4577,16 +4161,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4598,13 +4181,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4640,14 +4219,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4662,7 +4233,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4684,18 +4255,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4710,7 +4273,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -4729,20 +4292,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4777,7 +4330,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4787,17 +4340,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4826,7 +4377,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_10_01.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_10_01.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4845,7 +4395,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4853,16 +4403,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4874,13 +4423,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -4904,23 +4449,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-10-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4930,11 +4474,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4958,14 +4498,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4994,7 +4526,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5003,17 +4535,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -5029,7 +4557,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5053,18 +4580,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5075,14 +4601,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -5092,11 +4618,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5108,10 +4634,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -5137,7 +4659,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -5163,7 +4685,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5171,16 +4693,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5200,10 +4721,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -5225,14 +4742,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.SharingUpdate] @@ -5244,7 +4753,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5256,18 +4765,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.SharingUpdate] @@ -5279,7 +4780,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -5289,19 +4790,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2021_10_01.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2021_10_01.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_10_01.models.SharingUpdate] @@ -5333,7 +4823,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5343,14 +4833,12 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_compute_management_client.py index 77968ecec3b5..41ec51081a70 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_configuration.py index 4ab3d0eeb3cc..f10cad49d7d2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_metadata.json index 0ec054b4cae1..d4c630651bf8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_compute_management_client.py index 2f47e3ebfe69..6a5854434cf5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_configuration.py index db09880c539d..b868444e9aa5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/operations/_operations.py index c041a7cf32b6..bf26083975c5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -233,7 +233,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -257,14 +256,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -275,14 +273,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -292,11 +290,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -308,8 +306,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -352,7 +348,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -363,7 +358,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -375,11 +370,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -390,7 +384,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -400,12 +394,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -433,7 +423,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -441,16 +431,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -462,13 +451,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -491,7 +476,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -502,7 +486,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -514,11 +498,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -529,7 +512,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -539,12 +522,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -572,7 +551,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -580,16 +559,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -601,13 +579,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -619,7 +593,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -638,21 +611,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -662,11 +634,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -676,7 +644,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -695,21 +662,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -721,13 +687,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -738,7 +700,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet] @@ -761,16 +722,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -781,14 +741,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -798,11 +758,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -814,17 +774,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet] @@ -847,16 +802,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -867,14 +821,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -884,11 +838,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -900,10 +854,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -915,7 +865,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] @@ -938,17 +887,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -959,14 +907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -976,11 +924,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -992,10 +940,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -1039,7 +983,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1050,7 +993,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1063,11 +1006,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1078,7 +1020,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -1088,12 +1030,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1121,7 +1059,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1129,16 +1067,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1158,10 +1095,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -1184,7 +1117,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1195,7 +1127,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1208,11 +1140,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1223,7 +1154,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -1233,12 +1164,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1266,7 +1194,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -1274,16 +1202,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1295,13 +1222,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1313,7 +1236,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1332,21 +1254,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1356,11 +1277,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1380,7 +1297,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1399,22 +1315,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1426,19 +1341,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1462,15 +1372,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1481,14 +1390,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1498,11 +1407,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1514,10 +1423,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -1526,7 +1431,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -1550,16 +1454,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1570,14 +1473,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -1587,11 +1490,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1603,10 +1506,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -1651,7 +1550,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1662,7 +1560,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1676,11 +1574,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1691,7 +1588,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -1703,12 +1600,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1736,7 +1629,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1744,16 +1637,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1773,10 +1665,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -1798,7 +1686,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1809,7 +1696,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1821,11 +1708,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1836,7 +1722,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -1846,12 +1732,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -1879,7 +1761,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -1887,16 +1769,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1908,13 +1789,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -1926,7 +1803,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -1945,21 +1821,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1969,11 +1844,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -1995,7 +1866,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -2014,22 +1884,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2041,13 +1910,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -2058,7 +1923,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup] @@ -2081,16 +1945,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2101,14 +1964,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2118,11 +1981,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2134,16 +1997,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup] @@ -2166,15 +2024,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2185,14 +2042,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -2202,11 +2059,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2218,8 +2075,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -2246,7 +2101,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2272,7 +2127,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2281,16 +2136,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2310,10 +2164,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -2338,14 +2188,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2359,7 +2201,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2373,18 +2215,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2398,7 +2232,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -2410,19 +2244,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2456,7 +2279,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2466,24 +2289,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -2509,7 +2330,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -2518,16 +2339,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2539,13 +2359,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2571,14 +2387,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2592,7 +2400,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2606,18 +2414,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2631,7 +2431,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -2643,19 +2443,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -2689,7 +2478,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2699,17 +2488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -2728,22 +2515,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2753,11 +2539,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2771,14 +2553,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2806,7 +2580,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2815,17 +2589,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -2849,7 +2619,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -2868,23 +2637,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2896,13 +2664,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -2915,7 +2679,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -2938,17 +2701,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2959,14 +2721,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -2976,11 +2738,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2992,10 +2754,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -3013,22 +2771,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3038,11 +2795,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -3059,14 +2812,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3094,7 +2839,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3103,17 +2848,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -3141,7 +2882,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3165,15 +2905,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3184,14 +2923,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3201,11 +2940,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3217,8 +2956,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -3228,7 +2965,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -3252,16 +2988,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3272,14 +3007,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -3289,11 +3024,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3305,10 +3040,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -3330,7 +3061,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3341,7 +3071,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3353,11 +3083,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3368,7 +3097,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -3378,12 +3107,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3411,7 +3136,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3419,16 +3144,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3448,10 +3172,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -3473,7 +3193,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3484,7 +3203,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3496,11 +3215,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3511,7 +3229,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -3521,12 +3239,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3554,7 +3269,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -3562,16 +3277,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3583,13 +3297,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -3601,7 +3311,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -3620,21 +3329,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3644,11 +3352,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -3660,7 +3364,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -3679,21 +3382,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3705,13 +3407,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -3725,7 +3423,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3744,21 +3441,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3770,13 +3466,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -3813,7 +3505,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -3832,23 +3523,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3860,13 +3550,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -3878,7 +3564,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3897,21 +3582,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3923,13 +3607,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -3957,7 +3637,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -3976,7 +3655,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -3985,16 +3664,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4006,13 +3684,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -4040,7 +3714,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4066,7 +3740,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4075,16 +3749,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4104,10 +3777,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4134,14 +3803,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4155,7 +3816,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4171,18 +3832,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4196,7 +3849,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -4209,20 +3862,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4256,7 +3898,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4266,24 +3908,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -4309,7 +3949,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -4318,16 +3958,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4339,13 +3978,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -4374,14 +4009,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4395,7 +4022,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4411,18 +4038,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4436,7 +4055,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -4449,20 +4068,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -4496,7 +4104,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4506,17 +4114,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -4535,22 +4141,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4560,11 +4165,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4579,14 +4180,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4614,7 +4207,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4623,17 +4216,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4655,7 +4244,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -4674,23 +4262,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4702,13 +4289,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4722,7 +4305,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4741,22 +4323,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4768,13 +4349,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -4813,7 +4390,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4832,7 +4408,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4840,16 +4416,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4861,13 +4436,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -4899,7 +4470,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4918,7 +4488,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -4928,16 +4498,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4949,13 +4518,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -4967,7 +4532,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -4986,21 +4550,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5012,13 +4575,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -5026,7 +4585,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5045,20 +4603,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5070,13 +4627,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5090,7 +4643,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5109,22 +4661,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5136,13 +4687,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -5183,7 +4730,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5202,7 +4748,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5211,16 +4757,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5232,13 +4777,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -5275,7 +4816,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5294,7 +4834,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -5305,16 +4845,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5326,13 +4865,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -5347,7 +4882,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5366,22 +4900,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5393,13 +4926,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -5411,7 +4940,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5430,21 +4958,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5456,13 +4983,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -5479,7 +5002,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -5498,23 +5020,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5526,13 +5047,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -5562,7 +5079,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5584,16 +5100,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5604,14 +5119,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -5621,11 +5136,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5637,8 +5152,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -5667,7 +5180,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -5690,16 +5202,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5710,14 +5221,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -5727,11 +5238,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5743,15 +5254,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -5777,7 +5284,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -5785,16 +5292,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5808,13 +5314,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -5838,14 +5340,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5858,7 +5352,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5871,18 +5365,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5895,7 +5381,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -5906,19 +5392,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -5951,7 +5427,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5963,20 +5439,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6001,7 +5479,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6009,16 +5487,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6038,10 +5515,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -6064,14 +5537,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6084,7 +5549,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6097,18 +5562,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6118,7 +5575,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -6128,19 +5589,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6173,7 +5623,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6183,20 +5633,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -6221,7 +5673,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -6229,16 +5681,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6250,13 +5701,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6279,14 +5726,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6299,7 +5738,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6311,18 +5750,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6332,7 +5763,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -6341,19 +5776,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -6386,7 +5810,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6396,17 +5820,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6425,22 +5847,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6450,11 +5871,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6469,14 +5886,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6504,7 +5913,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6513,17 +5922,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6546,7 +5951,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -6565,22 +5969,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6592,13 +5995,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -6610,7 +6009,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -6629,21 +6027,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6655,13 +6052,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -6680,21 +6073,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6704,11 +6096,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -6721,14 +6109,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6755,7 +6135,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6764,17 +6144,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -6793,22 +6169,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6818,11 +6193,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6838,14 +6209,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6873,7 +6236,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6882,17 +6245,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -6909,7 +6268,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6928,21 +6286,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6952,11 +6309,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -6972,7 +6325,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -6995,17 +6347,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7016,14 +6367,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7033,11 +6384,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7049,10 +6400,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -7068,7 +6415,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -7091,17 +6437,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7112,14 +6457,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -7129,11 +6474,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7145,8 +6490,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7158,7 +6501,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] @@ -7181,17 +6523,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7202,14 +6543,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -7219,11 +6560,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7235,10 +6576,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -7256,22 +6593,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7281,11 +6617,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7302,14 +6634,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7337,7 +6661,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7346,17 +6670,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7375,21 +6695,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7399,11 +6718,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7413,14 +6728,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7447,7 +6754,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7456,17 +6763,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7485,21 +6788,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7509,11 +6811,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7523,14 +6821,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7557,7 +6847,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7566,17 +6856,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7595,21 +6881,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7619,11 +6904,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7633,14 +6914,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7667,7 +6940,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7676,17 +6949,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -7705,21 +6974,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7729,11 +6997,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -7743,14 +7007,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7777,7 +7033,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7786,23 +7042,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7831,7 +7083,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -7839,16 +7091,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7858,11 +7109,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -7886,14 +7133,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7904,7 +7143,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7917,18 +7156,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7939,7 +7170,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -7949,19 +7180,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7991,7 +7212,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8000,17 +7221,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -8032,7 +7249,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8051,22 +7267,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8078,13 +7293,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8103,21 +7314,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8127,11 +7337,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8143,14 +7349,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8177,7 +7375,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8186,17 +7384,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8208,7 +7402,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8227,21 +7420,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8251,11 +7443,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -8274,21 +7462,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8302,13 +7489,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -8320,14 +7503,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -8357,7 +7532,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8369,23 +7544,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -8411,7 +7584,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8419,16 +7592,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8442,13 +7614,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -8473,14 +7641,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8493,7 +7653,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8506,18 +7666,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8530,7 +7682,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -8540,20 +7692,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -8586,7 +7727,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8598,20 +7739,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -8636,7 +7779,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -8644,16 +7787,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8667,13 +7809,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8696,14 +7834,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8716,7 +7846,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8728,18 +7858,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8749,7 +7871,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -8758,19 +7884,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8803,7 +7918,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8815,17 +7930,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -8855,7 +7968,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8879,16 +7991,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8899,14 +8010,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -8916,11 +8027,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8932,15 +8043,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -8966,7 +8073,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8974,16 +8081,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9003,10 +8109,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -9028,14 +8130,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9048,7 +8142,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9060,18 +8154,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9084,7 +8170,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -9093,20 +8179,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9139,7 +8214,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9149,23 +8224,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -9191,7 +8264,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9199,16 +8272,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9220,13 +8292,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9249,14 +8317,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9269,7 +8329,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9281,18 +8341,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9305,7 +8357,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -9314,20 +8366,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -9360,7 +8402,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9370,17 +8412,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9399,22 +8439,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9424,11 +8463,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9443,14 +8478,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9478,7 +8505,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9487,17 +8514,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9518,7 +8541,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -9537,22 +8559,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9564,19 +8585,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9605,7 +8622,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9613,16 +8630,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9632,11 +8648,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -9663,14 +8675,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9681,7 +8685,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9696,18 +8700,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9718,7 +8714,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -9730,20 +8726,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9773,7 +8758,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9782,23 +8767,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9826,7 +8807,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9835,16 +8816,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9854,11 +8834,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -9887,14 +8863,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9905,7 +8873,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -9919,21 +8887,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9944,7 +8904,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9956,23 +8916,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10003,7 +8952,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10012,17 +8961,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -10034,7 +8979,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10053,21 +8997,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10079,13 +9022,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -10093,7 +9032,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10117,16 +9055,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10137,14 +9074,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -10154,11 +9091,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10170,17 +9107,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -10204,15 +9136,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10223,14 +9154,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -10240,11 +9171,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10256,8 +9187,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10269,7 +9198,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -10293,17 +9221,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10314,14 +9241,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -10331,11 +9258,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10347,10 +9274,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10361,7 +9284,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -10385,17 +9307,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10406,14 +9327,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -10423,11 +9344,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10439,15 +9360,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10478,7 +9395,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10487,16 +9404,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10506,11 +9422,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -10542,14 +9454,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10560,7 +9464,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -10576,7 +9480,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -10584,14 +9488,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10602,7 +9498,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -10616,24 +9512,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10664,7 +9549,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10673,23 +9558,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10718,7 +9599,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10726,16 +9607,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10745,11 +9625,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -10774,14 +9650,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10792,7 +9660,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10805,18 +9673,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10827,7 +9687,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -10837,20 +9697,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10880,7 +9729,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10889,23 +9738,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10934,7 +9779,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10942,16 +9787,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10961,11 +9805,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -10990,14 +9830,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11008,7 +9840,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11021,18 +9853,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11043,7 +9867,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -11053,20 +9877,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11096,7 +9909,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11105,23 +9918,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11150,7 +9959,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11158,16 +9967,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11177,11 +9985,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -11207,14 +10011,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11225,7 +10021,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11239,18 +10035,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11261,7 +10049,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -11272,20 +10060,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11315,7 +10092,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11324,23 +10101,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11369,7 +10142,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11377,16 +10150,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11396,11 +10168,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -11428,14 +10196,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11446,7 +10206,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11462,18 +10222,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11484,7 +10236,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -11497,20 +10249,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11540,7 +10281,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11549,23 +10290,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -11591,7 +10328,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11599,16 +10336,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11618,11 +10354,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -11647,14 +10379,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11665,7 +10389,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11678,18 +10402,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11700,7 +10416,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -11710,20 +10426,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11753,7 +10458,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11762,23 +10467,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11807,7 +10508,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -11815,16 +10516,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11834,11 +10534,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11864,14 +10560,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11882,7 +10570,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11896,18 +10584,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11918,7 +10598,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -11930,20 +10610,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11973,7 +10642,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11982,23 +10651,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12027,7 +10692,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12035,16 +10700,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12054,11 +10718,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -12084,14 +10744,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12102,7 +10754,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12116,18 +10768,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12138,7 +10782,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -12149,20 +10793,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12192,7 +10825,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12201,20 +10834,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -12240,7 +10869,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -12259,7 +10887,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12267,16 +10895,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12288,13 +10915,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -12319,7 +10942,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12330,7 +10952,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12343,11 +10965,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12358,7 +10979,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -12369,13 +10990,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12403,7 +11021,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12411,16 +11029,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12430,17 +11047,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -12466,7 +11079,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -12474,16 +11087,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12493,11 +11105,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -12521,14 +11129,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12539,7 +11139,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12552,18 +11152,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12574,7 +11166,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -12585,19 +11177,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12627,7 +11209,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12636,17 +11218,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -12676,7 +11254,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] @@ -12699,16 +11276,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12719,14 +11295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12736,11 +11312,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12752,8 +11328,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -12776,7 +11350,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -12801,7 +11375,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -12809,16 +11383,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12838,10 +11411,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -12863,14 +11432,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12881,7 +11442,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12893,18 +11454,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12912,7 +11465,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -12921,19 +11474,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -12964,7 +11506,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12974,20 +11516,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -13012,7 +11556,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -13020,16 +11564,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13049,10 +11592,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -13074,14 +11613,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13092,7 +11623,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13104,18 +11635,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13123,7 +11646,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -13132,19 +11659,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13175,7 +11691,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13185,17 +11701,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -13214,21 +11728,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13238,11 +11751,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -13252,14 +11761,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13286,7 +11787,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13295,17 +11796,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13319,7 +11816,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -13338,22 +11834,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13365,13 +11860,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -13379,7 +11870,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13401,16 +11891,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13421,14 +11910,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13438,11 +11927,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13454,16 +11943,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -13485,15 +11969,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13504,14 +11987,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -13521,11 +12004,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13537,8 +12020,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -13584,7 +12065,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13595,7 +12075,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13610,11 +12090,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13625,7 +12104,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -13637,12 +12116,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13670,7 +12145,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13678,16 +12153,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13707,10 +12181,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -13733,7 +12203,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13744,7 +12213,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13757,11 +12226,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13772,7 +12240,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -13782,12 +12250,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -13815,7 +12280,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -13823,16 +12288,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13844,13 +12308,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -13869,21 +12329,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13893,11 +12352,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -13910,14 +12365,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13944,7 +12391,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -13953,17 +12400,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -13985,7 +12428,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -14004,22 +12446,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14031,13 +12472,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -14045,7 +12482,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -14069,16 +12505,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14089,14 +12524,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -14106,11 +12541,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14122,17 +12557,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -14156,15 +12586,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14175,14 +12604,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -14192,11 +12621,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14208,8 +12637,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -14236,7 +12663,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -14262,7 +12689,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -14271,16 +12698,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14292,13 +12718,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -14325,14 +12747,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -14345,7 +12759,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14360,18 +12774,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -14384,7 +12790,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -14397,19 +12803,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -14442,7 +12837,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14452,17 +12847,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -14481,22 +12874,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14506,11 +12898,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -14524,14 +12912,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14559,7 +12939,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -14568,17 +12948,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -14601,7 +12977,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -14620,23 +12995,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14648,13 +13022,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -14700,7 +13070,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14711,7 +13080,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14725,11 +13094,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14740,7 +13108,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -14752,12 +13120,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14785,7 +13149,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14793,16 +13157,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14822,10 +13185,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -14849,7 +13208,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14860,7 +13218,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14874,11 +13232,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14889,7 +13246,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -14900,12 +13257,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14933,7 +13287,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -14941,16 +13295,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14962,13 +13315,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14983,7 +13332,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15002,21 +13350,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15026,11 +13373,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15054,7 +13397,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15073,22 +13415,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15100,13 +13441,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15127,7 +13464,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15151,17 +13487,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15172,14 +13507,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15189,11 +13524,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15205,10 +13540,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -15222,7 +13553,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -15246,16 +13576,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15266,14 +13595,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -15283,11 +13612,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15299,10 +13628,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -15329,7 +13654,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -15355,7 +13680,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15364,16 +13689,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15393,10 +13717,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -15423,14 +13743,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15444,7 +13756,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15460,18 +13772,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15485,7 +13789,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -15499,19 +13803,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15545,7 +13838,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15555,24 +13848,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -15598,7 +13889,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -15607,16 +13898,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15630,13 +13920,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15662,14 +13948,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15683,7 +13961,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15697,18 +13975,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15722,7 +13992,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -15734,19 +14004,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -15780,7 +14039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15790,17 +14049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -15823,22 +14080,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15848,11 +14104,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15872,14 +14124,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15907,7 +14151,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15916,17 +14160,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15951,7 +14191,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -15970,23 +14209,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15998,13 +14236,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -16017,7 +14251,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_11_01.models.CapacityReservation] @@ -16040,17 +14273,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16061,14 +14293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -16078,11 +14310,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16094,12 +14326,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16124,7 +14352,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16150,7 +14378,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16159,16 +14387,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16188,10 +14415,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -16219,14 +14442,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16240,7 +14455,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16256,18 +14471,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16281,7 +14488,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -16294,20 +14501,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16341,7 +14537,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16351,24 +14547,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -16394,7 +14588,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -16403,16 +14597,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16432,10 +14625,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -16463,14 +14652,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16484,7 +14665,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16500,18 +14681,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16525,7 +14698,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -16538,20 +14711,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16585,7 +14748,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16595,17 +14758,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -16624,22 +14785,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16649,11 +14809,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16668,14 +14824,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16703,7 +14851,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16712,17 +14860,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16744,7 +14888,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -16763,23 +14906,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16791,13 +14933,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -16809,7 +14947,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -16833,17 +14970,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16854,14 +14990,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -16871,11 +15007,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16887,12 +15023,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -16929,21 +15061,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16953,11 +15084,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -16969,14 +15096,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17003,7 +15122,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17012,17 +15131,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17041,21 +15156,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17065,11 +15179,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -17083,14 +15193,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17117,7 +15219,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17126,17 +15228,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -17155,21 +15253,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17179,11 +15276,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -17197,14 +15290,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17231,7 +15316,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17240,17 +15325,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -17262,7 +15343,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -17281,21 +15361,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17307,16 +15386,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -17342,7 +15417,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17368,7 +15443,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17378,16 +15453,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17407,10 +15481,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -17440,14 +15510,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17462,7 +15524,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17479,18 +15541,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17505,7 +15559,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -17519,20 +15573,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17567,7 +15610,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17577,17 +15620,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -17595,7 +15636,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -17621,7 +15662,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -17631,16 +15672,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17652,13 +15692,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -17689,14 +15725,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17711,7 +15739,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17728,18 +15756,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17754,7 +15774,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -17768,20 +15788,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -17816,7 +15826,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17826,17 +15836,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -17855,23 +15863,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17881,11 +15888,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17901,14 +15904,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17937,7 +15932,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17946,17 +15941,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17981,7 +15972,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -18000,7 +15990,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18008,16 +15998,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18029,13 +16018,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -18057,7 +16042,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18076,23 +16060,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18104,13 +16087,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -18138,7 +16117,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -18167,7 +16148,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18176,16 +16157,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18195,11 +16175,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -18227,14 +16203,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18246,7 +16214,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18261,18 +16229,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18284,7 +16244,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -18296,20 +16258,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18340,7 +16292,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18349,17 +16301,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18378,22 +16326,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18403,11 +16350,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -18422,14 +16365,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18457,7 +16392,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18466,17 +16401,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -18495,22 +16426,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18520,11 +16450,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -18540,14 +16466,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18575,7 +16493,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18584,24 +16502,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -18627,7 +16541,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -18636,16 +16550,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18665,10 +16578,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -18695,14 +16604,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18716,7 +16617,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18732,18 +16633,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18757,7 +16650,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -18770,19 +16663,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -18816,7 +16698,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18826,17 +16708,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -18861,23 +16741,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18887,11 +16766,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18914,14 +16789,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18950,7 +16817,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18959,17 +16826,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18993,7 +16856,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -19012,23 +16874,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19040,13 +16901,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -19060,7 +16917,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -19079,22 +16935,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19106,13 +16961,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19141,7 +16992,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -19165,7 +17015,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -19173,12 +17023,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19189,14 +17038,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -19206,11 +17055,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19222,10 +17071,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -19249,23 +17094,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19275,11 +17119,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -19305,14 +17145,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19341,7 +17173,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19350,17 +17182,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19379,22 +17207,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19404,11 +17231,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -19422,14 +17245,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19457,7 +17272,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19466,17 +17281,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19495,22 +17306,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19520,11 +17330,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -19538,14 +17344,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19573,7 +17371,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19582,17 +17380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19611,22 +17405,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19636,11 +17429,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -19655,14 +17444,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19690,7 +17471,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19699,17 +17480,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -19735,7 +17512,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -19754,23 +17530,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19782,13 +17557,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -19807,22 +17578,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19832,11 +17602,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -19850,14 +17616,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19885,7 +17643,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19894,17 +17652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -19918,7 +17672,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19937,22 +17690,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19962,18 +17714,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -19999,7 +17747,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -20008,16 +17756,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20031,13 +17778,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -20063,14 +17806,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -20084,7 +17819,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20098,18 +17833,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -20123,7 +17850,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -20135,19 +17862,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -20181,7 +17897,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20193,17 +17909,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -20227,7 +17941,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20252,23 +17966,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20282,13 +17995,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20310,14 +18019,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20327,7 +18028,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20336,18 +18037,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20357,7 +18050,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20365,19 +18058,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20409,7 +18092,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20422,20 +18105,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20460,23 +18141,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20490,13 +18170,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20517,14 +18193,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20534,7 +18202,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20542,18 +18210,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20563,7 +18223,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20571,19 +18231,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20615,7 +18264,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20628,17 +18277,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20667,7 +18314,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20691,16 +18337,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20711,14 +18356,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20728,11 +18373,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20744,10 +18389,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20756,7 +18397,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20775,21 +18415,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20801,20 +18440,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20840,7 +18475,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20849,16 +18484,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20878,10 +18512,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20908,14 +18538,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20929,7 +18551,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20945,18 +18567,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20970,7 +18584,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -20983,19 +18597,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21029,7 +18632,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21039,24 +18642,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21082,7 +18683,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21091,16 +18692,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21112,13 +18712,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21146,14 +18742,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21167,7 +18755,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21183,18 +18771,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21208,7 +18788,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21221,19 +18801,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21267,7 +18837,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21277,17 +18847,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21306,22 +18874,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21331,11 +18898,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21350,14 +18913,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21385,7 +18940,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21394,17 +18949,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21426,7 +18977,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21445,23 +18995,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21473,13 +19022,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21493,7 +19038,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21517,18 +19061,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21539,14 +19082,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21556,11 +19099,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21572,12 +19115,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21603,7 +19142,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21629,7 +19168,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21639,16 +19178,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21668,10 +19206,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21700,14 +19234,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21722,7 +19248,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21739,18 +19265,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21765,7 +19283,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21779,19 +19297,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21826,7 +19333,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21836,17 +19343,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21854,7 +19359,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21880,7 +19385,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21890,16 +19395,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21911,13 +19415,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21947,14 +19447,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21969,7 +19461,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21986,18 +19478,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22012,7 +19496,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22026,19 +19510,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22073,7 +19547,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22083,17 +19557,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22112,23 +19584,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22138,11 +19609,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22158,14 +19625,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22194,7 +19653,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22203,17 +19662,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22238,7 +19693,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22257,7 +19711,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22265,16 +19719,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22286,13 +19739,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22314,7 +19763,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22338,19 +19786,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22361,14 +19808,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22378,11 +19825,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22393,7 +19840,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_compute_management_client_enums.py index cd76651479e6..4e198654f248 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_compute_management_client_enums.py @@ -375,7 +375,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_models_py3.py index 893cb7d7212a..89d9a55f9941 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -265,7 +261,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -290,7 +286,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -305,7 +301,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -388,7 +384,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -445,7 +441,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -544,7 +540,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet] @@ -857,7 +853,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -975,7 +971,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1077,7 +1073,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup] @@ -1237,7 +1233,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.CapacityReservation] @@ -1472,7 +1468,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1683,7 +1679,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1866,7 +1862,7 @@ class DedicatedHostGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1982,7 +1978,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup] @@ -2186,7 +2182,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -2444,17 +2440,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2761,7 +2746,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2941,7 +2926,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_11_01.models.SubResource @@ -3049,7 +3034,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.Image] @@ -3083,7 +3068,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2021_11_01.models.SubResource @@ -3223,7 +3208,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3282,7 +3267,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3504,7 +3489,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3537,7 +3522,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -3823,7 +3808,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.Usage] @@ -3857,7 +3842,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4211,7 +4196,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. "AutomaticRepairs" :vartype service_name: str or ~azure.mgmt.compute.v2021_11_01.models.OrchestrationServiceNames @@ -4284,7 +4269,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -4446,7 +4431,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -4831,7 +4816,7 @@ class ProximityPlacementGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -4925,7 +4910,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup] @@ -4963,17 +4948,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5014,7 +4988,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2021_11_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2021_11_01.models.PublicIPAddressSkuTier """ @@ -5033,8 +5007,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2021_11_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2021_11_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5045,7 +5018,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5116,7 +5089,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5297,7 +5270,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6006,7 +5979,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6069,7 +6042,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6132,7 +6105,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6206,7 +6179,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6253,7 +6226,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6286,7 +6259,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.RunCommandDocumentBase] @@ -6322,7 +6295,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6667,7 +6640,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -6677,7 +6650,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -6706,7 +6679,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -6721,7 +6694,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6777,7 +6750,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource] @@ -6993,7 +6966,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7014,67 +6987,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7143,7 +7055,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7286,7 +7198,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -7504,7 +7416,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7996,7 +7908,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8086,7 +7998,7 @@ class VirtualMachineExtension(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8233,7 +8145,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8277,7 +8189,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8587,7 +8499,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_11_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_11_01.models.UserAssignedIdentitiesValue] """ @@ -8619,7 +8531,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2021_11_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_11_01.models.UserAssignedIdentitiesValue] """ @@ -8633,7 +8545,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8696,7 +8608,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8850,7 +8762,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9175,7 +9087,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -9208,10 +9120,10 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -9318,7 +9230,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9338,10 +9250,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -9507,10 +9419,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -9610,10 +9522,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -9668,7 +9580,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9915,7 +9827,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand] @@ -10064,7 +9976,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10315,7 +10227,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10569,10 +10481,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension] @@ -10827,7 +10739,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2021_11_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -10866,7 +10778,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue] """ @@ -10877,7 +10789,7 @@ def __init__( self.user_assigned_identities = user_assigned_identities -class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): +class VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue(_serialization.Model): # pylint: disable=name-too-long """VirtualMachineScaleSetIdentityUserAssignedIdentitiesValue. Variables are only populated by the server, and will be ignored when sending a request. @@ -10948,7 +10860,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -10975,7 +10887,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11130,10 +11042,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -11175,7 +11087,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet] @@ -11211,7 +11123,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetSku] @@ -11247,7 +11159,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet] @@ -11280,7 +11192,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -11331,10 +11243,10 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11437,7 +11349,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -11462,7 +11374,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_11_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -11493,7 +11405,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_11_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11512,7 +11424,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11784,10 +11696,10 @@ def __init__( self.allow_extension_operations = allow_extension_operations -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -11876,10 +11788,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -11906,7 +11820,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -11914,18 +11828,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -12234,7 +12136,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -12352,7 +12254,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -12453,12 +12355,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2021_11_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12491,7 +12393,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2021_11_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12626,7 +12528,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -12686,7 +12588,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -12834,7 +12736,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13221,7 +13123,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -13244,7 +13146,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -13422,10 +13324,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -13569,7 +13471,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM] @@ -13602,7 +13504,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -14455,14 +14357,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -14497,7 +14399,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -14511,7 +14413,7 @@ def __init__( self.configuration_reference = configuration_reference -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/operations/_operations.py index a6f5a332942d..ff3c86e0f3ad 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_11_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -58,7 +58,7 @@ def build_operations_list_request(**kwargs: Any) -> HttpRequest: return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -188,7 +188,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -244,7 +244,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -275,7 +275,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -311,7 +311,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -347,7 +347,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -380,7 +380,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -422,7 +422,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -448,7 +450,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -478,7 +480,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -512,7 +514,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -546,7 +548,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -615,7 +617,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -645,7 +647,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -669,7 +673,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -811,7 +815,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -874,7 +878,9 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -898,7 +904,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1058,7 +1064,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1089,7 +1095,7 @@ def build_ssh_public_keys_generate_key_pair_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1122,7 +1128,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1153,7 +1159,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -1199,7 +1205,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1234,7 +1240,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1269,7 +1275,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1301,7 +1307,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -1341,7 +1347,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1408,7 +1414,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -1456,7 +1462,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1487,7 +1493,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1516,7 +1522,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1548,7 +1554,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1590,7 +1596,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -1640,7 +1646,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1672,7 +1678,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1703,7 +1709,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1763,7 +1769,9 @@ def build_usage_list_request(location: str, subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -1825,7 +1833,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1969,7 +1977,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2000,7 +2008,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2031,7 +2039,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2064,7 +2072,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2157,7 +2165,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2379,7 +2387,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2419,7 +2427,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2450,7 +2458,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2481,7 +2489,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2512,7 +2520,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2546,7 +2554,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2580,7 +2588,7 @@ def build_virtual_machines_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2610,7 +2618,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2644,7 +2652,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2678,7 +2686,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2716,7 +2724,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2754,7 +2762,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2788,7 +2796,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -2829,7 +2837,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2860,7 +2868,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2890,7 +2898,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2916,7 +2926,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2947,7 +2957,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2978,7 +2988,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3019,7 +3029,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3053,7 +3063,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3087,7 +3097,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3121,7 +3131,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3155,7 +3165,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3189,7 +3199,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3223,7 +3233,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3257,7 +3267,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -3300,7 +3310,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3334,7 +3344,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3527,7 +3537,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3581,7 +3591,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3617,7 +3627,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3653,7 +3663,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3686,7 +3696,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -3726,7 +3736,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3756,7 +3766,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3903,7 +3915,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3939,7 +3951,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3975,7 +3987,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4008,7 +4020,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -4048,7 +4060,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -4084,7 +4096,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -4117,7 +4129,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4158,7 +4170,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4199,7 +4211,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -4279,7 +4291,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4312,7 +4324,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4347,7 +4359,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4382,7 +4394,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4414,7 +4426,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -4454,7 +4466,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4485,7 +4497,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4516,7 +4528,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4547,7 +4559,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4578,7 +4590,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4609,7 +4621,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4650,7 +4662,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4691,7 +4703,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4729,7 +4741,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4771,7 +4783,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4811,7 +4823,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4846,7 +4858,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4878,7 +4890,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4910,7 +4922,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4945,7 +4957,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -4985,7 +4997,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5025,7 +5037,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5057,7 +5069,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -5103,7 +5115,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5143,7 +5155,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5175,7 +5187,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5207,7 +5219,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5239,7 +5251,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5281,7 +5293,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5313,7 +5325,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5345,7 +5357,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5380,7 +5392,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5413,7 +5425,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5446,7 +5458,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5473,7 +5487,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5504,7 +5518,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5539,7 +5553,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5574,7 +5588,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5606,7 +5620,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5646,7 +5660,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5679,7 +5693,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5720,7 +5734,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5761,7 +5775,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5799,7 +5813,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5841,7 +5855,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5905,7 +5919,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5929,14 +5942,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5947,14 +5959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5964,11 +5976,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5980,8 +5992,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class AvailabilitySetsOperations: """ @@ -6024,7 +6034,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6035,7 +6044,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6047,11 +6056,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6062,7 +6070,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -6072,12 +6080,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6105,7 +6109,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -6113,16 +6117,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6134,13 +6137,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -6163,7 +6162,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6174,7 +6172,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6186,11 +6184,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6201,7 +6198,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -6211,12 +6208,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6244,7 +6237,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -6252,16 +6245,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6273,13 +6265,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6291,7 +6279,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -6310,21 +6297,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6334,11 +6320,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -6348,7 +6330,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -6367,21 +6348,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6393,13 +6373,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -6410,7 +6386,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -6432,16 +6407,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6452,14 +6426,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -6469,11 +6443,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6485,17 +6459,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -6517,16 +6486,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6537,14 +6505,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -6554,11 +6522,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6570,10 +6538,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -6585,7 +6549,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6607,17 +6570,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6628,14 +6590,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6645,11 +6607,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6661,10 +6623,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -6708,7 +6666,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6719,7 +6676,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6732,11 +6689,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6747,7 +6703,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -6757,12 +6713,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6790,7 +6742,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6798,16 +6750,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6827,10 +6778,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -6853,7 +6800,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6864,7 +6810,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6877,11 +6823,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6892,7 +6837,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -6902,12 +6847,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -6935,7 +6877,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -6943,16 +6885,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6964,13 +6905,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -6982,7 +6919,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7001,21 +6937,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7025,11 +6960,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -7049,7 +6980,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7068,22 +6998,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7095,19 +7024,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -7131,15 +7055,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7150,14 +7073,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -7167,11 +7090,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7183,10 +7106,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -7195,7 +7114,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -7219,16 +7137,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7239,14 +7156,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -7256,11 +7173,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7272,10 +7189,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -7320,7 +7233,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7331,7 +7243,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7345,11 +7257,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7360,7 +7271,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -7372,12 +7283,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7405,7 +7312,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -7413,16 +7320,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7442,10 +7348,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -7467,7 +7369,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7478,7 +7379,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7490,11 +7391,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7505,7 +7405,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -7515,12 +7415,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7548,7 +7444,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -7556,16 +7452,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7577,13 +7472,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -7595,7 +7486,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -7614,21 +7504,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7638,11 +7527,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -7664,7 +7549,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -7683,22 +7567,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7710,13 +7593,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -7725,7 +7604,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -7747,16 +7625,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7767,14 +7644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7784,11 +7661,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7800,16 +7677,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -7831,15 +7703,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7850,14 +7721,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -7867,11 +7738,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7883,8 +7754,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -7911,7 +7780,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -7937,7 +7806,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -7946,16 +7815,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7975,10 +7843,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -8003,14 +7867,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8023,7 +7879,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8037,18 +7893,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8061,7 +7909,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -8073,19 +7921,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8118,7 +7955,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8128,24 +7965,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -8171,7 +8006,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -8180,16 +8015,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8201,13 +8035,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -8233,14 +8063,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8253,7 +8075,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8267,18 +8089,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8291,7 +8105,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -8303,19 +8117,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] @@ -8348,7 +8151,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8358,17 +8161,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -8387,22 +8188,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8412,11 +8212,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -8430,14 +8226,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8465,7 +8253,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8474,17 +8262,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -8508,7 +8292,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -8527,23 +8310,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8555,13 +8337,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -8574,7 +8352,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -8596,17 +8373,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8617,14 +8393,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -8634,11 +8410,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8650,10 +8426,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -8671,22 +8443,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8696,11 +8467,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -8717,14 +8484,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8752,7 +8511,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8761,17 +8520,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -8799,7 +8554,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -8823,15 +8577,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8842,14 +8595,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -8859,11 +8612,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8875,8 +8628,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -8886,7 +8637,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -8910,16 +8660,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8930,14 +8679,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -8947,11 +8696,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8963,10 +8712,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -8988,7 +8733,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -8999,7 +8743,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9011,11 +8755,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9026,7 +8769,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -9036,12 +8779,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9069,7 +8808,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -9077,16 +8816,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9106,10 +8844,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -9131,7 +8865,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9142,7 +8875,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9154,11 +8887,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9169,7 +8901,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -9179,12 +8911,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9212,7 +8941,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -9220,16 +8949,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9241,13 +8969,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -9259,7 +8983,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9278,21 +9001,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9302,11 +9024,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -9316,7 +9034,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -9335,21 +9052,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9361,13 +9077,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -9381,7 +9093,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9400,21 +9111,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9426,13 +9136,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -9469,7 +9175,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -9488,23 +9193,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9516,13 +9220,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -9534,7 +9234,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9553,21 +9252,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9579,13 +9277,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -9613,7 +9307,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -9632,7 +9325,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -9641,16 +9334,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9662,13 +9354,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class VirtualMachineExtensionsOperations: @@ -9696,7 +9384,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9722,7 +9410,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9731,16 +9419,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9760,10 +9447,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -9790,14 +9473,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9811,7 +9486,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9827,18 +9502,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9852,7 +9519,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -9865,20 +9532,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9912,7 +9568,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9922,24 +9578,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9965,7 +9619,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9974,16 +9628,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9995,13 +9648,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -10030,14 +9679,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10051,7 +9692,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10067,18 +9708,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10092,7 +9725,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -10105,20 +9738,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -10152,7 +9774,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10162,17 +9784,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -10191,22 +9811,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10216,11 +9835,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10235,14 +9850,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10270,7 +9877,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10279,17 +9886,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10311,7 +9914,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10330,23 +9932,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10358,13 +9959,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10378,7 +9975,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -10397,22 +9993,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10424,13 +10019,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachineImagesOperations: @@ -10469,7 +10060,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -10488,7 +10078,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -10496,16 +10086,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10517,13 +10106,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10555,7 +10140,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10574,7 +10158,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -10584,16 +10168,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10605,13 +10188,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -10623,7 +10202,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10642,21 +10220,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10668,13 +10245,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -10682,7 +10255,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10701,20 +10273,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10726,13 +10297,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -10746,7 +10313,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10765,22 +10331,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10792,13 +10357,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -10839,7 +10400,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -10858,7 +10418,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -10867,16 +10427,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10888,13 +10447,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10931,7 +10486,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -10950,7 +10504,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -10961,16 +10515,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10982,13 +10535,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -11003,7 +10552,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11022,22 +10570,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11049,13 +10596,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -11067,7 +10610,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11086,21 +10628,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11112,13 +10653,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -11135,7 +10672,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -11154,23 +10690,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11182,13 +10717,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class UsageOperations: @@ -11218,7 +10749,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -11240,16 +10770,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11260,14 +10789,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -11277,11 +10806,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11293,8 +10822,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachinesOperations: # pylint: disable=too-many-public-methods """ @@ -11323,7 +10850,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -11345,16 +10871,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11365,14 +10890,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11382,11 +10907,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11398,15 +10923,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -11432,7 +10953,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11440,16 +10961,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11463,13 +10983,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -11493,14 +11009,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -11513,7 +11021,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11526,18 +11034,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -11550,7 +11050,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -11561,19 +11061,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -11606,7 +11096,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11618,20 +11108,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -11656,7 +11148,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11664,16 +11156,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11693,10 +11184,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -11719,14 +11206,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -11738,7 +11217,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11751,18 +11230,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -11771,7 +11242,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -11781,19 +11256,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -11825,7 +11289,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11835,20 +11299,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -11873,7 +11339,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11881,16 +11347,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11902,13 +11367,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11931,14 +11392,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -11950,7 +11403,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11962,18 +11415,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -11982,7 +11427,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -11991,19 +11440,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] @@ -12035,7 +11473,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12045,17 +11483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -12074,22 +11510,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12099,11 +11534,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12118,14 +11549,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12153,7 +11576,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12162,17 +11585,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12195,7 +11614,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -12214,22 +11632,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12241,13 +11658,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore - return deserialized - - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -12259,7 +11672,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -12278,21 +11690,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12304,13 +11715,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12329,21 +11736,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12353,11 +11759,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -12368,14 +11770,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12402,7 +11796,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12411,17 +11805,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -12440,22 +11830,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12465,11 +11854,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12485,14 +11870,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12520,7 +11897,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12529,17 +11906,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -12556,7 +11929,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12575,21 +11947,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12599,11 +11970,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -12619,7 +11986,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -12641,17 +12007,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12662,14 +12027,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12679,11 +12044,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12695,10 +12060,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -12714,7 +12075,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -12736,17 +12096,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12757,14 +12116,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -12774,11 +12133,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12790,8 +12149,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12803,7 +12160,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -12825,17 +12181,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -12846,14 +12201,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -12863,11 +12218,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12879,10 +12234,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -12900,22 +12251,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12925,11 +12275,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -12946,14 +12292,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12981,7 +12319,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12990,17 +12328,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13019,21 +12353,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13043,11 +12376,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13057,14 +12386,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13091,7 +12412,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13100,17 +12421,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13129,21 +12446,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13153,11 +12469,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13167,14 +12479,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13201,7 +12505,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13210,17 +12514,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13239,21 +12539,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13263,11 +12562,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13277,14 +12572,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13311,7 +12598,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13320,17 +12607,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13349,21 +12632,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13373,11 +12655,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13387,14 +12665,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13421,7 +12691,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13430,23 +12700,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -13475,7 +12741,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -13483,16 +12749,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13502,11 +12767,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -13530,14 +12791,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13548,7 +12801,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -13561,18 +12814,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13583,7 +12828,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -13593,19 +12838,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13635,7 +12870,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13644,17 +12879,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13676,7 +12907,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13695,22 +12925,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13722,13 +12951,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13747,21 +12972,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13771,11 +12995,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -13785,14 +13005,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13819,7 +13031,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13828,17 +13040,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -13850,7 +13058,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13869,21 +13076,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13893,11 +13099,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -13916,21 +13118,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13944,13 +13145,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -13962,14 +13159,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -13999,7 +13188,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14011,23 +13200,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -14053,7 +13240,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14061,16 +13248,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14084,13 +13270,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -14115,14 +13297,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -14135,7 +13309,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14148,18 +13322,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -14172,7 +13338,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -14182,20 +13348,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -14228,7 +13383,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14240,20 +13395,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -14278,7 +13435,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -14286,16 +13443,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14309,13 +13465,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14338,14 +13490,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -14357,7 +13501,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14369,18 +13513,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -14389,7 +13525,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -14398,19 +13538,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -14442,7 +13571,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14454,17 +13583,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods @@ -14494,7 +13621,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14518,16 +13644,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14538,14 +13663,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -14555,11 +13680,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14571,15 +13696,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -14605,7 +13726,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14613,16 +13734,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14642,10 +13762,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -14667,14 +13783,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14687,7 +13795,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14699,18 +13807,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14723,7 +13823,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -14732,20 +13832,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14778,7 +13867,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14788,23 +13877,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -14830,7 +13917,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -14838,16 +13925,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14859,13 +13945,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14888,14 +13970,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14908,7 +13982,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14920,18 +13994,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14944,7 +14010,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -14953,20 +14019,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -14999,7 +14055,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15009,17 +14065,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15038,22 +14092,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15063,11 +14116,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15082,14 +14131,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15117,7 +14158,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15126,17 +14167,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15157,7 +14194,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -15176,22 +14212,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15203,19 +14238,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -15244,7 +14275,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15252,16 +14283,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15271,11 +14301,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -15302,14 +14328,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15320,7 +14338,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -15335,18 +14353,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15357,7 +14367,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -15369,20 +14379,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -15412,7 +14411,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15421,23 +14420,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15465,7 +14460,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -15474,16 +14469,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15493,11 +14487,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -15526,14 +14516,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15544,7 +14526,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -15558,21 +14540,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15583,7 +14557,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15595,23 +14569,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15642,7 +14605,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15651,17 +14614,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -15673,7 +14632,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15692,21 +14650,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15718,13 +14675,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -15732,7 +14685,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15756,16 +14708,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15776,14 +14727,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -15793,11 +14744,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15809,17 +14760,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -15843,15 +14789,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15862,14 +14807,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -15879,11 +14824,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15895,8 +14840,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -15908,7 +14851,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -15932,17 +14874,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15953,14 +14894,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -15970,11 +14911,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15986,10 +14927,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -16000,7 +14937,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -16024,17 +14960,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16045,14 +14980,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -16062,11 +14997,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16078,15 +15013,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -16117,7 +15048,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16126,16 +15057,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16145,11 +15075,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -16181,14 +15107,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16199,7 +15117,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -16215,7 +15133,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -16223,14 +15141,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16241,7 +15151,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -16255,24 +15165,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16303,7 +15202,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16312,23 +15211,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16357,7 +15252,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16365,16 +15260,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16384,11 +15278,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -16413,14 +15303,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16431,7 +15313,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16444,18 +15326,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16466,7 +15340,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -16476,20 +15350,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16519,7 +15382,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16528,23 +15391,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16573,7 +15432,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16581,16 +15440,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16600,11 +15458,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -16629,14 +15483,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16647,7 +15493,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16660,18 +15506,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16682,7 +15520,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -16692,20 +15530,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16735,7 +15562,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16744,23 +15571,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -16789,7 +15612,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -16797,16 +15620,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16816,11 +15638,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -16846,14 +15664,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16864,7 +15674,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -16878,18 +15688,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16900,7 +15702,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -16911,20 +15713,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -16954,7 +15745,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16963,23 +15754,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17008,7 +15795,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17016,16 +15803,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17035,11 +15821,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -17067,14 +15849,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17085,7 +15859,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17101,18 +15875,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17123,7 +15889,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -17136,20 +15902,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17179,7 +15934,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17188,23 +15943,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -17230,7 +15981,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17238,16 +15989,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17257,11 +16007,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -17286,14 +16032,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17304,7 +16042,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17317,18 +16055,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17339,7 +16069,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -17349,20 +16079,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17392,7 +16111,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17401,23 +16120,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17446,7 +16161,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17454,16 +16169,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17473,11 +16187,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17503,14 +16213,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17521,7 +16223,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17535,18 +16237,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17557,7 +16251,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -17569,20 +16263,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17612,7 +16295,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17621,23 +16304,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17666,7 +16345,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17674,16 +16353,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17693,11 +16371,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -17723,14 +16397,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17741,7 +16407,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17755,18 +16421,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17777,7 +16435,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -17788,20 +16446,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17831,7 +16478,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17840,20 +16487,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -17879,7 +16522,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -17898,7 +16540,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -17906,16 +16548,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17927,13 +16568,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -17958,7 +16595,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17969,7 +16605,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17982,11 +16618,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17997,7 +16632,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -18008,13 +16643,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2021_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2021_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18042,7 +16674,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18050,16 +16682,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18069,17 +16700,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -18105,7 +16732,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -18113,16 +16740,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18132,11 +16758,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -18160,14 +16782,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18178,7 +16792,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18191,18 +16805,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18213,7 +16819,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -18224,19 +16830,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18266,7 +16862,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18275,17 +16871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class VirtualMachineSizesOperations: @@ -18315,7 +16907,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -18337,16 +16928,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18357,14 +16947,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -18374,11 +16964,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18390,8 +16980,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class ImagesOperations: """ @@ -18414,7 +17002,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -18439,7 +17027,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -18447,16 +17035,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18476,10 +17063,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -18501,14 +17084,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18519,7 +17094,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18531,18 +17106,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18550,7 +17117,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -18559,19 +17126,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18602,7 +17158,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18612,20 +17168,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -18650,7 +17208,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -18658,16 +17216,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18687,10 +17244,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -18712,14 +17265,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18730,7 +17275,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18742,18 +17287,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18761,7 +17298,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -18770,19 +17311,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18813,7 +17343,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18823,17 +17353,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -18852,21 +17380,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18876,11 +17403,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18890,14 +17413,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18924,7 +17439,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18933,17 +17448,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -18957,7 +17468,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -18976,22 +17486,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19003,13 +17512,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -19017,7 +17522,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19039,16 +17543,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19059,14 +17562,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19076,11 +17579,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19092,16 +17595,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19123,15 +17621,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19142,14 +17639,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19159,11 +17656,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19175,8 +17672,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -19222,7 +17717,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19233,7 +17727,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19248,11 +17742,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19263,7 +17756,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -19275,12 +17768,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19308,7 +17797,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -19316,16 +17805,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19345,10 +17833,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -19371,7 +17855,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19382,7 +17865,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19395,11 +17878,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19410,7 +17892,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -19420,12 +17902,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19453,7 +17932,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -19461,16 +17940,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19482,13 +17960,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -19507,21 +17981,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19531,11 +18004,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -19548,14 +18017,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19582,7 +18043,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -19591,17 +18052,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -19623,7 +18080,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19642,22 +18098,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19669,13 +18124,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -19683,7 +18134,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -19707,16 +18157,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19727,14 +18176,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -19744,11 +18193,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19760,17 +18209,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -19794,15 +18238,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19813,14 +18256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -19830,11 +18273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19846,8 +18289,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -19874,7 +18315,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -19900,7 +18341,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -19909,16 +18350,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19930,13 +18370,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -19963,14 +18399,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -19983,7 +18411,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19998,18 +18426,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -20022,7 +18442,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -20035,19 +18455,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RestorePoint] @@ -20080,7 +18489,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20090,17 +18499,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -20119,22 +18526,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20144,11 +18550,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -20162,14 +18564,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20197,7 +18591,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -20206,17 +18600,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -20239,7 +18629,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -20258,23 +18647,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20286,13 +18674,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -20338,7 +18722,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20349,7 +18732,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20363,11 +18746,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20378,7 +18760,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -20390,12 +18772,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20423,7 +18801,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -20431,16 +18809,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20460,10 +18837,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -20487,7 +18860,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20498,7 +18870,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20512,11 +18884,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20527,7 +18898,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -20538,12 +18909,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20571,7 +18939,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -20579,16 +18947,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20600,13 +18967,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20621,7 +18984,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20640,21 +19002,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20664,11 +19025,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20692,7 +19049,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20711,22 +19067,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20738,13 +19093,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -20765,7 +19116,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -20789,17 +19139,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20810,14 +19159,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -20827,11 +19176,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20843,10 +19192,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -20860,7 +19205,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -20884,16 +19228,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20904,14 +19247,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -20921,11 +19264,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20937,10 +19280,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -20967,7 +19306,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -20993,7 +19332,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21002,16 +19341,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21031,10 +19369,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -21061,14 +19395,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21082,7 +19408,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21098,18 +19424,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21123,7 +19441,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -21137,19 +19455,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21183,7 +19490,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21193,24 +19500,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -21236,7 +19541,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21245,16 +19550,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21268,13 +19572,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21300,14 +19600,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21321,7 +19613,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21335,18 +19627,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21360,7 +19644,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -21372,19 +19656,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21418,7 +19691,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21428,17 +19701,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -21461,22 +19732,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21486,11 +19756,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21510,14 +19776,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21545,7 +19803,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21554,17 +19812,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21589,7 +19843,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -21608,23 +19861,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21636,13 +19888,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -21655,7 +19903,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_11_01.models.CapacityReservation] @@ -21678,17 +19925,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21699,14 +19945,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -21716,11 +19962,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21732,12 +19978,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21762,7 +20004,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -21788,7 +20030,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -21797,16 +20039,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21826,10 +20067,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -21857,14 +20094,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -21878,7 +20107,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21894,18 +20123,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -21919,7 +20140,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -21932,20 +20153,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -21979,7 +20189,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21989,24 +20199,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -22032,7 +20240,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -22041,16 +20249,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22070,10 +20277,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -22101,14 +20304,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22122,7 +20317,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22138,18 +20333,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22163,7 +20350,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -22176,20 +20363,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22223,7 +20400,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22233,17 +20410,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -22262,22 +20437,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22287,11 +20461,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -22306,14 +20476,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22341,7 +20503,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22350,17 +20512,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -22382,7 +20540,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -22401,23 +20558,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22429,13 +20585,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22447,7 +20599,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -22471,17 +20622,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22492,14 +20642,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -22509,11 +20659,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22525,12 +20675,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -22567,21 +20713,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22591,11 +20736,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -22605,14 +20746,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22639,7 +20772,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22648,17 +20781,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -22677,21 +20806,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22701,11 +20829,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -22719,14 +20843,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22753,7 +20869,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22762,17 +20878,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -22791,21 +20903,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22815,11 +20926,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -22833,14 +20940,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22867,7 +20966,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22876,17 +20975,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -22898,7 +20993,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -22917,21 +21011,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22943,16 +21036,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -22978,7 +21067,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -23004,7 +21093,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23014,16 +21103,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23043,10 +21131,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -23076,14 +21160,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23098,7 +21174,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23115,18 +21191,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23141,7 +21209,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -23155,20 +21223,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23203,7 +21260,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23213,17 +21270,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -23231,7 +21286,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -23257,7 +21312,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23267,16 +21322,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23288,13 +21342,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -23325,14 +21375,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23347,7 +21389,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23364,18 +21406,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23390,7 +21424,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -23404,20 +21438,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -23452,7 +21476,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23462,17 +21486,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -23491,23 +21513,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23517,11 +21538,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23537,14 +21554,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23573,7 +21582,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23582,17 +21591,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23617,7 +21622,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -23636,7 +21640,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23644,16 +21648,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23665,13 +21668,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -23693,7 +21692,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -23712,23 +21710,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23740,13 +21737,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -23774,7 +21767,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -23803,7 +21798,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23812,16 +21807,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23831,11 +21825,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -23863,14 +21853,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23882,7 +21864,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -23897,18 +21879,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23920,7 +21894,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -23932,20 +21908,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -23976,7 +21942,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23985,17 +21951,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24014,22 +21976,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24039,11 +22000,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -24058,14 +22015,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24093,7 +22042,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24102,17 +22051,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -24131,22 +22076,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24156,11 +22100,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -24176,14 +22116,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24211,7 +22143,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24220,24 +22152,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -24263,7 +22191,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -24272,16 +22200,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24301,10 +22228,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -24331,14 +22254,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -24352,7 +22267,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24368,18 +22283,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -24393,7 +22300,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -24406,19 +22313,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -24452,7 +22348,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24462,17 +22358,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -24497,23 +22391,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24523,11 +22416,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24550,14 +22439,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24586,7 +22467,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24595,17 +22476,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24629,7 +22506,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2021_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -24648,23 +22524,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24676,13 +22551,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -24696,7 +22567,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -24715,22 +22585,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24742,13 +22611,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -24777,7 +22642,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -24801,7 +22665,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -24809,12 +22673,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24825,14 +22688,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -24842,11 +22705,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24858,10 +22721,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -24885,23 +22744,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24911,11 +22769,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -24941,14 +22795,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24977,7 +22823,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24986,17 +22832,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25015,22 +22857,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25040,11 +22881,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -25058,14 +22895,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25093,7 +22922,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25102,17 +22931,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25131,22 +22956,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25156,11 +22980,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -25174,14 +22994,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25209,7 +23021,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25218,17 +23030,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25247,22 +23055,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25272,11 +23079,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -25291,14 +23094,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25326,7 +23121,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25335,17 +23130,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -25371,7 +23162,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -25390,23 +23180,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25418,13 +23207,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -25443,22 +23228,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25468,11 +23252,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -25486,14 +23266,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25521,7 +23293,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25530,17 +23302,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -25554,7 +23322,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -25573,22 +23340,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25598,18 +23364,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -25635,7 +23397,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -25644,16 +23406,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25667,13 +23428,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -25699,14 +23456,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -25719,7 +23468,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25733,18 +23482,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -25757,7 +23498,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -25769,19 +23510,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_11_01.models.RunCommandResult] @@ -25814,7 +23544,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25826,17 +23556,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class LogAnalyticsOperations: @@ -25860,7 +23588,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -25885,23 +23613,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25915,13 +23642,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -25943,14 +23666,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -25960,7 +23675,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -25969,18 +23684,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -25990,7 +23697,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -25998,19 +23705,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26042,7 +23739,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26054,20 +23751,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26092,23 +23787,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26122,13 +23816,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26149,14 +23839,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26166,7 +23848,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26174,18 +23856,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26195,7 +23869,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26203,19 +23877,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2021_11_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26247,7 +23910,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26259,17 +23922,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26298,7 +23959,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26322,16 +23982,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26342,14 +24001,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26359,11 +24018,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26375,10 +24034,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26387,7 +24042,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26406,21 +24060,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26432,20 +24085,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26471,7 +24120,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26480,16 +24129,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26509,10 +24157,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26539,14 +24183,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26560,7 +24196,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26576,18 +24212,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26601,7 +24229,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -26614,19 +24242,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26660,7 +24277,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26670,24 +24287,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26713,7 +24328,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26722,16 +24337,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26743,13 +24357,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26777,14 +24387,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26798,7 +24400,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26814,18 +24416,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26839,7 +24433,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -26852,19 +24446,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26898,7 +24482,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26908,17 +24492,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -26937,22 +24519,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26962,11 +24543,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26981,14 +24558,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27016,7 +24585,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27025,17 +24594,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27057,7 +24622,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27076,23 +24640,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27104,13 +24667,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27124,7 +24683,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27148,18 +24706,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27170,14 +24727,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27187,11 +24744,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27203,12 +24760,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27234,7 +24787,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27260,7 +24813,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27270,16 +24823,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27299,10 +24851,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27331,14 +24879,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27353,7 +24893,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27370,18 +24910,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27396,7 +24928,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27410,19 +24942,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27457,7 +24978,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27467,17 +24988,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27485,7 +25004,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27511,7 +25030,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27521,16 +25040,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27542,13 +25060,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27578,14 +25092,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27600,7 +25106,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27617,18 +25123,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27643,7 +25141,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -27657,19 +25155,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27704,7 +25192,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27714,17 +25202,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -27743,23 +25229,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27769,11 +25254,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27789,14 +25270,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27825,7 +25298,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27834,17 +25307,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27869,7 +25338,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27888,7 +25356,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27896,16 +25364,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27917,13 +25384,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -27945,7 +25408,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27969,19 +25431,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27992,14 +25453,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28009,11 +25470,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28024,7 +25485,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_compute_management_client.py index d175ea1a9d29..e7c6109df7c7 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_configuration.py index 23f9b29e739c..61ae48ac51da 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_metadata.json index 76310488fd43..de1ce8d4c5da 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_compute_management_client.py index 300c2faa64c1..0da5ca9b5567 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2021-12-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_configuration.py index 1cb9434f660b..90c29065701d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2021-12-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/operations/_operations.py index edbb4352fda4..c6fd01195fce 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_12_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_12_01.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_12_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_12_01.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -1238,7 +1086,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1111,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1119,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1147,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -1327,14 +1170,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1346,7 +1181,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1360,18 +1195,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1380,7 +1207,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -1391,19 +1218,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1435,7 +1251,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1445,20 +1261,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -1483,7 +1301,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -1491,16 +1309,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1520,10 +1337,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -1548,14 +1361,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1567,7 +1372,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1582,18 +1387,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1602,7 +1399,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -1613,19 +1414,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1657,7 +1447,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1667,17 +1457,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -1689,7 +1477,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -1708,21 +1495,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1734,13 +1520,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -1759,21 +1541,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1783,11 +1564,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -1799,14 +1576,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1833,7 +1602,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1842,17 +1611,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -1860,7 +1625,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1883,16 +1647,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1903,14 +1666,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -1920,11 +1683,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1936,15 +1699,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.Snapshot] @@ -1967,15 +1725,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1986,14 +1743,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -2003,11 +1760,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2019,13 +1776,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2051,7 +1806,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2059,16 +1814,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2082,13 +1836,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -2114,14 +1864,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -2133,7 +1875,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2148,18 +1890,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -2171,7 +1905,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -2183,19 +1917,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -2227,7 +1950,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2239,17 +1962,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2268,21 +1989,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2292,11 +2012,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -2310,14 +2026,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2344,7 +2052,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2355,17 +2063,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -2392,7 +2096,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2418,7 +2122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2426,16 +2130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2455,10 +2158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2483,14 +2182,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2503,7 +2194,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2518,18 +2209,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2542,7 +2225,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2554,19 +2237,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2599,7 +2272,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2609,23 +2282,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2651,7 +2322,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2330,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2688,10 +2358,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2716,14 +2382,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2736,7 +2394,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2751,18 +2409,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2775,7 +2425,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -2787,19 +2437,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2832,7 +2473,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2483,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2866,7 +2505,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -2885,21 +2523,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2911,13 +2548,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -2936,21 +2569,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2960,11 +2592,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2978,14 +2606,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3012,7 +2632,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3021,17 +2641,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3041,7 +2657,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3064,16 +2679,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3084,14 +2698,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3101,11 +2715,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3117,15 +2731,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3148,15 +2757,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3167,14 +2775,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3184,11 +2792,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3200,8 +2808,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3214,7 +2820,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3236,17 +2841,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3257,14 +2861,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3274,11 +2878,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3290,10 +2894,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -3316,7 +2916,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -3341,7 +2945,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3349,16 +2953,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3378,10 +2981,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -3406,14 +3005,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3425,7 +3016,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3440,18 +3031,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3460,7 +3043,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -3471,19 +3058,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3515,7 +3091,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3525,23 +3101,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -3567,7 +3141,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -3575,16 +3149,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3604,10 +3177,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -3632,14 +3201,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3651,7 +3212,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3666,18 +3227,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3689,7 +3242,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -3701,19 +3254,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3745,7 +3287,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3755,17 +3297,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -3777,7 +3317,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -3796,21 +3335,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3822,13 +3360,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3847,21 +3381,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3871,11 +3404,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3889,14 +3418,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3923,7 +3444,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3932,17 +3453,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -3950,7 +3467,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -3973,16 +3489,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3993,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4010,11 +3525,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4026,15 +3541,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -4057,15 +3567,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4076,14 +3585,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -4093,11 +3602,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4109,8 +3618,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4123,7 +3630,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -4142,21 +3648,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4168,20 +3673,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -4207,7 +3708,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -4216,16 +3717,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4245,12 +3745,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -4278,14 +3774,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4294,12 +3782,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4317,18 +3805,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4337,12 +3817,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -4358,20 +3838,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4405,7 +3874,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4415,17 +3884,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -4441,7 +3908,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -4460,22 +3926,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4487,15 +3952,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -4512,22 +3973,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4537,14 +3997,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -4557,14 +4013,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4592,7 +4040,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4601,17 +4049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -4625,7 +4069,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -4649,17 +4092,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4670,14 +4112,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -4687,11 +4129,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4703,10 +4145,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -4749,7 +4187,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4768,23 +4205,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4796,13 +4232,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4818,7 +4250,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskRestorePoint] @@ -4841,18 +4272,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4863,14 +4293,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4880,11 +4310,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4896,17 +4326,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4932,7 +4358,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4942,16 +4368,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4965,13 +4390,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5003,14 +4424,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -5024,7 +4437,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5043,18 +4456,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -5068,7 +4473,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5084,19 +4489,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] @@ -5130,7 +4524,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5142,17 +4536,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5176,23 +4568,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5202,11 +4593,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5229,14 +4616,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_compute_management_client_enums.py index ce276a38d86c..b2ddbc1dc335 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_compute_management_client_enums.py @@ -30,8 +30,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -45,31 +45,28 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + disk and upload using write token in both disk and VM guest state""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -88,10 +85,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -101,17 +98,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -144,16 +141,16 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -161,13 +158,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -223,12 +220,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_models_py3.py index 715b334de2b3..d405826b2ef2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/models/_models_py3.py @@ -136,11 +136,11 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", and "Import". + "ImportSecure", and "UploadPreparedSecure". :vartype create_option: str or ~azure.mgmt.compute.v2021_12_01.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -207,7 +207,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", and "Import". + "CopyStart", "ImportSecure", and "UploadPreparedSecure". :paramtype create_option: str or ~azure.mgmt.compute.v2021_12_01.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -253,7 +253,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -302,7 +302,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -644,7 +644,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -721,7 +721,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] @@ -777,7 +777,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -889,7 +889,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -991,7 +991,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2021_12_01.models.Disk] @@ -1209,7 +1209,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2021_12_01.models.DiskRestorePoint] @@ -1626,7 +1626,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1750,7 +1750,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2021_12_01.models.AccessLevel @@ -1798,7 +1798,7 @@ def __init__( class ImageDiskReference(_serialization.Model): """The source image used for creating the disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: A relative uri containing either a Platform Image Repository or user image reference. Required. @@ -1864,7 +1864,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -1903,7 +1903,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_12_01.models.SourceVault @@ -1936,7 +1936,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2021_12_01.models.SourceVault @@ -2228,7 +2228,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2278,7 +2278,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2339,7 +2339,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2590,7 +2590,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2021_12_01.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/operations/_operations.py index 72f910618def..707d5351777c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2021_12_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -400,7 +400,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -513,7 +513,7 @@ def build_snapshots_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -547,7 +547,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +612,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -643,7 +643,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -697,7 +697,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -728,7 +728,7 @@ def build_disk_encryption_sets_list_associated_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -858,7 +858,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -943,7 +943,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -984,7 +984,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1022,7 +1022,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -1060,7 +1060,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1131,7 +1131,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1169,7 +1169,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1212,7 +1212,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_12_01.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_12_01.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2021_12_01.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2021_12_01.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -2409,7 +2257,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2282,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2290,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2318,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -2498,14 +2341,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2516,7 +2351,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2530,18 +2365,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2549,7 +2376,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -2560,19 +2387,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2603,7 +2419,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2613,20 +2429,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -2651,7 +2469,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2477,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2688,10 +2505,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -2716,14 +2529,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2734,7 +2539,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2749,18 +2554,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2768,7 +2565,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -2779,19 +2580,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2021_12_01.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -2822,7 +2612,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2832,17 +2622,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -2854,7 +2642,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -2873,21 +2660,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2899,13 +2685,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -2924,21 +2706,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2948,11 +2729,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2964,14 +2741,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2998,7 +2767,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3007,17 +2776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -3025,7 +2790,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3047,16 +2811,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3067,14 +2830,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -3084,11 +2847,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3100,15 +2863,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -3130,15 +2888,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3149,14 +2906,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -3166,11 +2923,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3182,13 +2939,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3214,7 +2969,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -3222,16 +2977,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3245,13 +2999,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -3277,14 +3027,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3295,7 +3037,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3310,18 +3052,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3332,7 +3066,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -3344,19 +3078,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -3387,7 +3110,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3399,17 +3122,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -3428,21 +3149,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3452,11 +3172,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3468,14 +3184,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3502,7 +3210,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -3513,17 +3221,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskEncryptionSetsOperations: @@ -3550,7 +3254,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3576,7 +3280,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3584,16 +3288,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3613,10 +3316,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3641,14 +3340,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3660,7 +3351,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3675,18 +3366,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3698,7 +3381,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3710,19 +3393,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3754,7 +3427,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3764,23 +3437,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3806,7 +3477,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3814,16 +3485,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3843,10 +3513,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -3871,14 +3537,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3890,7 +3548,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3905,18 +3563,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3928,7 +3578,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3940,19 +3590,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] @@ -3984,7 +3625,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3994,17 +3635,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4016,7 +3655,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4035,21 +3673,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4061,13 +3698,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4086,21 +3719,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4110,11 +3742,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4126,14 +3754,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4160,7 +3780,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4169,17 +3789,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4187,7 +3803,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4209,16 +3824,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4229,14 +3843,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4246,11 +3860,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4262,15 +3876,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4292,15 +3901,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4311,14 +3919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4328,11 +3936,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4344,8 +3952,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4358,7 +3964,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4380,17 +3985,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4401,14 +4005,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4418,11 +4022,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4434,10 +4038,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskAccessesOperations: """ @@ -4460,7 +4060,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -4485,7 +4089,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4493,16 +4097,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4522,10 +4125,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -4550,14 +4149,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4568,7 +4159,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4583,18 +4174,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4602,7 +4185,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -4613,19 +4200,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4656,7 +4232,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4666,23 +4242,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -4708,7 +4282,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -4716,16 +4290,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4745,10 +4318,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -4773,14 +4342,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4791,7 +4352,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4806,18 +4367,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4828,7 +4381,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -4840,19 +4393,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2021_12_01.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -4883,7 +4425,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4893,17 +4435,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -4915,7 +4455,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -4934,21 +4473,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4498,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -4985,21 +4519,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5009,11 +4542,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5025,14 +4554,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5059,7 +4580,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5068,17 +4589,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -5086,7 +4603,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5108,16 +4624,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5128,14 +4643,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -5145,11 +4660,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5161,15 +4676,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -5191,15 +4701,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5210,14 +4719,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -5227,11 +4736,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5243,8 +4752,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -5257,7 +4764,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -5276,21 +4782,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5302,20 +4807,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -5341,7 +4842,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -5350,16 +4851,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5379,12 +4879,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -5412,14 +4908,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5428,12 +4916,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5451,18 +4939,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5471,12 +4951,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -5492,20 +4972,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5539,7 +5008,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5549,17 +5018,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -5575,7 +5042,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -5594,22 +5060,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5621,15 +5086,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -5646,22 +5107,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5671,14 +5131,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -5691,14 +5147,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5726,7 +5174,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5735,17 +5183,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -5759,7 +5203,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -5783,17 +5226,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5804,14 +5246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -5821,11 +5263,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5837,10 +5279,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskRestorePointOperations: """ @@ -5883,7 +5321,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2021_12_01.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -5902,23 +5339,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5930,13 +5366,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -5952,7 +5384,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2021_12_01.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -5974,18 +5405,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5996,14 +5426,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -6013,11 +5443,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6029,17 +5459,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6065,7 +5491,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -6075,16 +5501,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6098,13 +5523,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6136,14 +5557,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6156,7 +5569,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6175,18 +5588,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6199,7 +5604,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -6215,19 +5620,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2021_12_01.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2021_12_01.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6260,7 +5654,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6272,17 +5666,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -6306,23 +5698,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2021-12-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6332,11 +5723,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -6359,14 +5746,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_compute_management_client.py index 754b78643766..0b12cf538de5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -95,7 +97,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -138,7 +158,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-01-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -158,7 +178,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_configuration.py index 8f729b4eacf6..a37b947f5a2d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-01-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_metadata.json index af8cb3891439..aab4e1048c3f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "galleries": "GalleriesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_compute_management_client.py index 745b3d28ed5e..7a71e5d8d46f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -96,7 +98,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -139,7 +159,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-01-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -159,7 +181,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_configuration.py index 1782bb15ed7f..b69881c2d2fe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-01-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/operations/_operations.py index 2e426376d58e..0f83f760f313 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -162,10 +161,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -190,14 +185,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -209,7 +196,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -224,18 +211,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -244,7 +223,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -255,19 +234,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_01_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_01_03.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -299,7 +267,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -309,20 +277,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -347,7 +317,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -355,16 +325,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,13 +345,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -407,14 +372,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -426,7 +383,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -440,18 +397,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -460,7 +409,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -471,19 +424,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_01_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_01_03.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -515,7 +457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -525,17 +467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -559,7 +499,6 @@ async def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -578,23 +517,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -606,13 +544,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -631,21 +565,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -655,11 +588,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -669,14 +598,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -703,7 +624,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -712,17 +633,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -730,7 +647,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -752,16 +668,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -772,14 +687,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -789,11 +704,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -805,15 +720,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -835,15 +745,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -854,14 +763,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -871,11 +780,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -887,8 +796,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -915,7 +822,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -941,7 +848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -950,16 +857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,10 +888,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -1014,14 +916,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1034,7 +928,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1052,18 +946,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1076,7 +962,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -1091,19 +977,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1136,7 +1011,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1146,24 +1021,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1189,7 +1062,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1198,16 +1071,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1219,13 +1091,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1254,14 +1122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1274,7 +1134,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1291,18 +1151,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1315,7 +1167,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -1330,19 +1182,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1375,7 +1216,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1385,17 +1226,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1410,7 +1249,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1429,22 +1267,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1456,13 +1293,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -1481,22 +1314,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1506,11 +1338,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1525,14 +1353,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1560,7 +1380,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1569,17 +1389,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -1592,7 +1408,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -1615,17 +1430,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1636,14 +1450,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -1653,11 +1467,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,10 +1483,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -1700,7 +1510,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1726,7 +1536,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1736,16 +1546,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1768,10 +1577,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -1805,14 +1610,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1827,7 +1624,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1849,18 +1646,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1875,7 +1664,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -1894,19 +1683,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1941,7 +1720,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1951,17 +1730,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -1969,7 +1746,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1995,7 +1772,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2005,16 +1782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2026,13 +1802,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2067,14 +1839,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2089,7 +1853,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2111,18 +1875,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2137,7 +1893,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -2156,20 +1912,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2204,7 +1949,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2214,17 +1959,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2253,7 +1996,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -2272,7 +2014,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2280,16 +2022,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2301,13 +2042,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -2331,23 +2068,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2357,11 +2093,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2385,14 +2117,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2421,7 +2145,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2430,17 +2154,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -2456,7 +2176,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion] @@ -2479,18 +2198,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2501,14 +2219,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -2518,11 +2236,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2534,10 +2252,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -2564,7 +2278,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2590,7 +2304,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2599,16 +2313,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2631,10 +2344,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -2663,14 +2372,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2684,7 +2385,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2702,18 +2403,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2727,7 +2420,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -2742,19 +2435,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2788,7 +2471,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2798,24 +2481,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2841,7 +2522,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2850,16 +2531,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2871,13 +2551,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2907,14 +2583,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2928,7 +2596,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2946,18 +2614,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2971,7 +2631,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -2986,20 +2646,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3033,7 +2682,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3043,17 +2692,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3069,7 +2716,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3088,22 +2734,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3115,13 +2760,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -3140,22 +2781,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3165,11 +2805,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3185,14 +2821,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3220,7 +2848,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3229,17 +2857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -3252,7 +2876,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryApplication] @@ -3275,17 +2898,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3296,14 +2918,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -3313,11 +2935,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3329,10 +2951,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -3360,7 +2978,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3386,7 +3004,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3396,16 +3014,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3428,10 +3045,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -3466,14 +3079,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3488,7 +3093,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3510,18 +3115,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3536,7 +3133,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -3555,21 +3152,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3604,7 +3190,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3614,17 +3200,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -3632,7 +3216,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3658,7 +3242,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3668,16 +3252,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3689,13 +3272,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3731,14 +3310,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3753,7 +3324,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3775,18 +3346,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3801,7 +3364,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -3820,20 +3383,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3868,7 +3421,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3878,17 +3431,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3917,7 +3468,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3936,7 +3486,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3944,16 +3494,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3965,13 +3514,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3995,23 +3540,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4021,11 +3565,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4049,14 +3589,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4085,7 +3617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4094,17 +3626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -4120,7 +3648,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4144,18 +3671,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4166,14 +3692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -4183,11 +3709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4199,10 +3725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -4228,7 +3750,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -4254,7 +3776,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4262,16 +3784,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4291,10 +3812,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -4316,14 +3833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4336,7 +3845,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4348,18 +3857,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4372,7 +3873,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -4382,19 +3883,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_01_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_01_03.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4427,7 +3917,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4437,17 +3927,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -4481,7 +3969,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.SharedGallery] @@ -4504,17 +3991,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4525,14 +4011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -4542,11 +4028,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4558,10 +4044,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -4570,7 +4052,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,21 +4070,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4615,13 +4095,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -4662,7 +4138,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImage] @@ -4685,18 +4160,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4707,14 +4181,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -4724,11 +4198,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4740,10 +4214,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -4757,7 +4227,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4776,22 +4245,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4803,13 +4271,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -4854,7 +4318,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -4878,19 +4341,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4901,14 +4363,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -4918,11 +4380,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4934,10 +4396,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -4961,7 +4419,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4980,23 +4437,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5008,13 +4464,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -5045,7 +4497,6 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5064,21 +4515,20 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5090,13 +4540,9 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -5131,7 +4577,6 @@ async def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5150,22 +4595,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4621,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4635,6 @@ def list( :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -5219,17 +4658,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -5257,11 +4695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,10 +4711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -5320,7 +4754,6 @@ async def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5339,23 +4772,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5367,13 +4799,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5387,7 +4815,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -5411,18 +4838,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5433,14 +4859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -5450,11 +4876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5465,7 +4891,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/models/_models_py3.py index 84cbde5e412c..02f3d78c0cc6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/models/_models_py3.py @@ -159,26 +159,6 @@ class CommunityGallery(PirCommunityGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this community gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class CommunityGalleryImage(PirCommunityGalleryResource): # pylint: disable=too-many-instance-attributes """Specifies information about the gallery image definition that you want to create or update. @@ -328,7 +308,7 @@ def __init__( class CommunityGalleryImageList(_serialization.Model): """The List Community Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImage] @@ -441,7 +421,7 @@ def __init__( class CommunityGalleryImageVersionList(_serialization.Model): """The List Community Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImageVersion] @@ -565,7 +545,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -690,7 +670,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -739,7 +719,7 @@ class Gallery(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -833,7 +813,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -934,7 +914,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.GalleryApplication] @@ -1107,7 +1087,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1179,7 +1159,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion] @@ -1312,12 +1292,12 @@ def __init__( class GalleryApplicationVersionPublishingProfile( GalleryArtifactPublishingProfileBase -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1522,7 +1502,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2022_01_03.models.ManagedArtifact @@ -1626,7 +1606,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -1737,7 +1717,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1938,7 +1918,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -1978,7 +1958,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2182,7 +2162,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2260,7 +2240,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion] @@ -2326,71 +2306,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) list[~azure.mgmt.compute.v2022_01_03.models.GalleryTargetExtendedLocation] """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - "target_extended_locations": {"key": "targetExtendedLocations", "type": "[GalleryTargetExtendedLocation]"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - target_extended_locations: Optional[List["_models.GalleryTargetExtendedLocation"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2022_01_03.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", - "Premium_LRS", and "StandardSSD_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2022_01_03.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2022_01_03.models.ReplicationMode - :keyword target_extended_locations: The target extended locations where the Image Version is - going to be replicated to. This property is updatable. - :paramtype target_extended_locations: - list[~azure.mgmt.compute.v2022_01_03.models.GalleryTargetExtendedLocation] - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - target_extended_locations=target_extended_locations, - **kwargs - ) - class GalleryImageVersionStorageProfile(_serialization.Model): """This is the storage profile of a Gallery Image Version. @@ -2505,7 +2420,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.Gallery] @@ -2550,32 +2465,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2022_01_03.models.GalleryArtifactVersionSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryArtifactVersionSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryArtifactVersionSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_01_03.models.HostCaching - :keyword source: The gallery artifact version source. - :paramtype source: ~azure.mgmt.compute.v2022_01_03.models.GalleryArtifactVersionSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryTargetExtendedLocation(_serialization.Model): """GalleryTargetExtendedLocation. @@ -2793,7 +2682,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3176,24 +3065,6 @@ class SharedGallery(PirSharedGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this shared gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class SharedGalleryDiskImage(_serialization.Model): """This is the disk image base class. @@ -3234,7 +3105,7 @@ class SharedGalleryDataDiskImage(SharedGalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_size_gb: This property indicates the size of the VHD to be created. :vartype disk_size_gb: int @@ -3404,7 +3275,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImage] @@ -3513,7 +3384,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImageVersion] @@ -3583,7 +3454,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_01_03.models.SharedGallery] @@ -3626,25 +3497,6 @@ class SharedGalleryOSDiskImage(SharedGalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryHostCaching """ - _validation = { - "disk_size_gb": {"readonly": True}, - } - - _attribute_map = { - "disk_size_gb": {"key": "diskSizeGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__( - self, *, host_caching: Optional[Union[str, "_models.SharedGalleryHostCaching"]] = None, **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryHostCaching - """ - super().__init__(host_caching=host_caching, **kwargs) - class SharingProfile(_serialization.Model): """Profile for gallery sharing to subscription or tenant. @@ -3768,7 +3620,7 @@ def __init__(self, *, summary: Optional[List["_models.RegionalSharingStatus"]] = class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. :code:`
    `:code:`
    ` Possible values are: :code:`
    `:code:`
    ` **Add** @@ -3879,7 +3731,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -3940,7 +3792,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -4024,7 +3876,7 @@ def __init__( class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/operations/_operations.py index fb6173d95622..3a519ce0f449 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_01_03/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -179,7 +179,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -233,7 +233,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -367,7 +367,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -398,7 +398,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -439,7 +439,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -522,7 +522,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -560,7 +560,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +592,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +627,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -694,7 +694,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -726,7 +726,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -757,7 +757,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -800,7 +800,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -843,7 +843,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -887,7 +887,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -927,7 +927,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -959,7 +959,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1130,7 +1130,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1170,7 +1170,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1239,7 +1239,7 @@ def build_community_galleries_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_get_request( +def build_community_gallery_images_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1271,7 +1271,7 @@ def build_community_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_list_request( +def build_community_gallery_images_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1302,7 +1302,7 @@ def build_community_gallery_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_get_request( +def build_community_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, @@ -1340,7 +1340,7 @@ def build_community_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_list_request( +def build_community_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1393,7 +1393,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1418,7 +1418,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1426,16 +1426,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1458,10 +1457,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -1486,14 +1481,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1504,7 +1491,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1519,18 +1506,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1538,7 +1517,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -1549,19 +1528,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_01_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_01_03.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1592,7 +1560,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1602,20 +1570,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1640,7 +1610,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1648,16 +1618,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,13 +1638,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -1700,14 +1665,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,7 +1675,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1732,18 +1689,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1751,7 +1700,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -1762,19 +1715,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_01_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_01_03.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1805,7 +1747,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1815,17 +1757,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -1849,7 +1789,6 @@ def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -1868,23 +1807,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1896,13 +1834,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -1921,21 +1855,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1945,11 +1878,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1959,14 +1888,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1993,7 +1914,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2002,17 +1923,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -2020,7 +1937,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2042,16 +1958,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2062,14 +1977,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2079,11 +1994,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2095,15 +2010,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2125,15 +2035,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2144,14 +2053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2161,11 +2070,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2177,8 +2086,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -2205,7 +2112,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2231,7 +2138,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2240,16 +2147,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,10 +2178,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -2304,14 +2206,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2324,7 +2218,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2342,18 +2236,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2366,7 +2252,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -2381,19 +2267,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2426,7 +2301,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,24 +2311,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2479,7 +2352,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2488,16 +2361,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2509,13 +2381,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2544,14 +2412,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2564,7 +2424,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2581,18 +2441,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2605,7 +2457,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -2620,19 +2472,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] @@ -2665,7 +2506,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2675,17 +2516,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -2700,7 +2539,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2719,22 +2557,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2746,13 +2583,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -2771,22 +2604,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2796,11 +2628,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -2815,14 +2643,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2850,7 +2670,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2859,17 +2679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -2882,7 +2698,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2904,17 +2719,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2925,14 +2739,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -2942,11 +2756,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,10 +2772,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -2989,7 +2799,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3015,7 +2825,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3025,16 +2835,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3057,10 +2866,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -3094,14 +2899,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3116,7 +2913,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3138,18 +2935,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3164,7 +2953,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -3183,19 +2972,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3230,7 +3009,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3240,17 +3019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -3258,7 +3035,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3284,7 +3061,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3294,16 +3071,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3315,13 +3091,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3356,14 +3128,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3378,7 +3142,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3164,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3426,7 +3182,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -3445,20 +3201,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3493,7 +3238,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3503,17 +3248,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3542,7 +3285,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3561,7 +3303,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3569,16 +3311,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3590,13 +3331,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3620,23 +3357,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3646,11 +3382,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -3674,14 +3406,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3710,7 +3434,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3719,17 +3443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -3745,7 +3465,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryImageVersion] @@ -3768,18 +3487,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3790,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -3807,11 +3525,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3823,10 +3541,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -3853,7 +3567,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3879,7 +3593,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3888,16 +3602,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3920,10 +3633,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -3952,14 +3661,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3973,7 +3674,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3991,18 +3692,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4016,7 +3709,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -4031,19 +3724,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4077,7 +3760,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4087,24 +3770,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -4130,7 +3811,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4139,16 +3820,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4160,13 +3840,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4196,14 +3872,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4217,7 +3885,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4235,18 +3903,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4260,7 +3920,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -4275,20 +3935,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4322,7 +3971,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4332,17 +3981,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4358,7 +4005,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -4377,22 +4023,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4404,13 +4049,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -4429,22 +4070,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4454,11 +4094,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4474,14 +4110,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4509,7 +4137,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4518,17 +4146,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -4541,7 +4165,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -4563,17 +4186,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4584,14 +4206,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -4601,11 +4223,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4617,10 +4239,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -4648,7 +4266,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4674,7 +4292,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4684,16 +4302,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4716,10 +4333,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -4754,14 +4367,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4776,7 +4381,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4798,18 +4403,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4824,7 +4421,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -4843,21 +4440,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4892,7 +4478,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4902,17 +4488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -4920,7 +4504,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4946,7 +4530,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4956,16 +4540,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4977,13 +4560,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5019,14 +4598,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5041,7 +4612,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5063,18 +4634,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5089,7 +4652,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -5108,20 +4671,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5156,7 +4709,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5166,17 +4719,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5205,7 +4756,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_01_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5224,7 +4774,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -5232,16 +4782,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5253,13 +4802,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -5283,23 +4828,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5309,11 +4853,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5337,14 +4877,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5373,7 +4905,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5382,17 +4914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -5408,7 +4936,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5432,18 +4959,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5454,14 +4980,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -5471,11 +4997,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5487,10 +5013,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -5516,7 +5038,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -5542,7 +5064,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5550,16 +5072,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5579,10 +5100,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -5604,14 +5121,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.SharingUpdate] @@ -5623,7 +5132,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5635,18 +5144,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.SharingUpdate] @@ -5658,7 +5159,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -5668,19 +5169,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_01_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_01_03.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_01_03.models.SharingUpdate] @@ -5712,7 +5202,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5722,17 +5212,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -5766,7 +5254,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5788,17 +5275,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5809,14 +5295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -5826,11 +5312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5842,10 +5328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -5854,7 +5336,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5873,21 +5354,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5899,13 +5379,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -5946,7 +5422,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5968,18 +5443,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5990,14 +5464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -6007,11 +5481,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6023,10 +5497,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -6040,7 +5510,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6059,22 +5528,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6086,13 +5554,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -6137,7 +5601,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_01_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -6161,19 +5624,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6184,14 +5646,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -6201,11 +5663,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6217,10 +5679,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -6244,7 +5702,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6263,23 +5720,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6291,13 +5747,9 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -6328,7 +5780,6 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -6347,21 +5798,20 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6373,13 +5823,9 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -6414,7 +5860,6 @@ def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6433,22 +5878,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6460,13 +5904,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterable["_models.CommunityGalleryImage"]: @@ -6476,7 +5916,6 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -6500,17 +5939,16 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6521,14 +5959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -6538,11 +5976,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6554,10 +5992,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -6601,7 +6035,6 @@ def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_01_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6620,23 +6053,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-01-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,13 +6080,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6668,7 +6096,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -6692,18 +6119,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6714,14 +6140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -6731,11 +6157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6746,7 +6172,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_compute_management_client.py index ede6b2b7be49..2bbf83a71a63 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_configuration.py index e5b471f561f6..10629f815c56 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_metadata.json index b156f5397041..2ec1956d0e03 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_compute_management_client.py index f149b355a39f..e50d3db1c336 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_configuration.py index 0bc5e80658a3..ae96c8a88df6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/operations/_operations.py index 2eaedc162acf..e4dde36f3c6b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -234,7 +234,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -258,14 +257,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -276,14 +274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -293,11 +291,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -309,8 +307,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -339,7 +335,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -361,16 +356,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -381,14 +375,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -398,11 +392,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -414,8 +408,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -444,7 +436,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] @@ -467,16 +458,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -487,14 +477,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -504,11 +494,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -520,8 +510,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -550,7 +538,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -574,16 +561,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -594,14 +580,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -611,11 +597,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -627,15 +613,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -661,7 +643,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -669,16 +651,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -698,10 +679,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -723,14 +700,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -743,7 +712,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -755,18 +724,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -779,7 +740,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -788,20 +749,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -834,7 +784,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -844,23 +794,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -886,7 +834,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -894,16 +842,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -915,13 +862,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -944,14 +887,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -964,7 +899,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -976,18 +911,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1000,7 +927,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1009,20 +936,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1055,7 +972,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1065,17 +982,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1094,22 +1009,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1119,11 +1033,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1138,14 +1048,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1173,7 +1075,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1182,17 +1084,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1213,7 +1111,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1232,22 +1129,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1259,19 +1155,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -1300,7 +1192,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1308,16 +1200,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1327,11 +1218,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1358,14 +1245,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1376,7 +1255,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1391,18 +1270,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1413,7 +1284,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -1425,20 +1296,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -1468,7 +1328,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1477,23 +1337,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1521,7 +1377,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1530,16 +1386,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1549,11 +1404,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1582,14 +1433,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1600,7 +1443,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1614,21 +1457,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1639,7 +1474,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1651,23 +1486,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1698,7 +1522,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1707,17 +1531,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1729,7 +1549,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1748,21 +1567,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1774,13 +1592,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1788,7 +1602,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1812,16 +1625,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1832,14 +1644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1849,11 +1661,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1865,17 +1677,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1899,15 +1706,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1918,14 +1724,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -1935,11 +1741,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1951,8 +1757,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -1964,7 +1768,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -1988,17 +1791,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2009,14 +1811,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2026,11 +1828,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2042,10 +1844,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2056,7 +1854,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2080,17 +1877,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2101,14 +1897,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2118,11 +1914,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2134,15 +1930,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2173,7 +1965,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2182,16 +1974,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2201,11 +1992,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2237,14 +2024,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2255,7 +2034,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2271,7 +2050,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2279,14 +2058,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2297,7 +2068,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2311,24 +2082,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2359,7 +2119,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2368,23 +2128,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2413,7 +2169,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2421,16 +2177,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2440,11 +2195,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2469,14 +2220,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2487,7 +2230,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2500,18 +2243,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2522,7 +2257,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2532,20 +2267,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2575,7 +2299,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2584,23 +2308,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2629,7 +2349,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2637,16 +2357,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2656,11 +2375,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2685,14 +2400,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2703,7 +2410,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2716,18 +2423,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2738,7 +2437,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2748,20 +2447,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2791,7 +2479,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2800,23 +2488,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2845,7 +2529,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2853,16 +2537,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2872,11 +2555,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -2902,14 +2581,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2920,7 +2591,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2934,18 +2605,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2956,7 +2619,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -2967,20 +2630,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3010,7 +2662,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3019,23 +2671,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3064,7 +2712,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3072,16 +2720,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3091,11 +2738,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3123,14 +2766,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3141,7 +2776,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3157,18 +2792,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3179,7 +2806,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3192,20 +2819,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3235,7 +2851,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3244,23 +2860,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3286,7 +2898,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3294,16 +2906,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3313,11 +2924,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3342,14 +2949,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3360,7 +2959,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3373,18 +2972,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3395,7 +2986,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3405,20 +2996,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3448,7 +3028,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3457,23 +3037,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3502,7 +3078,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3510,16 +3086,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3529,11 +3104,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3559,14 +3130,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3577,7 +3140,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3591,18 +3154,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3613,7 +3168,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3625,20 +3180,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3668,7 +3212,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3677,23 +3221,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3722,7 +3262,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3730,16 +3270,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3749,11 +3288,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -3779,14 +3314,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3797,7 +3324,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3811,18 +3338,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3833,7 +3352,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -3844,20 +3363,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3887,7 +3395,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3896,20 +3404,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -3935,7 +3439,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -3954,7 +3457,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3962,16 +3465,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3983,13 +3485,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4014,7 +3512,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4025,7 +3522,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4038,11 +3535,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4053,7 +3549,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4064,13 +3560,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4098,7 +3591,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4106,16 +3599,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4125,17 +3617,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4161,7 +3649,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4169,16 +3657,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4188,11 +3675,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4216,14 +3699,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4234,7 +3709,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4247,18 +3722,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4269,7 +3736,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4280,19 +3747,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4322,7 +3779,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4331,20 +3788,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4369,7 +3822,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4395,7 +3848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4404,16 +3857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4433,10 +3885,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4464,14 +3912,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4485,7 +3925,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4501,18 +3941,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4526,7 +3958,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4539,20 +3971,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4586,7 +4007,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4596,24 +4017,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4639,7 +4058,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4648,16 +4067,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4677,10 +4095,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -4708,14 +4122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4729,7 +4135,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4745,18 +4151,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4770,7 +4168,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -4783,20 +4181,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4830,7 +4218,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4840,17 +4228,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -4869,22 +4255,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4894,11 +4279,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4913,14 +4294,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4948,7 +4321,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4957,17 +4330,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4989,7 +4358,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5008,23 +4376,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5036,13 +4403,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5054,7 +4417,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5078,17 +4440,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5099,14 +4460,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5116,11 +4477,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5132,12 +4493,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5174,21 +4531,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5198,11 +4554,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5214,14 +4566,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5248,7 +4592,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5257,17 +4601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5286,21 +4626,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5310,11 +4649,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5328,14 +4663,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5362,7 +4689,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5371,17 +4698,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5400,21 +4723,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5424,11 +4746,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5442,14 +4760,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5476,7 +4786,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5485,17 +4795,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5507,7 +4813,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5526,21 +4831,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5552,16 +4856,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5587,7 +4887,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5613,7 +4913,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5623,16 +4923,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5652,10 +4951,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -5685,14 +4980,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5707,7 +4994,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5724,18 +5011,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5750,7 +5029,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -5764,20 +5043,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5812,7 +5080,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5822,17 +5090,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -5840,7 +5106,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5866,7 +5132,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5876,16 +5142,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5897,13 +5162,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -5934,14 +5195,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5956,7 +5209,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5973,18 +5226,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5999,7 +5244,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6013,20 +5258,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6061,7 +5296,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6071,17 +5306,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6100,23 +5333,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6126,11 +5358,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6146,14 +5374,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6182,7 +5402,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6191,17 +5411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6226,7 +5442,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6245,7 +5460,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6253,16 +5468,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6274,13 +5488,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6302,7 +5512,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6321,23 +5530,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6349,13 +5557,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6383,7 +5587,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6412,7 +5618,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6421,16 +5627,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6440,11 +5645,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6472,14 +5673,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6491,7 +5684,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6506,18 +5699,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6529,7 +5714,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6541,20 +5728,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6585,7 +5762,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6594,17 +5771,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6623,22 +5796,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,11 +5820,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -6667,14 +5835,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6702,7 +5862,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6711,17 +5871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6740,22 +5896,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6765,11 +5920,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6785,14 +5936,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6820,7 +5963,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6829,24 +5972,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -6872,7 +6011,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6881,16 +6020,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6910,10 +6048,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -6940,14 +6074,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -6961,7 +6087,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6977,18 +6103,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7002,7 +6120,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -7015,19 +6133,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7061,7 +6168,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7071,17 +6178,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7106,23 +6211,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7132,11 +6236,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7159,14 +6259,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7195,7 +6287,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7204,17 +6296,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7238,7 +6326,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7257,23 +6344,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7285,13 +6371,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7305,7 +6387,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7324,22 +6405,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7351,13 +6431,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7386,7 +6462,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7410,7 +6485,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7418,12 +6493,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7434,14 +6508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7451,11 +6525,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7467,10 +6541,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -7494,23 +6564,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7520,11 +6589,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7550,14 +6615,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7586,7 +6643,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7595,17 +6652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7624,22 +6677,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7649,11 +6701,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -7667,14 +6715,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7702,7 +6742,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7711,17 +6751,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7740,22 +6776,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7765,11 +6800,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -7783,14 +6814,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7818,7 +6841,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7827,17 +6850,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7856,22 +6875,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7881,11 +6899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -7900,14 +6914,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7935,7 +6941,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7944,17 +6950,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7980,7 +6982,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7999,23 +7000,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8027,13 +7027,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8052,22 +7048,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8077,11 +7072,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8095,14 +7086,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8130,7 +7113,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8139,17 +7122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8163,7 +7142,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8182,22 +7160,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8207,18 +7184,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -8244,7 +7217,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8253,16 +7226,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8276,13 +7248,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8308,14 +7276,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8329,7 +7289,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8343,18 +7303,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8368,7 +7320,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -8380,19 +7332,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8426,7 +7367,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8438,17 +7379,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -8476,7 +7415,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8502,7 +7441,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8511,16 +7450,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8540,10 +7478,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8570,14 +7504,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8591,7 +7517,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8607,18 +7533,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8632,7 +7550,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8645,20 +7563,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8692,7 +7599,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8702,24 +7609,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8745,7 +7650,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8754,16 +7659,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8775,13 +7679,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8810,14 +7710,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8831,7 +7723,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8847,18 +7739,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8872,7 +7756,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8885,20 +7769,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8932,7 +7805,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8942,17 +7815,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8971,22 +7842,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8996,11 +7866,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9015,14 +7881,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9050,7 +7908,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9059,17 +7917,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9091,7 +7945,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9110,23 +7963,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9138,13 +7990,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9158,7 +8006,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9177,22 +8024,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9204,13 +8050,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -9240,7 +8082,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -9263,16 +8104,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9283,14 +8123,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9300,11 +8140,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9316,15 +8156,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9350,7 +8186,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9358,16 +8194,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9381,13 +8216,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9411,14 +8242,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9431,7 +8254,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9444,18 +8267,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9468,7 +8283,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9479,19 +8294,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9524,7 +8329,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9536,20 +8341,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9574,7 +8381,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9582,16 +8389,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9611,10 +8417,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9637,14 +8439,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9657,7 +8451,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9670,18 +8464,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9691,7 +8477,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9701,19 +8491,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9746,7 +8525,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9756,20 +8535,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9794,7 +8575,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9802,16 +8583,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9823,13 +8603,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9852,14 +8628,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9872,7 +8640,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9884,18 +8652,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9905,7 +8665,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -9914,19 +8678,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9959,7 +8712,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9969,17 +8722,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9998,22 +8749,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10023,11 +8773,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10042,14 +8788,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10077,7 +8815,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10086,17 +8824,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10119,7 +8853,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10138,22 +8871,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10165,13 +8897,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10183,7 +8911,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10202,21 +8929,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10228,13 +8954,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10253,21 +8975,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10277,11 +8998,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10294,14 +9011,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10328,7 +9037,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10337,17 +9046,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -10366,22 +9071,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10391,11 +9095,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10411,14 +9111,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10446,7 +9138,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10455,17 +9147,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10482,7 +9170,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10501,21 +9188,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10525,11 +9211,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -10545,7 +9227,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -10568,17 +9249,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10589,14 +9269,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10606,11 +9286,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10622,10 +9302,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -10641,7 +9317,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -10664,17 +9339,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10685,14 +9359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10702,11 +9376,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10718,8 +9392,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10731,7 +9403,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] @@ -10754,17 +9425,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10775,14 +9445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10792,11 +9462,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10808,10 +9478,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10829,22 +9495,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10854,11 +9519,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -10875,14 +9536,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10910,7 +9563,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10919,17 +9572,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10948,21 +9597,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10972,11 +9620,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -10986,14 +9630,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11020,7 +9656,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11029,17 +9665,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11058,21 +9690,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11082,11 +9713,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11096,14 +9723,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11130,7 +9749,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11139,17 +9758,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11168,21 +9783,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11192,11 +9806,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11206,14 +9816,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11240,7 +9842,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11249,17 +9851,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11278,21 +9876,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11302,11 +9899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11316,14 +9909,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11350,7 +9935,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11359,23 +9944,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11404,7 +9985,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11412,16 +9993,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11431,11 +10011,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11459,14 +10035,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11477,7 +10045,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11490,18 +10058,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11512,7 +10072,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11522,19 +10082,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11564,7 +10114,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11573,17 +10123,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11605,7 +10151,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11624,22 +10169,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11651,13 +10195,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11676,21 +10216,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11700,11 +10239,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11716,14 +10251,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11750,7 +10277,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11759,17 +10286,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11781,7 +10304,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11800,21 +10322,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11824,11 +10345,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11847,21 +10364,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11875,13 +10391,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -11893,14 +10405,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -11930,7 +10434,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11942,23 +10446,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -11984,7 +10486,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11992,16 +10494,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12015,13 +10516,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12046,14 +10543,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12066,7 +10555,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12079,18 +10568,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12103,7 +10584,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12113,20 +10594,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12159,7 +10629,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12171,20 +10641,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12209,7 +10681,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12217,16 +10689,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12240,13 +10711,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12269,14 +10736,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12289,7 +10748,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12301,18 +10760,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12322,7 +10773,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12331,19 +10786,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12376,7 +10820,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12388,17 +10832,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -12437,7 +10879,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12456,7 +10897,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12464,16 +10905,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12485,13 +10925,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12523,7 +10959,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12542,7 +10977,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12552,16 +10987,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12573,13 +11007,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -12591,7 +11021,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12610,21 +11039,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12636,13 +11064,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -12650,7 +11074,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12669,20 +11092,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12694,13 +11116,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -12714,7 +11132,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12733,22 +11150,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12760,13 +11176,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -12778,7 +11190,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12797,21 +11208,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12823,13 +11233,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -12870,7 +11276,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12889,7 +11294,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -12898,16 +11303,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12919,13 +11323,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12962,7 +11362,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12981,7 +11380,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -12992,16 +11391,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13013,13 +11411,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13034,7 +11428,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13053,22 +11446,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13080,13 +11472,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -13098,7 +11486,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13117,21 +11504,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13143,13 +11529,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13166,7 +11548,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13185,23 +11566,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13213,13 +11593,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13256,7 +11632,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13275,23 +11650,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13303,13 +11677,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -13321,7 +11691,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13340,21 +11709,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13366,13 +11734,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -13400,7 +11764,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13419,7 +11782,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13428,16 +11791,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13449,13 +11811,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -13499,7 +11857,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13510,7 +11867,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13522,11 +11879,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13537,7 +11893,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -13547,12 +11903,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13580,7 +11932,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13588,16 +11940,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13609,13 +11960,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -13638,7 +11985,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13649,7 +11995,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13661,11 +12007,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13676,7 +12021,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -13686,12 +12031,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13719,7 +12060,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13727,16 +12068,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13748,13 +12088,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -13766,7 +12102,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13785,21 +12120,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13809,11 +12143,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -13823,7 +12153,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13842,21 +12171,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13868,13 +12196,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -13885,7 +12209,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet] @@ -13908,16 +12231,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13928,14 +12250,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -13945,11 +12267,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13961,17 +12283,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet] @@ -13994,16 +12311,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14014,14 +12330,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14031,11 +12347,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14047,10 +12363,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -14062,7 +12374,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] @@ -14085,17 +12396,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14106,14 +12416,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -14123,11 +12433,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14139,10 +12449,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -14186,7 +12492,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14197,7 +12502,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14210,11 +12515,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14225,7 +12529,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -14235,12 +12539,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14268,7 +12568,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14276,16 +12576,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14305,10 +12604,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -14331,7 +12626,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14342,7 +12636,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14355,11 +12649,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14370,7 +12663,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -14380,12 +12673,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14413,7 +12703,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14421,16 +12711,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14442,13 +12731,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14460,7 +12745,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14479,21 +12763,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14503,11 +12786,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14527,7 +12806,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14546,22 +12824,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14573,19 +12850,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14609,15 +12881,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14628,14 +12899,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14645,11 +12916,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14661,10 +12932,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14673,7 +12940,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14697,16 +12963,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14717,14 +12982,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14734,11 +12999,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14750,10 +13015,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -14798,7 +13059,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14809,7 +13069,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14823,11 +13083,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14838,7 +13097,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -14850,12 +13109,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14883,7 +13138,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -14891,16 +13146,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14920,10 +13174,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -14945,7 +13195,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14956,7 +13205,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14968,11 +13217,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14983,7 +13231,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -14993,12 +13241,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15026,7 +13270,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15034,16 +13278,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15055,13 +13298,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15073,7 +13312,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15092,21 +13330,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15116,11 +13353,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15142,7 +13375,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15161,22 +13393,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15188,13 +13419,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15205,7 +13432,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup] @@ -15228,16 +13454,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15248,14 +13473,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15265,11 +13490,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15281,16 +13506,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup] @@ -15313,15 +13533,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15332,14 +13551,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15349,11 +13568,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15365,8 +13584,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -15393,7 +13610,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15419,7 +13636,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15428,16 +13645,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15457,10 +13673,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -15485,14 +13697,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15506,7 +13710,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15520,18 +13724,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15545,7 +13741,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -15557,19 +13753,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15603,7 +13788,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15613,24 +13798,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15656,7 +13839,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15665,16 +13848,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15686,13 +13868,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15718,14 +13896,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15739,7 +13909,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15753,18 +13923,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15778,7 +13940,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -15790,19 +13952,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15836,7 +13987,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15846,17 +13997,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -15875,22 +14024,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15900,11 +14048,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15918,14 +14062,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15953,7 +14089,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15962,17 +14098,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15996,7 +14128,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -16015,23 +14146,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16043,13 +14173,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -16062,7 +14188,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -16085,17 +14210,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16106,14 +14230,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -16123,11 +14247,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16139,10 +14263,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -16160,22 +14280,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16185,11 +14304,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -16206,14 +14321,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16241,7 +14348,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16250,17 +14357,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -16288,7 +14391,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16312,15 +14414,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16331,14 +14432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16348,11 +14449,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16364,8 +14465,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -16375,7 +14474,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16399,16 +14497,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16419,14 +14516,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16436,11 +14533,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16452,10 +14549,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -16477,7 +14570,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16488,7 +14580,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16500,11 +14592,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16515,7 +14606,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -16525,12 +14616,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16558,7 +14645,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16566,16 +14653,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16595,10 +14681,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -16620,7 +14702,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16631,7 +14712,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16643,11 +14724,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16658,7 +14738,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -16668,12 +14748,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16701,7 +14778,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16709,16 +14786,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16730,13 +14806,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -16748,7 +14820,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16767,21 +14838,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16791,11 +14861,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -16807,7 +14873,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16826,21 +14891,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16852,13 +14916,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -16872,7 +14932,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -16891,21 +14950,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16917,13 +14975,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -16947,7 +15001,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -16972,7 +15026,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -16980,16 +15034,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17009,10 +15062,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17034,14 +15083,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17052,7 +15093,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17064,18 +15105,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17083,7 +15116,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -17092,19 +15125,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17135,7 +15157,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17145,20 +15167,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17183,7 +15207,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17191,16 +15215,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17220,10 +15243,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -17245,14 +15264,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17263,7 +15274,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17275,18 +15286,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17294,7 +15297,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -17303,19 +15310,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17346,7 +15342,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17356,17 +15352,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -17385,21 +15379,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17409,11 +15402,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17423,14 +15412,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17457,7 +15438,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17466,17 +15447,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17490,7 +15467,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -17509,22 +15485,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17536,13 +15511,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -17550,7 +15521,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17572,16 +15542,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17592,14 +15561,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17609,11 +15578,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17625,16 +15594,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17656,15 +15620,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17675,14 +15638,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17692,11 +15655,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17708,8 +15671,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -17755,7 +15716,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17766,7 +15726,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17781,11 +15741,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17796,7 +15755,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -17808,12 +15767,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17841,7 +15796,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17849,16 +15804,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17878,10 +15832,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -17904,7 +15854,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17915,7 +15864,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17928,11 +15877,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17943,7 +15891,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -17953,12 +15901,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17986,7 +15931,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17994,16 +15939,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18015,13 +15959,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -18040,21 +15980,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18064,11 +16003,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18081,14 +16016,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18115,7 +16042,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18124,17 +16051,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18156,7 +16079,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18175,22 +16097,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18202,13 +16123,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -18216,7 +16133,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18240,16 +16156,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18260,14 +16175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18277,11 +16192,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18293,17 +16208,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18327,15 +16237,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18346,14 +16255,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18363,11 +16272,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18379,8 +16288,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -18407,7 +16314,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -18433,7 +16340,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -18442,16 +16349,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18463,13 +16369,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -18496,14 +16398,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -18516,7 +16410,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18531,18 +16425,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -18555,7 +16441,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -18568,19 +16454,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -18613,7 +16488,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18623,17 +16498,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18652,22 +16525,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18677,11 +16549,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18695,14 +16563,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18730,7 +16590,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18739,17 +16599,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18772,7 +16628,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -18791,23 +16646,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18819,13 +16673,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -18871,7 +16721,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18882,7 +16731,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18896,11 +16745,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18911,7 +16759,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -18923,12 +16771,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18956,7 +16800,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -18964,16 +16808,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18993,10 +16836,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -19020,7 +16859,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19031,7 +16869,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19045,11 +16883,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19060,7 +16897,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -19071,12 +16908,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19104,7 +16938,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19112,16 +16946,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19133,13 +16966,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -19154,7 +16983,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19173,21 +17001,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19197,11 +17024,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -19225,7 +17048,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19244,22 +17066,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19271,13 +17092,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -19298,7 +17115,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19322,17 +17138,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19343,14 +17158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19360,11 +17175,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19376,10 +17191,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -19393,7 +17204,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19417,16 +17227,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19437,14 +17246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19454,11 +17263,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19470,10 +17279,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -19500,7 +17305,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -19526,7 +17331,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19535,16 +17340,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19564,10 +17368,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -19594,14 +17394,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19615,7 +17407,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19631,18 +17423,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19656,7 +17440,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -19670,19 +17454,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19716,7 +17489,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19726,24 +17499,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -19769,7 +17540,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19778,16 +17549,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19801,13 +17571,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19833,14 +17599,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19854,7 +17612,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19868,18 +17626,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19893,7 +17643,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -19905,19 +17655,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19951,7 +17690,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19961,17 +17700,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -19994,22 +17731,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20019,11 +17755,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20043,14 +17775,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20078,7 +17802,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20087,17 +17811,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20122,7 +17842,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -20141,23 +17860,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20169,13 +17887,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -20188,7 +17902,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_01.models.CapacityReservation] @@ -20211,17 +17924,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20232,14 +17944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -20249,11 +17961,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20265,10 +17977,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -20291,7 +17999,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20316,23 +18024,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20346,13 +18053,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20374,14 +18077,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20391,7 +18086,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20400,18 +18095,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20421,7 +18108,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20429,19 +18116,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20473,7 +18150,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20486,20 +18163,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20524,23 +18199,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20554,13 +18228,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20581,14 +18251,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20598,7 +18260,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20606,18 +18268,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20627,7 +18281,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20635,19 +18289,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20679,7 +18322,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20692,17 +18335,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20731,7 +18372,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20755,16 +18395,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20775,14 +18414,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20792,11 +18431,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20808,10 +18447,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20820,7 +18455,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20839,21 +18473,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20865,20 +18498,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20904,7 +18533,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20913,16 +18542,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20942,10 +18570,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20972,14 +18596,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20993,7 +18609,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21009,18 +18625,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21034,7 +18642,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21047,19 +18655,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21093,7 +18690,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21103,24 +18700,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21146,7 +18741,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21155,16 +18750,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21176,13 +18770,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21210,14 +18800,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21231,7 +18813,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21247,18 +18829,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21272,7 +18846,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21285,19 +18859,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21331,7 +18895,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21341,17 +18905,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21370,22 +18932,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21395,11 +18956,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21414,14 +18971,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21449,7 +18998,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21458,17 +19007,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21490,7 +19035,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21509,23 +19053,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21537,13 +19080,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21557,7 +19096,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21581,18 +19119,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21603,14 +19140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21620,11 +19157,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21636,12 +19173,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21667,7 +19200,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21693,7 +19226,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21703,16 +19236,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21732,10 +19264,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21764,14 +19292,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21786,7 +19306,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21803,18 +19323,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21829,7 +19341,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21843,19 +19355,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21890,7 +19391,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21900,17 +19401,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21918,7 +19417,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21944,7 +19443,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21954,16 +19453,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21975,13 +19473,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22011,14 +19505,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22033,7 +19519,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22050,18 +19536,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22076,7 +19554,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22090,19 +19568,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22137,7 +19605,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22147,17 +19615,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22176,23 +19642,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22202,11 +19667,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22222,14 +19683,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22258,7 +19711,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22267,17 +19720,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22302,7 +19751,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22321,7 +19769,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22329,16 +19777,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22350,13 +19797,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22378,7 +19821,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22402,19 +19844,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22425,14 +19866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22442,11 +19883,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22457,7 +19898,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_compute_management_client_enums.py index 2dbc143ab97a..5687a9f647b1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_compute_management_client_enums.py @@ -389,7 +389,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_models_py3.py index 5863edc0cbb5..291780c6aa6f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -265,7 +261,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -296,7 +292,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -316,7 +312,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -399,7 +395,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -456,7 +452,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -555,7 +551,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet] @@ -868,7 +864,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -986,7 +982,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1088,7 +1084,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup] @@ -1248,7 +1244,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.CapacityReservation] @@ -1483,7 +1479,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1694,7 +1690,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1877,7 +1873,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2007,7 +2003,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup] @@ -2040,7 +2036,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group.:code:`
    `:code:`
    `Minimum api-version: 2022-03-01. @@ -2260,7 +2256,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -2518,17 +2514,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2849,7 +2834,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3029,7 +3014,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_03_01.models.SubResource @@ -3137,7 +3122,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.Image] @@ -3171,7 +3156,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_03_01.models.SubResource @@ -3313,7 +3298,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3372,7 +3357,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3594,7 +3579,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3627,7 +3612,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -3924,7 +3909,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -3957,7 +3942,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.Usage] @@ -3991,7 +3976,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4345,7 +4330,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4421,7 +4406,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -4583,7 +4568,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -4982,7 +4967,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5093,7 +5078,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup] @@ -5153,17 +5138,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5204,7 +5178,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2022_03_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2022_03_01.models.PublicIPAddressSkuTier """ @@ -5223,8 +5197,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2022_03_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2022_03_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5235,7 +5208,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5306,7 +5279,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5539,7 +5512,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6248,7 +6221,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6311,7 +6284,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6374,7 +6347,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6448,7 +6421,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6495,7 +6468,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6528,7 +6501,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.RunCommandDocumentBase] @@ -6564,7 +6537,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6909,7 +6882,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -6919,7 +6892,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -6948,7 +6921,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -6963,7 +6936,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7019,7 +6992,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource] @@ -7235,7 +7208,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7256,67 +7229,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7385,7 +7297,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7528,7 +7440,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -7746,7 +7658,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8238,7 +8150,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8477,7 +8389,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8521,7 +8433,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8836,7 +8748,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_01.models.UserAssignedIdentitiesValue] """ @@ -8868,7 +8780,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2022_03_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_01.models.UserAssignedIdentitiesValue] """ @@ -8882,7 +8794,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8945,7 +8857,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9099,7 +9011,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9424,7 +9336,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -9457,10 +9369,10 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -9567,7 +9479,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9587,10 +9499,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -9756,10 +9668,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -9859,10 +9771,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -9917,7 +9829,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10164,7 +10076,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand] @@ -10313,7 +10225,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10564,7 +10476,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10842,10 +10754,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension] @@ -11105,7 +11017,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_01.models.UserAssignedIdentitiesValue] """ @@ -11139,7 +11051,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_01.models.UserAssignedIdentitiesValue] """ @@ -11193,7 +11105,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11220,7 +11132,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11375,10 +11287,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -11420,7 +11332,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet] @@ -11456,7 +11368,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetSku] @@ -11492,7 +11404,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet] @@ -11525,7 +11437,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -11576,10 +11488,10 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(SubResource): +class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11682,7 +11594,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -11707,7 +11619,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -11738,7 +11650,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11757,7 +11669,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12050,10 +11962,10 @@ def __init__( self.allow_extension_operations = allow_extension_operations -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12142,10 +12054,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12172,7 +12086,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12180,18 +12094,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -12500,7 +12402,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -12618,7 +12520,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): +class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -12719,12 +12621,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12757,7 +12659,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12913,7 +12815,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -12973,7 +12875,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -13121,7 +13023,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13520,7 +13422,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -13543,7 +13445,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -13726,10 +13628,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -13873,7 +13775,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM] @@ -13906,7 +13808,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -14759,14 +14661,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -14811,7 +14713,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -14867,7 +14769,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -15087,7 +14989,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/operations/_operations.py index 2dc38f6a3127..b82cec2dfc22 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -176,7 +176,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -210,7 +210,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -248,7 +248,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -286,7 +286,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -320,7 +320,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -361,7 +361,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -392,7 +392,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -422,7 +422,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -448,7 +450,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -479,7 +481,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -510,7 +512,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -551,7 +553,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -585,7 +587,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -619,7 +621,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -653,7 +655,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -687,7 +689,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -721,7 +723,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -755,7 +757,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -789,7 +791,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -832,7 +834,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -866,7 +868,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -900,7 +902,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -935,7 +937,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -970,7 +972,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1002,7 +1004,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1042,7 +1044,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1073,7 +1075,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1104,7 +1106,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1137,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1168,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1197,7 +1199,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1238,7 +1240,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1279,7 +1281,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1317,7 +1319,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1359,7 +1361,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1399,7 +1401,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1434,7 +1436,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1466,7 +1468,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1498,7 +1500,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1533,7 +1535,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1573,7 +1575,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1613,7 +1615,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1645,7 +1647,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1691,7 +1693,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1731,7 +1733,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1763,7 +1765,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1795,7 +1797,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1827,7 +1829,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1869,7 +1871,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1901,7 +1903,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1933,7 +1935,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1968,7 +1970,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2003,7 +2005,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2038,7 +2040,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2070,7 +2072,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2110,7 +2112,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2143,7 +2145,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2205,7 +2209,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2349,7 +2353,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2380,7 +2384,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2411,7 +2415,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2444,7 +2448,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2537,7 +2541,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2759,7 +2763,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2799,7 +2803,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2830,7 +2834,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2861,7 +2865,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2892,7 +2896,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2926,7 +2930,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2994,7 +2998,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3042,7 +3046,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3073,7 +3077,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3102,7 +3106,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3134,7 +3138,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3165,7 +3169,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3207,7 +3211,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3257,7 +3261,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3289,7 +3293,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3320,7 +3324,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3353,7 +3357,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3386,7 +3390,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3417,7 +3421,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3463,7 +3467,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3593,7 +3597,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3649,7 +3653,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3680,7 +3684,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3716,7 +3720,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3752,7 +3756,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3785,7 +3789,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -3827,7 +3831,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3853,7 +3859,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3883,7 +3889,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3917,7 +3923,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3951,7 +3957,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4020,7 +4026,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4050,7 +4056,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4074,7 +4082,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4216,7 +4224,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4279,7 +4287,9 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4303,7 +4313,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4463,7 +4473,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4626,7 +4636,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4680,7 +4690,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4716,7 +4726,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4752,7 +4762,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4785,7 +4795,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4825,7 +4835,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4855,7 +4865,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5002,7 +5014,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5038,7 +5050,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5074,7 +5086,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5107,7 +5119,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5147,7 +5159,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5183,7 +5195,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5216,7 +5228,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5257,7 +5269,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5298,7 +5310,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5378,7 +5390,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5411,7 +5423,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5444,7 +5456,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5477,7 +5489,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5504,7 +5518,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5535,7 +5549,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5570,7 +5584,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5605,7 +5619,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5637,7 +5651,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5677,7 +5691,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5710,7 +5724,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5751,7 +5765,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5792,7 +5806,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5830,7 +5844,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5872,7 +5886,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5936,7 +5950,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5960,14 +5973,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5978,14 +5990,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5995,11 +6007,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6011,8 +6023,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6041,7 +6051,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6063,16 +6072,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6083,14 +6091,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6100,11 +6108,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6116,8 +6124,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6146,7 +6152,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6168,16 +6173,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6188,14 +6192,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6205,11 +6209,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6221,8 +6225,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6251,7 +6253,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6275,16 +6276,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6295,14 +6295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6312,11 +6312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6328,15 +6328,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6362,7 +6358,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6370,16 +6366,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6399,10 +6394,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6424,14 +6415,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6444,7 +6427,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6456,18 +6439,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6480,7 +6455,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6489,20 +6464,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6535,7 +6499,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6545,23 +6509,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6587,7 +6549,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6595,16 +6557,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6616,13 +6577,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6645,14 +6602,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6665,7 +6614,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6677,18 +6626,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6701,7 +6642,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6710,20 +6651,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6756,7 +6687,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6766,17 +6697,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6795,22 +6724,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6820,11 +6748,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6839,14 +6763,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6874,7 +6790,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6883,17 +6799,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6914,7 +6826,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6933,22 +6844,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6960,19 +6870,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7001,7 +6907,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7009,16 +6915,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7028,11 +6933,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7059,14 +6960,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7077,7 +6970,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7092,18 +6985,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7114,7 +6999,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7126,20 +7011,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7169,7 +7043,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7178,23 +7052,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7222,7 +7092,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7231,16 +7101,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7250,11 +7119,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7283,14 +7148,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7301,7 +7158,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7315,21 +7172,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7340,7 +7189,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7352,23 +7201,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7399,7 +7237,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7408,17 +7246,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7430,7 +7264,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7449,21 +7282,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7475,13 +7307,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7489,7 +7317,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7513,16 +7340,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7533,14 +7359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7550,11 +7376,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7566,17 +7392,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7600,15 +7421,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7619,14 +7439,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7636,11 +7456,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7652,8 +7472,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7665,7 +7483,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7689,17 +7506,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7710,14 +7526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7727,11 +7543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7743,10 +7559,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7757,7 +7569,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7781,17 +7592,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7802,14 +7612,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7819,11 +7629,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7835,15 +7645,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -7874,7 +7680,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7883,16 +7689,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7902,11 +7707,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -7938,14 +7739,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7956,7 +7749,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -7972,7 +7765,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -7980,14 +7773,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7998,7 +7783,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8012,24 +7797,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8060,7 +7834,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8069,23 +7843,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8114,7 +7884,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8122,16 +7892,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8141,11 +7910,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8170,14 +7935,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8188,7 +7945,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8201,18 +7958,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8223,7 +7972,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8233,20 +7982,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8276,7 +8014,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8285,23 +8023,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8330,7 +8064,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8338,16 +8072,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8357,11 +8090,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8386,14 +8115,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8404,7 +8125,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8417,18 +8138,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8439,7 +8152,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8449,20 +8162,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8492,7 +8194,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8501,23 +8203,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8546,7 +8244,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8554,16 +8252,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8573,11 +8270,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -8603,14 +8296,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8621,7 +8306,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8635,18 +8320,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8657,7 +8334,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8668,20 +8345,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8711,7 +8377,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8720,23 +8386,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8765,7 +8427,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8773,16 +8435,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8792,11 +8453,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -8824,14 +8481,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8842,7 +8491,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8858,18 +8507,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8880,7 +8521,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8893,20 +8534,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8936,7 +8566,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8945,23 +8575,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8987,7 +8613,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8995,16 +8621,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9014,11 +8639,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9043,14 +8664,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9061,7 +8674,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9074,18 +8687,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9096,7 +8701,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9106,20 +8711,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9149,7 +8743,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9158,23 +8752,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9203,7 +8793,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9211,16 +8801,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9230,11 +8819,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9260,14 +8845,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9278,7 +8855,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9292,18 +8869,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9314,7 +8883,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9326,20 +8895,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9369,7 +8927,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9378,23 +8936,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9423,7 +8977,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9431,16 +8985,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9450,11 +9003,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9480,14 +9029,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9498,7 +9039,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9512,18 +9053,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9534,7 +9067,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9545,20 +9078,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9588,7 +9110,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9597,20 +9119,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -9636,7 +9154,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9655,7 +9172,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9663,16 +9180,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9684,13 +9200,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -9715,7 +9227,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9726,7 +9237,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9739,11 +9250,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9754,7 +9264,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -9765,13 +9275,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9799,7 +9306,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9807,16 +9314,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9826,17 +9332,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9862,7 +9364,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9870,16 +9372,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9889,11 +9390,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -9917,14 +9414,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9935,7 +9424,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9948,18 +9437,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9970,7 +9451,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -9981,19 +9462,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10023,7 +9494,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10032,20 +9503,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10070,7 +9537,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10096,7 +9563,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10105,16 +9572,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10134,10 +9600,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10165,14 +9627,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10186,7 +9640,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10202,18 +9656,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10227,7 +9673,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10240,20 +9686,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10287,7 +9722,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10297,24 +9732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10340,7 +9773,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10349,16 +9782,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10378,10 +9810,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -10409,14 +9837,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10430,7 +9850,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10446,18 +9866,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10471,7 +9883,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -10484,20 +9896,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10531,7 +9933,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10541,17 +9943,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10570,22 +9970,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10595,11 +9994,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10614,14 +10009,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10649,7 +10036,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10658,17 +10045,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10690,7 +10073,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10709,23 +10091,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10737,13 +10118,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10755,7 +10132,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10779,17 +10155,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10800,14 +10175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10817,11 +10192,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10833,12 +10208,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10875,21 +10246,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10899,11 +10269,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10913,14 +10279,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10947,7 +10305,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10956,17 +10314,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10985,21 +10339,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11009,11 +10362,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11027,14 +10376,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11061,7 +10402,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11070,17 +10411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11099,21 +10436,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11123,11 +10459,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11141,14 +10473,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11175,7 +10499,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11184,17 +10508,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11206,7 +10526,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11225,21 +10544,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11251,16 +10569,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11286,7 +10600,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11312,7 +10626,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11322,16 +10636,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11351,10 +10664,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11384,14 +10693,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11406,7 +10707,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11423,18 +10724,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11449,7 +10742,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -11463,20 +10756,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11511,7 +10793,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11521,17 +10803,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -11539,7 +10819,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11565,7 +10845,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11575,16 +10855,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11596,13 +10875,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11633,14 +10908,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11655,7 +10922,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11672,18 +10939,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11698,7 +10957,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -11712,20 +10971,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11760,7 +11009,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11770,17 +11019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -11799,23 +11046,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11825,11 +11071,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11845,14 +11087,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11881,7 +11115,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11890,17 +11124,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11925,7 +11155,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11944,7 +11173,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11952,16 +11181,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11973,13 +11201,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12001,7 +11225,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12020,23 +11243,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12048,13 +11270,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12082,7 +11300,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12111,7 +11331,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12120,16 +11340,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12139,11 +11358,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12171,14 +11386,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12190,7 +11397,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12205,18 +11412,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12228,7 +11427,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12240,20 +11441,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12284,7 +11475,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12293,17 +11484,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12322,22 +11509,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12347,11 +11533,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -12366,14 +11548,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12401,7 +11575,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12410,17 +11584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12439,22 +11609,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12464,11 +11633,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12484,14 +11649,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12519,7 +11676,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12528,24 +11685,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12571,7 +11724,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12580,16 +11733,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12609,10 +11761,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -12639,14 +11787,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12660,7 +11800,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12676,18 +11816,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12701,7 +11833,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12714,19 +11846,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12760,7 +11881,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12770,17 +11891,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -12805,23 +11924,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12831,11 +11949,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12858,14 +11972,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12894,7 +12000,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12903,17 +12009,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12937,7 +12039,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -12956,23 +12057,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12984,13 +12084,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13004,7 +12100,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13023,22 +12118,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13050,13 +12144,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13085,7 +12175,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13109,7 +12198,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13117,12 +12206,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13133,14 +12221,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13150,11 +12238,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13166,10 +12254,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13193,23 +12277,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13219,11 +12302,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13249,14 +12328,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13285,7 +12356,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13294,17 +12365,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13323,22 +12390,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13348,11 +12414,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -13366,14 +12428,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13401,7 +12455,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13410,17 +12464,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13439,22 +12489,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13464,11 +12513,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -13482,14 +12527,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13517,7 +12554,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13526,17 +12563,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13555,22 +12588,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13580,11 +12612,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -13599,14 +12627,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13634,7 +12654,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13643,17 +12663,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13679,7 +12695,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13698,23 +12713,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13726,13 +12740,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13751,22 +12761,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13776,11 +12785,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -13794,14 +12799,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13829,7 +12826,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13838,17 +12835,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -13862,7 +12855,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13881,22 +12873,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13906,18 +12897,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -13943,7 +12930,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13952,16 +12939,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13975,13 +12961,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14007,14 +12989,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -14027,7 +13001,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14041,18 +13015,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -14065,7 +13031,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14077,19 +13043,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -14122,7 +13077,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14134,17 +13089,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -14172,7 +13125,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14198,7 +13151,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14207,16 +13160,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14236,10 +13188,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -14266,14 +13214,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14287,7 +13227,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14303,18 +13243,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14328,7 +13260,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -14341,20 +13273,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14388,7 +13309,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14398,24 +13319,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14441,7 +13360,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14450,16 +13369,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14471,13 +13389,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14506,14 +13420,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14527,7 +13433,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14543,18 +13449,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14568,7 +13466,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -14581,20 +13479,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14628,7 +13515,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14638,17 +13525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14667,22 +13552,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14692,11 +13576,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14711,14 +13591,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14746,7 +13618,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14755,17 +13627,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14787,7 +13655,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14806,23 +13673,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14834,13 +13700,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14854,7 +13716,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14873,22 +13734,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14900,13 +13760,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -14936,7 +13792,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -14958,16 +13813,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14978,14 +13832,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -14995,11 +13849,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15011,15 +13865,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15045,7 +13895,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15053,16 +13903,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15076,13 +13925,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15106,14 +13951,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15126,7 +13963,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15139,18 +13976,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15163,7 +13992,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15174,19 +14003,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15219,7 +14038,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15231,20 +14050,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15269,7 +14090,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15277,16 +14098,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15306,10 +14126,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15332,14 +14148,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15351,7 +14159,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15364,18 +14172,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15384,7 +14184,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15394,19 +14198,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15438,7 +14231,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15448,20 +14241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15486,7 +14281,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15494,16 +14289,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15515,13 +14309,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15544,14 +14334,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15563,7 +14345,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15575,18 +14357,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15595,7 +14369,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15604,19 +14382,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] @@ -15648,7 +14415,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15658,17 +14425,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15687,22 +14452,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15712,11 +14476,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15731,14 +14491,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15766,7 +14518,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15775,17 +14527,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15808,7 +14556,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -15827,22 +14574,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15854,13 +14600,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -15872,7 +14614,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15891,21 +14632,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15917,13 +14657,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -15942,21 +14678,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15966,11 +14701,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -15981,14 +14712,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16015,7 +14738,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16024,17 +14747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -16053,22 +14772,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16078,11 +14796,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -16098,14 +14812,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16133,7 +14839,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16142,17 +14848,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16169,7 +14871,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16188,21 +14889,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16212,11 +14912,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -16232,7 +14928,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16254,17 +14949,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16275,14 +14969,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16292,11 +14986,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16308,10 +15002,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -16327,7 +15017,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16349,17 +15038,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16370,14 +15058,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16387,11 +15075,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16403,8 +15091,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16416,7 +15102,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16438,17 +15123,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16459,14 +15143,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16476,11 +15160,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16492,10 +15176,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16513,22 +15193,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16538,11 +15217,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16559,14 +15234,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16594,7 +15261,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16603,17 +15270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16632,21 +15295,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16656,11 +15318,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16670,14 +15328,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16704,7 +15354,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16713,17 +15363,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16742,21 +15388,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16766,11 +15411,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16780,14 +15421,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16814,7 +15447,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16823,17 +15456,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16852,21 +15481,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16876,11 +15504,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16890,14 +15514,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16924,7 +15540,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16933,17 +15549,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16962,21 +15574,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16986,11 +15597,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17000,14 +15607,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17034,7 +15633,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17043,23 +15642,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17088,7 +15683,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17096,16 +15691,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17115,11 +15709,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17143,14 +15733,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17161,7 +15743,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17174,18 +15756,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17196,7 +15770,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -17206,19 +15780,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17248,7 +15812,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17257,17 +15821,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17289,7 +15849,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17308,22 +15867,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17335,13 +15893,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17360,21 +15914,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17384,11 +15937,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17398,14 +15947,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17432,7 +15973,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17441,17 +15982,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17463,7 +16000,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17482,21 +16018,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17506,11 +16041,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17529,21 +16060,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17557,13 +16087,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17575,14 +16101,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17612,7 +16130,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17624,23 +16142,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17666,7 +16182,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17674,16 +16190,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17697,13 +16212,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17728,14 +16239,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17748,7 +16251,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17761,18 +16264,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17785,7 +16280,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -17795,20 +16290,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17841,7 +16325,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17853,20 +16337,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -17891,7 +16377,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17899,16 +16385,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17922,13 +16407,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -17951,14 +16432,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -17970,7 +16443,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17982,18 +16455,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -18002,7 +16467,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18011,19 +16480,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RunCommandResult] @@ -18055,7 +16513,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18067,17 +16525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -18116,7 +16572,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18135,7 +16590,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18143,16 +16598,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18164,13 +16618,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18202,7 +16652,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18221,7 +16670,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18231,16 +16680,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18252,13 +16700,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18270,7 +16714,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18289,21 +16732,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18315,13 +16757,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -18329,7 +16767,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18348,20 +16785,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18373,13 +16809,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18393,7 +16825,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18412,22 +16843,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18439,13 +16869,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -18455,7 +16881,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18474,21 +16899,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18500,13 +16924,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -18547,7 +16967,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18566,7 +16985,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18575,16 +16994,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18596,13 +17014,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18639,7 +17053,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18658,7 +17071,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18669,16 +17082,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18690,13 +17102,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18711,7 +17119,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18730,22 +17137,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18757,13 +17163,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -18775,7 +17177,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18794,21 +17195,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18820,13 +17220,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18843,7 +17239,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18862,23 +17257,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18890,13 +17284,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -18933,7 +17323,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18952,23 +17341,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18980,13 +17368,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -18998,7 +17382,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19017,21 +17400,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19043,13 +17425,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -19077,7 +17455,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19096,7 +17473,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -19105,16 +17482,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19126,13 +17502,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -19176,7 +17548,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19187,7 +17558,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19199,11 +17570,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19214,7 +17584,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -19224,12 +17594,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19257,7 +17623,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19265,16 +17631,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19286,13 +17651,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -19315,7 +17676,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19326,7 +17686,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19338,11 +17698,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19353,7 +17712,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -19363,12 +17722,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19396,7 +17751,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19404,16 +17759,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19425,13 +17779,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -19443,7 +17793,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19462,21 +17811,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19486,11 +17834,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -19500,7 +17844,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19519,21 +17862,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19545,13 +17887,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -19562,7 +17900,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19584,16 +17921,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19604,14 +17940,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19621,11 +17957,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19637,17 +17973,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19669,16 +18000,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19689,14 +18019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19706,11 +18036,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19722,10 +18052,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -19737,7 +18063,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -19759,17 +18084,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19780,14 +18104,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -19797,11 +18121,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19813,10 +18137,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -19860,7 +18180,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19871,7 +18190,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19884,11 +18203,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19899,7 +18217,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -19909,12 +18227,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19942,7 +18256,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -19950,16 +18264,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19979,10 +18292,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -20005,7 +18314,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20016,7 +18324,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20029,11 +18337,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20044,7 +18351,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -20054,12 +18361,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20087,7 +18391,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20095,16 +18399,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20116,13 +18419,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20134,7 +18433,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20153,21 +18451,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20177,11 +18474,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20201,7 +18494,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20220,22 +18512,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20247,19 +18538,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20283,15 +18569,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20302,14 +18587,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20319,11 +18604,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20335,10 +18620,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -20347,7 +18628,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20371,16 +18651,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20391,14 +18670,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20408,11 +18687,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20424,10 +18703,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -20472,7 +18747,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20483,7 +18757,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20497,11 +18771,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20512,7 +18785,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -20524,12 +18797,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20557,7 +18826,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20565,16 +18834,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20594,10 +18862,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -20619,7 +18883,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20630,7 +18893,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20642,11 +18905,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20657,7 +18919,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -20666,13 +18928,9 @@ def update( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20700,7 +18958,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20708,16 +18966,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20729,13 +18986,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20747,7 +19000,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20766,21 +19018,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20790,11 +19041,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20816,7 +19063,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20835,22 +19081,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20862,13 +19107,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -20877,7 +19118,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20899,16 +19139,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20919,14 +19158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -20936,11 +19175,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20952,16 +19191,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20983,15 +19217,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21002,14 +19235,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21019,11 +19252,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21035,8 +19268,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -21063,7 +19294,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21089,7 +19320,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21098,16 +19329,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21127,10 +19357,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -21155,14 +19381,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21175,7 +19393,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21189,18 +19407,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21213,7 +19423,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -21225,19 +19435,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21270,7 +19469,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21280,24 +19479,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21323,7 +19520,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21332,16 +19529,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21353,13 +19549,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21385,14 +19577,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21405,7 +19589,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21419,18 +19603,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21443,7 +19619,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -21455,19 +19631,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] @@ -21500,7 +19665,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21510,17 +19675,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -21539,22 +19702,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21564,11 +19726,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21582,14 +19740,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21617,7 +19767,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21626,17 +19776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21660,7 +19806,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -21679,23 +19824,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21707,13 +19851,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -21726,7 +19866,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -21748,17 +19887,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21769,14 +19907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -21786,11 +19924,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21802,10 +19940,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -21823,22 +19957,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21848,11 +19981,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -21869,14 +19998,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21904,7 +20025,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21913,17 +20034,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -21951,7 +20068,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -21975,15 +20091,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21994,14 +20109,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22011,11 +20126,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22027,8 +20142,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -22038,7 +20151,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22062,16 +20174,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22082,14 +20193,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22099,11 +20210,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22115,10 +20226,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -22140,7 +20247,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22151,7 +20257,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22163,11 +20269,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22178,7 +20283,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -22188,12 +20293,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22221,7 +20322,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22229,16 +20330,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22258,10 +20358,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -22283,7 +20379,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22294,7 +20389,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22306,11 +20401,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22321,7 +20415,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -22331,12 +20425,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22364,7 +20455,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22372,16 +20463,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22393,13 +20483,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22411,7 +20497,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22430,21 +20515,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22454,11 +20538,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -22468,7 +20548,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22487,21 +20566,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22513,13 +20591,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -22533,7 +20607,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22552,21 +20625,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22578,13 +20650,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -22608,7 +20676,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22633,7 +20701,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22641,16 +20709,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22670,10 +20737,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -22695,14 +20758,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22713,7 +20768,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22725,18 +20780,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22744,7 +20791,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -22753,19 +20800,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22796,7 +20832,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22806,20 +20842,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22844,7 +20882,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22852,16 +20890,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22881,10 +20918,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -22906,14 +20939,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22924,7 +20949,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22936,18 +20961,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22955,7 +20972,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -22964,19 +20985,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23007,7 +21017,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23017,17 +21027,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -23046,21 +21054,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23070,11 +21077,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23084,14 +21087,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23118,7 +21113,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23127,17 +21122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23151,7 +21142,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -23170,22 +21160,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23197,13 +21186,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -23211,7 +21196,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23233,16 +21217,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23253,14 +21236,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23270,11 +21253,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23286,16 +21269,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23317,15 +21295,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23336,14 +21313,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23353,11 +21330,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23369,8 +21346,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23416,7 +21391,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23427,7 +21401,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23442,11 +21416,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23457,7 +21430,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23469,12 +21442,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23502,7 +21471,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23510,16 +21479,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23539,10 +21507,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23565,7 +21529,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23576,7 +21539,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23589,11 +21552,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23604,7 +21566,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23614,12 +21576,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23647,7 +21606,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23655,16 +21614,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23676,13 +21634,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -23701,21 +21655,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23725,11 +21678,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23742,14 +21691,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23776,7 +21717,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23785,17 +21726,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23817,7 +21754,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23836,22 +21772,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23863,13 +21798,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -23877,7 +21808,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23901,16 +21831,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23921,14 +21850,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -23938,11 +21867,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23954,17 +21883,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23988,15 +21912,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24007,14 +21930,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24024,11 +21947,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24040,8 +21963,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -24068,7 +21989,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -24094,7 +22015,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -24103,16 +22024,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24124,13 +22044,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -24157,14 +22073,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -24177,7 +22085,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24192,18 +22100,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -24216,7 +22116,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -24229,19 +22129,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_01.models.RestorePoint] @@ -24274,7 +22163,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24284,17 +22173,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -24313,22 +22200,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24338,11 +22224,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24356,14 +22238,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24391,7 +22265,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24400,17 +22274,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24433,7 +22303,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24452,23 +22321,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24480,13 +22348,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -24532,7 +22396,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24543,7 +22406,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24557,11 +22420,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24572,7 +22434,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -24584,12 +22446,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24617,7 +22475,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24625,16 +22483,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24654,10 +22511,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -24681,7 +22534,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24692,7 +22544,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24706,11 +22558,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24721,7 +22572,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -24732,12 +22583,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24765,7 +22613,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24773,16 +22621,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24794,13 +22641,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -24815,7 +22658,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -24834,21 +22676,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24858,11 +22699,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -24886,7 +22723,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24905,22 +22741,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24932,13 +22767,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -24959,7 +22790,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -24983,17 +22813,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25004,14 +22833,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25021,11 +22850,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25037,10 +22866,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -25054,7 +22879,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25078,16 +22902,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25098,14 +22921,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25115,11 +22938,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25131,10 +22954,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -25161,7 +22980,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -25187,7 +23006,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25196,16 +23015,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25225,10 +23043,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -25255,14 +23069,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25276,7 +23082,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25292,18 +23098,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25317,7 +23115,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -25331,19 +23129,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25377,7 +23164,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25387,24 +23174,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -25430,7 +23215,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25439,16 +23224,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25462,13 +23246,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25494,14 +23274,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25515,7 +23287,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25529,18 +23301,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25554,7 +23318,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -25566,19 +23330,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25612,7 +23365,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25622,17 +23375,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25655,22 +23406,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25680,11 +23430,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25704,14 +23450,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25739,7 +23477,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25748,17 +23486,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25783,7 +23517,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -25802,23 +23535,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25830,13 +23562,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -25849,7 +23577,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_01.models.CapacityReservation] @@ -25872,17 +23599,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25893,14 +23619,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -25910,11 +23636,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25926,10 +23652,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -25952,7 +23674,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -25977,23 +23699,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26007,13 +23728,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26035,14 +23752,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26052,7 +23761,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26061,18 +23770,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26082,7 +23783,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26090,19 +23791,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26134,7 +23825,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26146,20 +23837,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26184,23 +23873,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26214,13 +23902,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26241,14 +23925,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26258,7 +23934,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26266,18 +23942,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26287,7 +23955,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26295,19 +23963,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26339,7 +23996,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26351,17 +24008,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26390,7 +24045,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26414,16 +24068,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26434,14 +24087,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26451,11 +24104,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26467,10 +24120,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26479,7 +24128,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26498,21 +24146,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26524,20 +24171,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26563,7 +24206,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26572,16 +24215,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26601,10 +24243,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26631,14 +24269,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26652,7 +24282,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26668,18 +24298,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26693,7 +24315,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -26706,19 +24328,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26752,7 +24363,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26762,24 +24373,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26805,7 +24414,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26814,16 +24423,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26835,13 +24443,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26869,14 +24473,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26890,7 +24486,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26906,18 +24502,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26931,7 +24519,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -26944,19 +24532,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26990,7 +24568,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27000,17 +24578,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27029,22 +24605,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27054,11 +24629,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27073,14 +24644,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27108,7 +24671,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27117,17 +24680,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27149,7 +24708,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27168,23 +24726,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27196,13 +24753,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27216,7 +24769,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27240,18 +24792,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27262,14 +24813,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27279,11 +24830,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27295,12 +24846,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27326,7 +24873,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27352,7 +24899,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27362,16 +24909,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27391,10 +24937,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27423,14 +24965,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27445,7 +24979,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27462,18 +24996,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27488,7 +25014,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27502,19 +25028,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27549,7 +25064,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27559,17 +25074,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27577,7 +25090,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27603,7 +25116,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27613,16 +25126,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27634,13 +25146,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27670,14 +25178,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27692,7 +25192,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27709,18 +25209,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27735,7 +25227,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -27749,19 +25241,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27796,7 +25278,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27806,17 +25288,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -27835,23 +25315,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27861,11 +25340,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27881,14 +25356,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27917,7 +25384,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27926,17 +25393,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27961,7 +25424,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27980,7 +25442,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27988,16 +25450,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28009,13 +25470,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28037,7 +25494,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28061,19 +25517,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28084,14 +25539,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28101,11 +25556,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28116,7 +25571,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_compute_management_client.py index 1bcd1f443e37..4ba402e07c25 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_configuration.py index 6742f81ccebf..650b33b7b01d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_metadata.json index 0dc25a036b2b..d17e26b6d740 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_compute_management_client.py index 58f38ed312d4..e881a4a3faf4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_configuration.py index 40602de9461f..ee21bd5b5f7a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/operations/_operations.py index 1d0a70cc25aa..7b164ccce521 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_03_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_03_02.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_03_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_03_02.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -1238,7 +1086,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1115,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1123,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1151,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -1328,14 +1175,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1347,7 +1186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1362,18 +1201,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1382,7 +1213,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -1393,19 +1228,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1437,7 +1261,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1447,23 +1271,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -1489,7 +1311,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1497,16 +1319,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1526,10 +1347,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -1554,14 +1371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1573,7 +1382,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1588,18 +1397,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1611,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -1623,19 +1424,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1667,7 +1457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1677,17 +1467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -1699,7 +1487,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,21 +1505,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1744,13 +1530,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -1769,21 +1551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,11 +1574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1811,14 +1588,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1845,7 +1614,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1854,17 +1623,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -1872,7 +1637,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1895,16 +1659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1915,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -1932,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1948,15 +1711,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -1979,15 +1737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1998,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -2015,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2031,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2045,7 +1800,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2064,21 +1818,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2090,20 +1843,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -2129,7 +1878,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -2138,16 +1887,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2167,12 +1915,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -2200,14 +1944,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2216,12 +1952,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2239,18 +1975,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2259,12 +1987,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -2280,20 +2008,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2327,7 +2044,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2337,17 +2054,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -2363,7 +2078,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -2382,22 +2096,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2409,15 +2122,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -2434,22 +2143,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2459,14 +2167,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -2479,14 +2183,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2514,7 +2210,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2523,17 +2219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -2547,7 +2239,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2571,17 +2262,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2592,14 +2282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -2609,11 +2299,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2625,10 +2315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -2654,7 +2340,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2680,7 +2366,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2374,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2717,10 +2402,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2745,14 +2426,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2765,7 +2438,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2780,18 +2453,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2804,7 +2469,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2816,19 +2481,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2861,7 +2516,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2871,23 +2526,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2913,7 +2566,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2921,16 +2574,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2950,10 +2602,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2978,14 +2626,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2998,7 +2638,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3013,18 +2653,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3037,7 +2669,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3049,19 +2681,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3094,7 +2717,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3104,17 +2727,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3128,7 +2749,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3147,21 +2767,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2792,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3198,21 +2813,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,11 +2836,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3240,14 +2850,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3274,7 +2876,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3283,17 +2885,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3303,7 +2901,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -3326,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3346,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3363,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3379,15 +2975,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -3410,15 +3001,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3429,14 +3019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3446,11 +3036,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3462,8 +3052,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3476,7 +3064,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3498,17 +3085,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3519,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3536,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3552,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -3598,7 +3180,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -3617,23 +3198,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3645,13 +3225,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -3667,7 +3243,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskRestorePoint] @@ -3690,18 +3265,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3712,14 +3286,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -3729,11 +3303,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3745,17 +3319,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3781,7 +3351,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -3791,16 +3361,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3383,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -3852,14 +3417,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -3873,7 +3430,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3892,18 +3449,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -3917,7 +3466,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -3933,19 +3482,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -3979,7 +3517,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3991,17 +3529,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -4025,23 +3561,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4051,11 +3586,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -4078,14 +3609,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4114,7 +3637,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -4125,17 +3648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -4159,7 +3678,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4184,7 +3703,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4192,16 +3711,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4221,10 +3739,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -4248,14 +3762,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4267,7 +3773,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4281,18 +3787,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4301,7 +3799,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -4312,19 +3810,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4356,7 +3843,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4366,20 +3853,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4404,7 +3893,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4412,16 +3901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4441,10 +3929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -4469,14 +3953,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4488,7 +3964,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4503,18 +3979,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4523,7 +3991,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -4534,19 +4006,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4578,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4588,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -4610,7 +4069,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -4629,21 +4087,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4655,13 +4112,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4680,21 +4133,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4704,11 +4156,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4720,14 +4168,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4754,7 +4194,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4763,17 +4203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -4781,7 +4217,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4804,16 +4239,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4824,14 +4258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4841,11 +4275,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4857,15 +4291,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_02.models.Snapshot] @@ -4888,15 +4317,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4907,14 +4335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4924,11 +4352,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4368,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4972,7 +4398,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4980,16 +4406,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4428,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5035,14 +4456,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -5054,7 +4467,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5069,18 +4482,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -5092,7 +4497,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -5104,19 +4509,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] @@ -5148,7 +4542,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5160,17 +4554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5189,21 +4581,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5213,11 +4604,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5231,14 +4618,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_compute_management_client_enums.py index d3171ea09c67..e950992f4727 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_compute_management_client_enums.py @@ -32,7 +32,7 @@ class CopyCompletionErrorReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): COPY_SOURCE_NOT_FOUND = "CopySourceNotFound" """Indicates that the source snapshot was deleted while the background copy of the resource - #: created via CopyStart operation was in progress.""" + created via CopyStart operation was in progress.""" class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -40,8 +40,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -55,31 +55,28 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + disk and upload using write token in both disk and VM guest state""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -100,10 +97,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -113,17 +110,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -156,19 +153,19 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" PREMIUM_V2_LRS = "PremiumV2_LRS" """Premium SSD v2 locally redundant storage. Best for production and performance-sensitive - #: workloads that consistently require low latency and high IOPS and throughput.""" + workloads that consistently require low latency and high IOPS and throughput.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -176,13 +173,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -238,12 +235,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_models_py3.py index 9cd78d4a723d..c32bc11fd225 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/models/_models_py3.py @@ -135,7 +135,7 @@ class CopyCompletionError(_serialization.Model): """Indicates the error details if the background copy of a resource created via the CopyStart operation fails. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar error_code: Indicates the error code if the background copy of a resource created via the CopyStart operation fails. Required. "CopySourceNotFound" @@ -176,11 +176,11 @@ class CreationData(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", and "Import". + "ImportSecure", and "UploadPreparedSecure". :vartype create_option: str or ~azure.mgmt.compute.v2022_03_02.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -247,7 +247,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", and "Import". + "CopyStart", "ImportSecure", and "UploadPreparedSecure". :paramtype create_option: str or ~azure.mgmt.compute.v2022_03_02.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -293,7 +293,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -342,7 +342,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -684,7 +684,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -761,7 +761,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] @@ -817,7 +817,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -938,7 +938,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -1049,7 +1049,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_02.models.Disk] @@ -1274,7 +1274,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_02.models.DiskRestorePoint] @@ -1662,7 +1662,7 @@ class EncryptionSetIdentity(_serialization.Model): :vartype tenant_id: str :ivar user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_02.models.UserAssignedIdentitiesValue] """ @@ -1696,7 +1696,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_03_02.models.UserAssignedIdentitiesValue] """ @@ -1710,7 +1710,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1834,7 +1834,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2022_03_02.models.AccessLevel @@ -1964,7 +1964,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -2003,7 +2003,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2022_03_02.models.SourceVault @@ -2036,7 +2036,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2022_03_02.models.SourceVault @@ -2328,7 +2328,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2378,7 +2378,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2485,7 +2485,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2745,7 +2745,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_02.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/operations/_operations.py index 869295277811..902a27512dd3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_02/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -276,7 +276,7 @@ def build_disks_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +406,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +460,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -491,7 +491,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -532,7 +532,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -570,7 +570,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -608,7 +608,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -639,7 +639,7 @@ def build_disk_accesses_list_private_endpoint_connections_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -673,7 +673,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -738,7 +738,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -769,7 +769,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +823,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -894,7 +894,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -932,7 +932,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -975,7 +975,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1139,7 +1139,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_03_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_03_02.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_03_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_03_02.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -2409,7 +2257,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2286,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2294,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2322,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -2499,14 +2346,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2517,7 +2356,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2532,18 +2371,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2551,7 +2382,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -2562,19 +2397,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2605,7 +2429,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2615,23 +2439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -2657,7 +2479,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2665,16 +2487,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2694,10 +2515,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -2722,14 +2539,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2549,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2564,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2578,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -2789,19 +2590,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_03_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2622,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -2864,7 +2652,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -2883,21 +2670,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2909,13 +2695,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2934,21 +2716,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,11 +2739,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2974,14 +2751,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2777,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3017,17 +2786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -3035,7 +2800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3057,16 +2821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3077,14 +2840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3094,11 +2857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3110,15 +2873,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3140,15 +2898,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3176,11 +2933,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2949,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3206,7 +2961,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,21 +2979,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3251,20 +3004,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -3290,7 +3039,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -3299,16 +3048,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3328,12 +3076,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -3361,14 +3105,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3377,12 +3113,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3136,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3420,12 +3148,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -3441,20 +3169,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3488,7 +3205,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3498,17 +3215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -3524,7 +3239,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -3543,22 +3257,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3570,15 +3283,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -3595,22 +3304,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,14 +3328,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -3640,14 +3344,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3675,7 +3371,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3684,17 +3380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -3708,7 +3400,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3732,17 +3423,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3753,14 +3443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -3770,11 +3460,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3786,10 +3476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -3815,7 +3501,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3841,7 +3527,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3849,16 +3535,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3878,10 +3563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3906,14 +3587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -3925,7 +3598,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3940,18 +3613,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -3963,7 +3628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3975,19 +3640,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -4019,7 +3674,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4029,23 +3684,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4071,7 +3724,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4079,16 +3732,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4108,10 +3760,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4136,14 +3784,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -4155,7 +3795,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4170,18 +3810,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -4193,7 +3825,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -4205,19 +3837,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] @@ -4249,7 +3872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4259,17 +3882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4281,7 +3902,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4300,21 +3920,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4326,13 +3945,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4351,21 +3966,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4375,11 +3989,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4391,14 +4001,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4425,7 +4027,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4434,17 +4036,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4452,7 +4050,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4474,16 +4071,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4494,14 +4090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4511,11 +4107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4527,15 +4123,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4557,15 +4148,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4576,14 +4166,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4593,11 +4183,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4609,8 +4199,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4623,7 +4211,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4645,17 +4232,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4683,11 +4269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,10 +4285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -4745,7 +4327,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4764,23 +4345,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4792,13 +4372,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4814,7 +4390,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,18 +4411,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4858,14 +4432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4875,11 +4449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4891,17 +4465,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4927,7 +4497,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4937,16 +4507,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4529,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4998,14 +4563,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5018,7 +4575,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5037,18 +4594,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5061,7 +4610,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5077,19 +4626,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5122,7 +4660,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5134,17 +4672,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5168,23 +4704,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5194,11 +4729,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -5221,14 +4752,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5257,7 +4780,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -5268,17 +4791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -5302,7 +4821,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5327,7 +4846,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5335,16 +4854,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5364,10 +4882,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -5391,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5409,7 +4915,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5423,18 +4929,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,7 +4940,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -5453,19 +4951,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5496,7 +4983,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5506,20 +4993,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5544,7 +5033,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5552,16 +5041,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5581,10 +5069,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -5609,14 +5093,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5627,7 +5103,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5642,18 +5118,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5661,7 +5129,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -5672,19 +5144,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_03_02.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5715,7 +5176,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5725,17 +5186,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -5747,7 +5206,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -5766,21 +5224,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5792,13 +5249,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5817,21 +5270,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5841,11 +5293,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5857,14 +5305,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5891,7 +5331,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5900,17 +5340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -5918,7 +5354,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5940,16 +5375,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5960,14 +5394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -5977,11 +5411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5993,15 +5427,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -6023,15 +5452,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6042,14 +5470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -6059,11 +5487,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6075,13 +5503,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6107,7 +5533,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -6115,16 +5541,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6138,13 +5563,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6170,14 +5591,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,7 +5601,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6203,18 +5616,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6225,7 +5630,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -6237,19 +5642,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_03_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6280,7 +5674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6292,17 +5686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -6321,21 +5713,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6345,11 +5736,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6361,14 +5748,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_compute_management_client.py index 19ce8690771f..d01f24fffba2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -95,7 +97,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -138,7 +158,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -158,7 +178,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_configuration.py index 4ad81c322d15..331be0492835 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_metadata.json index bf68d6ad5df5..f8e85bcf6e1e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "galleries": "GalleriesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_compute_management_client.py index 89e8535c564e..07a424e3ad8f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -96,7 +98,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -139,7 +159,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-03-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -159,7 +181,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_configuration.py index 3976a4014b54..6d87c2d282ec 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-03-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/operations/_operations.py index 101ecf18eff8..f2238d69a144 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -162,10 +161,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -190,14 +185,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -209,7 +196,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -224,18 +211,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -244,7 +223,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -255,19 +234,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_03_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_03_03.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -299,7 +267,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -309,20 +277,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -347,7 +317,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -355,16 +325,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,13 +345,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -407,14 +372,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -426,7 +383,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -440,18 +397,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -460,7 +409,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -471,19 +424,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_03_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_03_03.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -515,7 +457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -525,17 +467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -559,7 +499,6 @@ async def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -578,23 +517,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -606,13 +544,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -631,21 +565,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -655,11 +588,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -669,14 +598,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -703,7 +624,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -712,17 +633,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -730,7 +647,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -752,16 +668,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -772,14 +687,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -789,11 +704,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -805,15 +720,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -835,15 +745,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -854,14 +763,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -871,11 +780,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -887,8 +796,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -915,7 +822,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -941,7 +848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -950,16 +857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,10 +888,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -1014,14 +916,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1034,7 +928,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1052,18 +946,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1076,7 +962,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -1091,19 +977,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1136,7 +1011,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1146,24 +1021,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1189,7 +1062,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1198,16 +1071,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1219,13 +1091,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1254,14 +1122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1274,7 +1134,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1291,18 +1151,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1315,7 +1167,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -1330,19 +1182,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1375,7 +1216,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1385,17 +1226,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1410,7 +1249,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1429,22 +1267,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1456,13 +1293,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -1481,22 +1314,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1506,11 +1338,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1525,14 +1353,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1560,7 +1380,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1569,17 +1389,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -1592,7 +1408,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -1615,17 +1430,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1636,14 +1450,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -1653,11 +1467,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,10 +1483,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -1700,7 +1510,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1726,7 +1536,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1736,16 +1546,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1768,10 +1577,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -1805,14 +1610,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1827,7 +1624,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1849,18 +1646,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1875,7 +1664,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -1894,19 +1683,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1941,7 +1720,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1951,17 +1730,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -1969,7 +1746,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1995,7 +1772,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2005,16 +1782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2026,13 +1802,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2067,14 +1839,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2089,7 +1853,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2111,18 +1875,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2137,7 +1893,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -2156,20 +1912,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2204,7 +1949,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2214,17 +1959,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2253,7 +1996,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -2272,7 +2014,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2280,16 +2022,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2301,13 +2042,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -2331,23 +2068,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2357,11 +2093,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2385,14 +2117,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2421,7 +2145,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2430,17 +2154,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -2456,7 +2176,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion] @@ -2479,18 +2198,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2501,14 +2219,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -2518,11 +2236,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2534,10 +2252,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -2564,7 +2278,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2590,7 +2304,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2599,16 +2313,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2631,10 +2344,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -2663,14 +2372,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2684,7 +2385,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2702,18 +2403,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2727,7 +2420,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -2742,19 +2435,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2788,7 +2471,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2798,24 +2481,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2841,7 +2522,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2850,16 +2531,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2871,13 +2551,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2907,14 +2583,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2928,7 +2596,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2946,18 +2614,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2971,7 +2631,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -2986,20 +2646,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3033,7 +2682,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3043,17 +2692,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3069,7 +2716,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3088,22 +2734,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3115,13 +2760,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -3140,22 +2781,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3165,11 +2805,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3185,14 +2821,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3220,7 +2848,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3229,17 +2857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -3252,7 +2876,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryApplication] @@ -3275,17 +2898,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3296,14 +2918,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -3313,11 +2935,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3329,10 +2951,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -3360,7 +2978,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3386,7 +3004,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3396,16 +3014,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3428,10 +3045,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -3466,14 +3079,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3488,7 +3093,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3510,18 +3115,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3536,7 +3133,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -3555,21 +3152,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3604,7 +3190,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3614,17 +3200,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -3632,7 +3216,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3658,7 +3242,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3668,16 +3252,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3689,13 +3272,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3731,14 +3310,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3753,7 +3324,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3775,18 +3346,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3801,7 +3364,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -3820,20 +3383,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3868,7 +3421,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3878,17 +3431,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3917,7 +3468,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3936,7 +3486,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3944,16 +3494,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3965,13 +3514,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3995,23 +3540,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4021,11 +3565,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4049,14 +3589,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4085,7 +3617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4094,17 +3626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -4120,7 +3648,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4144,18 +3671,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4166,14 +3692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -4183,11 +3709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4199,10 +3725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -4228,7 +3750,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -4254,7 +3776,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4262,16 +3784,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4291,10 +3812,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -4316,14 +3833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4336,7 +3845,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4348,18 +3857,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4372,7 +3873,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -4382,19 +3883,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_03_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_03_03.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4427,7 +3917,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4437,17 +3927,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -4481,7 +3969,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.SharedGallery] @@ -4504,17 +3991,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4525,14 +4011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -4542,11 +4028,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4558,10 +4044,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -4570,7 +4052,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,21 +4070,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4615,13 +4095,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -4662,7 +4138,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImage] @@ -4685,18 +4160,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4707,14 +4181,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -4724,11 +4198,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4740,10 +4214,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -4757,7 +4227,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4776,22 +4245,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4803,13 +4271,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -4854,7 +4318,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -4878,19 +4341,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4901,14 +4363,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -4918,11 +4380,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4934,10 +4396,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -4961,7 +4419,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4980,23 +4437,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5008,13 +4464,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -5045,7 +4497,6 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5064,21 +4515,20 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5090,13 +4540,9 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -5131,7 +4577,6 @@ async def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5150,22 +4595,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4621,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4635,6 @@ def list( :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -5219,17 +4658,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -5257,11 +4695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,10 +4711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -5320,7 +4754,6 @@ async def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5339,23 +4772,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5367,13 +4799,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5387,7 +4815,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -5411,18 +4838,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5433,14 +4859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -5450,11 +4876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5465,7 +4891,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/models/_models_py3.py index 580bae7db393..6b4f6c2d6ebb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/models/_models_py3.py @@ -159,26 +159,6 @@ class CommunityGallery(PirCommunityGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - "type": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "type": {"key": "type", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this community gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class CommunityGalleryImage(PirCommunityGalleryResource): # pylint: disable=too-many-instance-attributes """Specifies information about the gallery image definition that you want to create or update. @@ -362,7 +342,7 @@ def __init__( class CommunityGalleryImageList(_serialization.Model): """The List Community Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImage] @@ -475,7 +455,7 @@ def __init__( class CommunityGalleryImageVersionList(_serialization.Model): """The List Community Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImageVersion] @@ -599,7 +579,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -724,7 +704,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -773,7 +753,7 @@ class Gallery(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -867,7 +847,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -979,7 +959,7 @@ def __init__( class GalleryApplicationCustomAction(_serialization.Model): """A custom action that can be performed with a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1037,7 +1017,7 @@ class GalleryApplicationCustomActionParameter(_serialization.Model): """The definition of a parameter that can be passed to a custom action of a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1105,7 +1085,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.GalleryApplication] @@ -1289,7 +1269,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1370,7 +1350,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion] @@ -1503,12 +1483,12 @@ def __init__( class GalleryApplicationVersionPublishingProfile( GalleryArtifactPublishingProfileBase -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1686,18 +1666,6 @@ class GalleryApplicationVersionSafetyProfile(GalleryArtifactSafetyProfileBase): :vartype allow_deletion_of_replicated_locations: bool """ - _attribute_map = { - "allow_deletion_of_replicated_locations": {"key": "allowDeletionOfReplicatedLocations", "type": "bool"}, - } - - def __init__(self, *, allow_deletion_of_replicated_locations: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword allow_deletion_of_replicated_locations: Indicates whether or not removing this Gallery - Image Version from replicated regions is allowed. - :paramtype allow_deletion_of_replicated_locations: bool - """ - super().__init__(allow_deletion_of_replicated_locations=allow_deletion_of_replicated_locations, **kwargs) - class GalleryApplicationVersionUpdate(UpdateResourceDefinition): """Specifies information about the gallery Application Version that you want to update. @@ -1776,7 +1744,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2022_03_03.models.ManagedArtifact @@ -1905,7 +1873,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -2060,7 +2028,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2261,7 +2229,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -2301,7 +2269,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2505,7 +2473,7 @@ class GalleryImageVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2592,7 +2560,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion] @@ -2658,71 +2626,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) list[~azure.mgmt.compute.v2022_03_03.models.GalleryTargetExtendedLocation] """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - "target_extended_locations": {"key": "targetExtendedLocations", "type": "[GalleryTargetExtendedLocation]"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - target_extended_locations: Optional[List["_models.GalleryTargetExtendedLocation"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2022_03_03.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2022_03_03.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2022_03_03.models.ReplicationMode - :keyword target_extended_locations: The target extended locations where the Image Version is - going to be replicated to. This property is updatable. - :paramtype target_extended_locations: - list[~azure.mgmt.compute.v2022_03_03.models.GalleryTargetExtendedLocation] - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - target_extended_locations=target_extended_locations, - **kwargs - ) - class GalleryImageVersionSafetyProfile(GalleryArtifactSafetyProfileBase): """This is the safety profile of the Gallery Image Version. @@ -2884,7 +2787,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.Gallery] @@ -2929,32 +2832,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2022_03_03.models.GalleryDiskImageSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryDiskImageSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryDiskImageSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_03_03.models.HostCaching - :keyword source: The source for the disk image. - :paramtype source: ~azure.mgmt.compute.v2022_03_03.models.GalleryDiskImageSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryTargetExtendedLocation(_serialization.Model): """GalleryTargetExtendedLocation. @@ -3201,7 +3078,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3618,24 +3495,6 @@ class SharedGallery(PirSharedGalleryResource): :vartype unique_id: str """ - _validation = { - "name": {"readonly": True}, - "location": {"readonly": True}, - } - - _attribute_map = { - "name": {"key": "name", "type": "str"}, - "location": {"key": "location", "type": "str"}, - "unique_id": {"key": "identifier.uniqueId", "type": "str"}, - } - - def __init__(self, *, unique_id: Optional[str] = None, **kwargs: Any) -> None: - """ - :keyword unique_id: The unique id of this shared gallery. - :paramtype unique_id: str - """ - super().__init__(unique_id=unique_id, **kwargs) - class SharedGalleryDiskImage(_serialization.Model): """This is the disk image base class. @@ -3676,7 +3535,7 @@ class SharedGalleryDataDiskImage(SharedGalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_size_gb: This property indicates the size of the VHD to be created. :vartype disk_size_gb: int @@ -3860,7 +3719,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImage] @@ -3969,7 +3828,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImageVersion] @@ -4039,7 +3898,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_03_03.models.SharedGallery] @@ -4082,25 +3941,6 @@ class SharedGalleryOSDiskImage(SharedGalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryHostCaching """ - _validation = { - "disk_size_gb": {"readonly": True}, - } - - _attribute_map = { - "disk_size_gb": {"key": "diskSizeGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__( - self, *, host_caching: Optional[Union[str, "_models.SharedGalleryHostCaching"]] = None, **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryHostCaching - """ - super().__init__(host_caching=host_caching, **kwargs) - class SharingProfile(_serialization.Model): """Profile for gallery sharing to subscription or tenant. @@ -4224,7 +4064,7 @@ def __init__(self, *, summary: Optional[List["_models.RegionalSharingStatus"]] = class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. :code:`
    `:code:`
    ` Possible values are: :code:`
    `:code:`
    ` **Add** @@ -4365,7 +4205,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -4435,7 +4275,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -4519,7 +4359,7 @@ def __init__( class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/operations/_operations.py index 9d8a1fa13383..91283b8033f8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_03_03/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -179,7 +179,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -233,7 +233,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -367,7 +367,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -398,7 +398,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -439,7 +439,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -522,7 +522,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -560,7 +560,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +592,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +627,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -694,7 +694,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -726,7 +726,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -757,7 +757,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -800,7 +800,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -843,7 +843,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -887,7 +887,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -927,7 +927,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -959,7 +959,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1130,7 +1130,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1170,7 +1170,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1239,7 +1239,7 @@ def build_community_galleries_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_get_request( +def build_community_gallery_images_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1271,7 +1271,7 @@ def build_community_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_list_request( +def build_community_gallery_images_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1302,7 +1302,7 @@ def build_community_gallery_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_get_request( +def build_community_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, @@ -1340,7 +1340,7 @@ def build_community_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_list_request( +def build_community_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1393,7 +1393,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1418,7 +1418,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1426,16 +1426,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1458,10 +1457,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -1486,14 +1481,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1504,7 +1491,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1519,18 +1506,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1538,7 +1517,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -1549,19 +1528,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_03_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_03_03.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1592,7 +1560,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1602,20 +1570,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1640,7 +1610,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1648,16 +1618,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,13 +1638,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -1700,14 +1665,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,7 +1675,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1732,18 +1689,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1751,7 +1700,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -1762,19 +1715,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_03_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_03_03.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1805,7 +1747,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1815,17 +1757,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -1849,7 +1789,6 @@ def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -1868,23 +1807,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1896,13 +1834,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -1921,21 +1855,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1945,11 +1878,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1959,14 +1888,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1993,7 +1914,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2002,17 +1923,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -2020,7 +1937,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2042,16 +1958,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2062,14 +1977,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2079,11 +1994,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2095,15 +2010,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2125,15 +2035,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2144,14 +2053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2161,11 +2070,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2177,8 +2086,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -2205,7 +2112,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2231,7 +2138,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2240,16 +2147,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,10 +2178,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -2304,14 +2206,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2324,7 +2218,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2342,18 +2236,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2366,7 +2252,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -2381,19 +2267,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2426,7 +2301,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,24 +2311,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2479,7 +2352,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2488,16 +2361,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2509,13 +2381,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2544,14 +2412,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2564,7 +2424,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2581,18 +2441,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2605,7 +2457,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -2620,19 +2472,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] @@ -2665,7 +2506,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2675,17 +2516,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -2700,7 +2539,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2719,22 +2557,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2746,13 +2583,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -2771,22 +2604,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2796,11 +2628,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -2815,14 +2643,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2850,7 +2670,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2859,17 +2679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -2882,7 +2698,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2904,17 +2719,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2925,14 +2739,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -2942,11 +2756,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,10 +2772,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -2989,7 +2799,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3015,7 +2825,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3025,16 +2835,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3057,10 +2866,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -3094,14 +2899,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3116,7 +2913,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3138,18 +2935,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3164,7 +2953,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -3183,19 +2972,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3230,7 +3009,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3240,17 +3019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -3258,7 +3035,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3284,7 +3061,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3294,16 +3071,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3315,13 +3091,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3356,14 +3128,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3378,7 +3142,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3164,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3426,7 +3182,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -3445,20 +3201,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3493,7 +3238,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3503,17 +3248,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3542,7 +3285,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3561,7 +3303,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3569,16 +3311,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3590,13 +3331,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3620,23 +3357,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3646,11 +3382,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -3674,14 +3406,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3710,7 +3434,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3719,17 +3443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -3745,7 +3465,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryImageVersion] @@ -3768,18 +3487,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3790,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -3807,11 +3525,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3823,10 +3541,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -3853,7 +3567,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3879,7 +3593,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3888,16 +3602,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3920,10 +3633,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -3952,14 +3661,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3973,7 +3674,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3991,18 +3692,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4016,7 +3709,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -4031,19 +3724,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4077,7 +3760,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4087,24 +3770,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -4130,7 +3811,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4139,16 +3820,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4160,13 +3840,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4196,14 +3872,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4217,7 +3885,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4235,18 +3903,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4260,7 +3920,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -4275,20 +3935,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4322,7 +3971,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4332,17 +3981,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4358,7 +4005,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -4377,22 +4023,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4404,13 +4049,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -4429,22 +4070,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4454,11 +4094,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4474,14 +4110,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4509,7 +4137,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4518,17 +4146,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -4541,7 +4165,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -4563,17 +4186,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4584,14 +4206,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -4601,11 +4223,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4617,10 +4239,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -4648,7 +4266,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4674,7 +4292,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4684,16 +4302,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4716,10 +4333,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -4754,14 +4367,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4776,7 +4381,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4798,18 +4403,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4824,7 +4421,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -4843,21 +4440,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4892,7 +4478,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4902,17 +4488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -4920,7 +4504,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4946,7 +4530,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4956,16 +4540,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4977,13 +4560,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5019,14 +4598,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5041,7 +4612,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5063,18 +4634,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5089,7 +4652,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -5108,20 +4671,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5156,7 +4709,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5166,17 +4719,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5205,7 +4756,6 @@ def get( :keyword expand: The expand expression to apply on the operation. "ReplicationStatus" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_03_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5224,7 +4774,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -5232,16 +4782,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5253,13 +4802,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -5283,23 +4828,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5309,11 +4853,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5337,14 +4877,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5373,7 +4905,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5382,17 +4914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -5408,7 +4936,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5432,18 +4959,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5454,14 +4980,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -5471,11 +4997,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5487,10 +5013,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -5516,7 +5038,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -5542,7 +5064,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5550,16 +5072,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5579,10 +5100,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -5604,14 +5121,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.SharingUpdate] @@ -5623,7 +5132,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5635,18 +5144,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.SharingUpdate] @@ -5658,7 +5159,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -5668,19 +5169,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_03_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_03_03.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_03_03.models.SharingUpdate] @@ -5712,7 +5202,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5722,17 +5212,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -5766,7 +5254,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5788,17 +5275,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5809,14 +5295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -5826,11 +5312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5842,10 +5328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -5854,7 +5336,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5873,21 +5354,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5899,13 +5379,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -5946,7 +5422,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5968,18 +5443,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5990,14 +5464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -6007,11 +5481,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6023,10 +5497,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -6040,7 +5510,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6059,22 +5528,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6086,13 +5554,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -6137,7 +5601,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_03_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -6161,19 +5624,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6184,14 +5646,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -6201,11 +5663,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6217,10 +5679,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -6244,7 +5702,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6263,23 +5720,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6291,13 +5747,9 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -6328,7 +5780,6 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -6347,21 +5798,20 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6373,13 +5823,9 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -6414,7 +5860,6 @@ def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6433,22 +5878,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6460,13 +5904,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterable["_models.CommunityGalleryImage"]: @@ -6476,7 +5916,6 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -6500,17 +5939,16 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6521,14 +5959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -6538,11 +5976,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6554,10 +5992,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -6601,7 +6035,6 @@ def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_03_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6620,23 +6053,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-03-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,13 +6080,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6668,7 +6096,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -6692,18 +6119,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6714,14 +6140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -6731,11 +6157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6746,7 +6172,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_compute_management_client.py index cda8368e8492..8217fcc58b79 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -69,7 +71,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -91,7 +111,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-04-04" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -111,7 +131,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_configuration.py index 68c44c2068c0..afe4385d5b76 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-04-04") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_metadata.json index ef974c743cdb..b3da08f581fe 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "cloud_service_role_instances": "CloudServiceRoleInstancesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_compute_management_client.py index 1c656f16434f..5877e86d8727 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -69,7 +71,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -91,7 +111,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-04-04" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -111,7 +133,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_configuration.py index 9fb1479e7043..3681e45b4acf 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-04-04") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/operations/_operations.py index 6e279761e81a..0ed428e719a3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -105,22 +105,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -130,11 +129,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -148,14 +143,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -183,7 +170,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -192,17 +179,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -225,7 +208,6 @@ async def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_04_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -244,23 +226,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -272,13 +253,9 @@ async def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -292,7 +269,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -311,22 +287,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -338,13 +313,9 @@ async def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -366,7 +337,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_04_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.RoleInstance] @@ -389,18 +359,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -411,14 +380,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -428,11 +397,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -444,10 +413,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -465,22 +430,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -490,11 +454,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -509,14 +469,6 @@ async def begin_restart( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -544,7 +496,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -553,17 +505,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -582,22 +530,21 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -607,11 +554,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage( @@ -626,14 +569,6 @@ async def begin_reimage( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -661,7 +596,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -670,17 +605,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -699,22 +630,21 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -724,11 +654,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_rebuild( @@ -744,14 +670,6 @@ async def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -779,7 +697,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -788,17 +706,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_remote_desktop_file( @@ -812,8 +726,7 @@ async def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Async iterator of the response bytes or the result of cls(response) + :return: AsyncIterator[bytes] or the result of cls(response) :rtype: AsyncIterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -831,22 +744,21 @@ async def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -862,10 +774,6 @@ async def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -899,7 +807,6 @@ async def get( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -918,22 +825,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -945,13 +851,9 @@ async def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -964,7 +866,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudServiceRole] @@ -987,17 +888,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1008,14 +908,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1025,11 +925,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1041,10 +941,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -1070,7 +966,7 @@ async def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1099,7 +995,7 @@ async def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1107,16 +1003,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1136,10 +1031,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload async def begin_create_or_update( self, @@ -1162,14 +1053,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1181,7 +1064,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1194,18 +1077,10 @@ async def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1217,7 +1092,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -1227,20 +1102,9 @@ async def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudService or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1272,7 +1136,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1282,23 +1146,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1327,7 +1189,7 @@ async def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1335,16 +1197,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1356,13 +1217,9 @@ async def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1385,14 +1242,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1404,7 +1253,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1416,18 +1265,10 @@ async def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1439,7 +1280,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -1448,20 +1289,9 @@ async def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1493,7 +1323,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1503,17 +1333,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1532,21 +1360,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1556,11 +1383,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1572,14 +1395,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1606,7 +1421,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1615,17 +1430,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -1635,7 +1446,6 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -1654,21 +1464,20 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1680,13 +1489,9 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -1698,7 +1503,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1717,21 +1521,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1743,13 +1546,9 @@ async def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: @@ -1757,7 +1556,6 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1780,15 +1578,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1799,14 +1596,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1816,11 +1613,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1832,8 +1629,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -1842,7 +1637,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -1865,16 +1659,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1885,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1902,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1918,10 +1711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1939,21 +1728,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1963,11 +1751,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -1979,14 +1763,6 @@ async def begin_start( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2013,7 +1789,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2022,17 +1798,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2051,21 +1823,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2075,11 +1846,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -2092,14 +1859,6 @@ async def begin_power_off( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2126,7 +1885,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2135,23 +1894,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2180,7 +1935,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2188,16 +1943,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2207,11 +1961,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2234,14 +1984,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2252,7 +1994,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2264,18 +2006,10 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2286,7 +2020,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -2296,19 +2030,8 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2338,7 +2061,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2347,23 +2070,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2392,7 +2111,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2400,16 +2119,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2419,11 +2137,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -2447,14 +2161,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2465,7 +2171,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2478,18 +2184,10 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2500,7 +2198,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -2511,19 +2209,8 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2553,7 +2240,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2562,23 +2249,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2607,7 +2290,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2615,16 +2298,15 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2634,11 +2316,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_rebuild( @@ -2663,14 +2341,6 @@ async def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2681,7 +2351,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2695,18 +2365,10 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2717,7 +2379,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -2729,19 +2391,8 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2771,7 +2422,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2780,23 +2431,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2825,7 +2472,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2833,16 +2480,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2852,11 +2498,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -2879,14 +2521,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2897,7 +2531,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2909,18 +2543,10 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2931,7 +2557,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -2941,19 +2567,8 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2983,7 +2598,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2992,17 +2607,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3030,7 +2641,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3059,7 +2670,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3068,16 +2679,15 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3087,11 +2697,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_walk_update_domain( @@ -3119,14 +2725,6 @@ async def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3138,7 +2736,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3154,18 +2752,10 @@ async def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3177,7 +2767,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -3190,20 +2780,9 @@ async def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3234,7 +2813,7 @@ async def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3243,17 +2822,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_update_domain( @@ -3271,7 +2846,6 @@ async def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -3290,22 +2864,21 @@ async def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3317,13 +2890,9 @@ async def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -3335,7 +2904,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.UpdateDomain] @@ -3358,17 +2926,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3379,14 +2946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -3396,11 +2963,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3412,10 +2979,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -3446,7 +3009,6 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3465,21 +3027,20 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3491,13 +3052,9 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSVersion"]: @@ -3507,7 +3064,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.OSVersion] @@ -3530,16 +3086,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3550,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -3567,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3583,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace_async async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -3596,7 +3147,6 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -3615,21 +3165,20 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3641,13 +3190,9 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSFamily"]: @@ -3657,7 +3202,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_04_04.models.OSFamily] @@ -3680,16 +3224,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3700,14 +3243,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -3717,11 +3260,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3732,7 +3275,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/models/_models_py3.py index 99e811b84c0f..977cec82a4bc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/models/_models_py3.py @@ -115,7 +115,7 @@ class CloudService(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -379,7 +379,7 @@ def __init__(self, *, role_instance: Optional["_models.InstanceViewStatusesSumma class CloudServiceListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -696,7 +696,7 @@ def __init__( class CloudServiceRoleListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.CloudServiceRole] @@ -1099,7 +1099,7 @@ def __init__(self, **kwargs: Any) -> None: class LoadBalancerConfiguration(_serialization.Model): """Describes the load balancer configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1146,7 +1146,7 @@ def __init__( class LoadBalancerConfigurationProperties(_serialization.Model): """Describes the properties of the load balancer configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar frontend_ip_configurations: Specifies the frontend IP to be used for the load balancer. Only IPv4 frontend IP address is supported. Each load balancer configuration must have exactly @@ -1184,7 +1184,7 @@ class LoadBalancerFrontendIPConfiguration(_serialization.Model): """Specifies the frontend IP to be used for the load balancer. Only IPv4 frontend IP address is supported. Each load balancer configuration must have exactly one frontend IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the resource that is unique within the set of frontend IP configurations used by the load balancer. This name can be used to access the resource. @@ -1222,7 +1222,7 @@ def __init__( self.properties = properties -class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): +class LoadBalancerFrontendIPConfigurationProperties(_serialization.Model): # pylint: disable=name-too-long """Describes a cloud service IP Configuration. :ivar public_ip_address: The reference to the public ip address resource. @@ -1309,7 +1309,7 @@ def __init__(self, *, properties: Optional["_models.OSFamilyProperties"] = None, class OSFamilyListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.OSFamily] @@ -1421,7 +1421,7 @@ def __init__(self, *, properties: Optional["_models.OSVersionProperties"] = None class OSVersionListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.OSVersion] @@ -1543,7 +1543,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1743,7 +1743,7 @@ def __init__( class RoleInstanceListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.RoleInstance] @@ -1833,7 +1833,7 @@ def __init__( class RoleInstances(_serialization.Model): """Specifies a list of role instances from the cloud service. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar role_instances: List of cloud service role instance names. Value of '*' will signify all role instances of the cloud service. Required. @@ -2030,7 +2030,7 @@ def __init__(self, **kwargs: Any) -> None: class UpdateDomainListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_04_04.models.UpdateDomain] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/operations/_operations.py index bf6c316dd1c6..327952c1fee2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_04_04/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -39,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_cloud_service_role_instances_delete_request( +def build_cloud_service_role_instances_delete_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -71,7 +71,7 @@ def build_cloud_service_role_instances_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_request( +def build_cloud_service_role_instances_get_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, @@ -111,7 +111,7 @@ def build_cloud_service_role_instances_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_instance_view_request( +def build_cloud_service_role_instances_get_instance_view_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -143,7 +143,7 @@ def build_cloud_service_role_instances_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_list_request( +def build_cloud_service_role_instances_list_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, @@ -181,7 +181,7 @@ def build_cloud_service_role_instances_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_restart_request( +def build_cloud_service_role_instances_restart_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -213,7 +213,7 @@ def build_cloud_service_role_instances_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_reimage_request( +def build_cloud_service_role_instances_reimage_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -245,7 +245,7 @@ def build_cloud_service_role_instances_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_rebuild_request( +def build_cloud_service_role_instances_rebuild_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -277,7 +277,7 @@ def build_cloud_service_role_instances_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_remote_desktop_file_request( +def build_cloud_service_role_instances_get_remote_desktop_file_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -372,7 +372,7 @@ def build_cloud_service_roles_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_create_or_update_request( +def build_cloud_services_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -502,7 +502,7 @@ def build_cloud_services_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_get_instance_view_request( +def build_cloud_services_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -749,7 +749,7 @@ def build_cloud_services_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_delete_instances_request( +def build_cloud_services_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -783,7 +783,7 @@ def build_cloud_services_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_walk_update_domain_request( +def build_cloud_services_update_domain_walk_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -818,7 +818,7 @@ def build_cloud_services_update_domain_walk_update_domain_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_get_update_domain_request( +def build_cloud_services_update_domain_get_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -850,7 +850,7 @@ def build_cloud_services_update_domain_get_update_domain_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_list_update_domains_request( +def build_cloud_services_update_domain_list_update_domains_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -881,7 +881,7 @@ def build_cloud_services_update_domain_list_update_domains_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_version_request( +def build_cloud_service_operating_systems_get_os_version_request( # pylint: disable=name-too-long location: str, os_version_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_cloud_service_operating_systems_get_os_version_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_versions_request( +def build_cloud_service_operating_systems_list_os_versions_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +942,7 @@ def build_cloud_service_operating_systems_list_os_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_family_request( +def build_cloud_service_operating_systems_get_os_family_request( # pylint: disable=name-too-long location: str, os_family_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -973,7 +973,7 @@ def build_cloud_service_operating_systems_get_os_family_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_families_request( +def build_cloud_service_operating_systems_list_os_families_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1040,22 +1040,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1065,11 +1064,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -1083,14 +1078,6 @@ def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1118,7 +1105,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1127,17 +1114,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -1160,7 +1143,6 @@ def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_04_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -1179,23 +1161,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1207,13 +1188,9 @@ def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -1227,7 +1204,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1246,22 +1222,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1273,13 +1248,9 @@ def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1301,7 +1272,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_04_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.RoleInstance] :raises ~azure.core.exceptions.HttpResponseError: @@ -1323,18 +1293,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1345,14 +1314,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -1362,11 +1331,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1378,10 +1347,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1399,22 +1364,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1424,11 +1388,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -1443,14 +1403,6 @@ def begin_restart( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1478,7 +1430,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1487,17 +1439,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1516,22 +1464,21 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1541,11 +1488,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage( @@ -1560,14 +1503,6 @@ def begin_reimage( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1595,7 +1530,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1604,17 +1539,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1633,22 +1564,21 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1658,11 +1588,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_rebuild( @@ -1678,14 +1604,6 @@ def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1713,7 +1631,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1722,17 +1640,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_remote_desktop_file( @@ -1746,8 +1660,7 @@ def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Iterator of the response bytes or the result of cls(response) + :return: Iterator[bytes] or the result of cls(response) :rtype: Iterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -1765,22 +1678,21 @@ def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1796,10 +1708,6 @@ def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -1833,7 +1741,6 @@ def get( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -1852,22 +1759,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1879,13 +1785,9 @@ def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1898,7 +1800,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudServiceRole] :raises ~azure.core.exceptions.HttpResponseError: @@ -1920,17 +1821,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1941,14 +1841,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1958,11 +1858,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1974,10 +1874,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -2003,7 +1899,7 @@ def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -2032,7 +1928,7 @@ def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2040,16 +1936,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2069,10 +1964,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload def begin_create_or_update( self, @@ -2095,14 +1986,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2114,7 +1997,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2127,18 +2010,10 @@ def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2150,7 +2025,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -2160,20 +2035,9 @@ def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudService or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2205,7 +2069,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2215,23 +2079,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -2260,7 +2122,7 @@ def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2268,16 +2130,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2289,13 +2150,9 @@ def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2318,14 +2175,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2337,7 +2186,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2349,18 +2198,10 @@ def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2372,7 +2213,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -2381,20 +2222,9 @@ def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceUpdate or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_04_04.models.CloudService] @@ -2426,7 +2256,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,17 +2266,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2465,21 +2293,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2489,11 +2316,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2503,14 +2326,6 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2537,7 +2352,7 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2546,17 +2361,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -2566,7 +2377,6 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -2585,21 +2395,20 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2611,13 +2420,9 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -2629,7 +2434,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -2648,21 +2452,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2674,13 +2477,9 @@ def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: @@ -2688,7 +2487,6 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -2710,15 +2508,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2729,14 +2526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2746,11 +2543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2762,8 +2559,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -2772,7 +2567,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -2794,16 +2588,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2814,14 +2607,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2831,11 +2624,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2847,10 +2640,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -2868,21 +2657,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2892,11 +2680,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2906,14 +2690,6 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2940,7 +2716,7 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2949,17 +2725,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2978,21 +2750,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3002,11 +2773,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3017,14 +2784,6 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3051,7 +2810,7 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3060,23 +2819,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3105,7 +2860,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3113,16 +2868,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3132,11 +2886,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -3159,14 +2909,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3177,7 +2919,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3189,18 +2931,10 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3211,7 +2945,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -3221,19 +2955,8 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3263,7 +2986,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3272,23 +2995,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3317,7 +3036,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3325,16 +3044,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3344,11 +3062,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -3372,14 +3086,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3390,7 +3096,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3403,18 +3109,10 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3425,7 +3123,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -3436,19 +3134,8 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3478,7 +3165,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3487,23 +3174,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3532,7 +3215,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3540,16 +3223,15 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3559,11 +3241,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_rebuild( @@ -3588,14 +3266,6 @@ def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3606,7 +3276,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3620,18 +3290,10 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3642,7 +3304,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -3654,19 +3316,8 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3696,7 +3347,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3705,23 +3356,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3750,7 +3397,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3758,16 +3405,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3777,11 +3423,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -3804,14 +3446,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3822,7 +3456,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3834,18 +3468,10 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3856,7 +3482,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -3866,19 +3492,8 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3908,7 +3523,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3917,17 +3532,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3955,7 +3566,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3984,7 +3595,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3993,16 +3604,15 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4012,11 +3622,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_walk_update_domain( @@ -4044,14 +3650,6 @@ def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4063,7 +3661,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4079,18 +3677,10 @@ def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4102,7 +3692,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -4115,20 +3705,9 @@ def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4159,7 +3738,7 @@ def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4168,17 +3747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_update_domain( @@ -4196,7 +3771,6 @@ def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -4215,22 +3789,21 @@ def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4242,13 +3815,9 @@ def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -4260,7 +3829,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.UpdateDomain] :raises ~azure.core.exceptions.HttpResponseError: @@ -4282,17 +3850,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4303,14 +3870,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -4320,11 +3887,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4336,10 +3903,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -4370,7 +3933,6 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4389,21 +3951,20 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4415,13 +3976,9 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OSVersion"]: @@ -4431,7 +3988,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.OSVersion] :raises ~azure.core.exceptions.HttpResponseError: @@ -4453,16 +4009,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4473,14 +4028,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -4490,11 +4045,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4506,10 +4061,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -4519,7 +4070,6 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_04_04.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -4538,21 +4088,20 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-04-04")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4564,13 +4113,9 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OSFamily"]: @@ -4580,7 +4125,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_04_04.models.OSFamily] :raises ~azure.core.exceptions.HttpResponseError: @@ -4602,16 +4146,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4622,14 +4165,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -4639,11 +4182,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4654,7 +4197,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_compute_management_client.py index 15a7071ad4a1..5ec01ca4cf54 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-07-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_configuration.py index 09e53eb4ff09..584750853fcd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-07-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_metadata.json index 03d5d508acd3..5ae63eb35960 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_compute_management_client.py index 95718dfc2daf..af276befb96f 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-07-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_configuration.py index 1106b362766d..b9f22c81bdbf 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-07-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/operations/_operations.py index 608698a1b9dc..3429ace300a3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_07_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_07_02.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_07_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_07_02.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -1238,7 +1086,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1115,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1123,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1151,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -1328,14 +1175,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1347,7 +1186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1362,18 +1201,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1382,7 +1213,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -1393,19 +1228,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1437,7 +1261,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1447,23 +1271,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -1489,7 +1311,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1497,16 +1319,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1526,10 +1347,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -1554,14 +1371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1573,7 +1382,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1588,18 +1397,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1611,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -1623,19 +1424,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1667,7 +1457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1677,17 +1467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -1699,7 +1487,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,21 +1505,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1744,13 +1530,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -1769,21 +1551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,11 +1574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1811,14 +1588,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1845,7 +1614,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1854,17 +1623,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -1872,7 +1637,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1895,16 +1659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1915,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -1932,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1948,15 +1711,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -1979,15 +1737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1998,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -2015,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2031,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2045,7 +1800,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2064,21 +1818,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2090,20 +1843,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -2129,7 +1878,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -2138,16 +1887,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2167,12 +1915,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -2200,14 +1944,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2216,12 +1952,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2239,18 +1975,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2259,12 +1987,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -2280,20 +2008,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2327,7 +2044,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2337,17 +2054,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -2363,7 +2078,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -2382,22 +2096,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2409,15 +2122,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -2434,22 +2143,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2459,14 +2167,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -2479,14 +2183,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2514,7 +2210,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2523,17 +2219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -2547,7 +2239,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2571,17 +2262,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2592,14 +2282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -2609,11 +2299,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2625,10 +2315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -2654,7 +2340,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2680,7 +2366,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2374,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2717,10 +2402,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2745,14 +2426,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2765,7 +2438,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2780,18 +2453,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2804,7 +2469,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2816,19 +2481,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2861,7 +2516,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2871,23 +2526,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2913,7 +2566,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2921,16 +2574,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2950,10 +2602,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2978,14 +2626,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2998,7 +2638,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3013,18 +2653,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3037,7 +2669,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3049,19 +2681,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3094,7 +2717,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3104,17 +2727,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3128,7 +2749,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3147,21 +2767,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2792,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3198,21 +2813,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,11 +2836,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3240,14 +2850,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3274,7 +2876,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3283,17 +2885,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3303,7 +2901,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -3326,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3346,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3363,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3379,15 +2975,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -3410,15 +3001,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3429,14 +3019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3446,11 +3036,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3462,8 +3052,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3476,7 +3064,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3498,17 +3085,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3519,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3536,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3552,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -3598,7 +3180,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -3617,23 +3198,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3645,13 +3225,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -3667,7 +3243,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskRestorePoint] @@ -3690,18 +3265,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3712,14 +3286,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -3729,11 +3303,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3745,17 +3319,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3781,7 +3351,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -3791,16 +3361,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3383,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -3852,14 +3417,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -3873,7 +3430,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3892,18 +3449,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -3917,7 +3466,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -3933,19 +3482,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -3979,7 +3517,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3991,17 +3529,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -4025,23 +3561,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4051,11 +3586,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -4078,14 +3609,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4114,7 +3637,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -4125,17 +3648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -4159,7 +3678,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4184,7 +3703,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4192,16 +3711,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4221,10 +3739,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -4248,14 +3762,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4267,7 +3773,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4281,18 +3787,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4301,7 +3799,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -4312,19 +3810,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4356,7 +3843,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4366,20 +3853,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4404,7 +3893,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4412,16 +3901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4441,10 +3929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -4469,14 +3953,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4488,7 +3964,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4503,18 +3979,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4523,7 +3991,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -4534,19 +4006,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4578,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4588,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -4610,7 +4069,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -4629,21 +4087,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4655,13 +4112,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4680,21 +4133,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4704,11 +4156,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4720,14 +4168,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4754,7 +4194,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4763,17 +4203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -4781,7 +4217,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4804,16 +4239,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4824,14 +4258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4841,11 +4275,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4857,15 +4291,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_07_02.models.Snapshot] @@ -4888,15 +4317,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4907,14 +4335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4924,11 +4352,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4368,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4972,7 +4398,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4980,16 +4406,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4428,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5035,14 +4456,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -5054,7 +4467,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5069,18 +4482,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -5092,7 +4497,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -5104,19 +4509,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] @@ -5148,7 +4542,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5160,17 +4554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5189,21 +4581,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5213,11 +4604,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5231,14 +4618,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_compute_management_client_enums.py index d3171ea09c67..e950992f4727 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_compute_management_client_enums.py @@ -32,7 +32,7 @@ class CopyCompletionErrorReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): COPY_SOURCE_NOT_FOUND = "CopySourceNotFound" """Indicates that the source snapshot was deleted while the background copy of the resource - #: created via CopyStart operation was in progress.""" + created via CopyStart operation was in progress.""" class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -40,8 +40,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -55,31 +55,28 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + disk and upload using write token in both disk and VM guest state""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -100,10 +97,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -113,17 +110,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -156,19 +153,19 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" PREMIUM_V2_LRS = "PremiumV2_LRS" """Premium SSD v2 locally redundant storage. Best for production and performance-sensitive - #: workloads that consistently require low latency and high IOPS and throughput.""" + workloads that consistently require low latency and high IOPS and throughput.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -176,13 +173,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -238,12 +235,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_models_py3.py index 138c8a9de987..758765671841 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/models/_models_py3.py @@ -135,7 +135,7 @@ class CopyCompletionError(_serialization.Model): """Indicates the error details if the background copy of a resource created via the CopyStart operation fails. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar error_code: Indicates the error code if the background copy of a resource created via the CopyStart operation fails. Required. "CopySourceNotFound" @@ -176,11 +176,11 @@ class CreationData(_serialization.Model): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", and "Import". + "ImportSecure", and "UploadPreparedSecure". :vartype create_option: str or ~azure.mgmt.compute.v2022_07_02.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -253,7 +253,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", and "Import". + "CopyStart", "ImportSecure", and "UploadPreparedSecure". :paramtype create_option: str or ~azure.mgmt.compute.v2022_07_02.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -304,7 +304,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -353,7 +353,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -715,7 +715,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -792,7 +792,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] @@ -848,7 +848,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -969,7 +969,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -1080,7 +1080,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2022_07_02.models.Disk] @@ -1305,7 +1305,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2022_07_02.models.DiskRestorePoint] @@ -1708,7 +1708,7 @@ class EncryptionSetIdentity(_serialization.Model): :vartype tenant_id: str :ivar user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_07_02.models.UserAssignedIdentitiesValue] """ @@ -1742,7 +1742,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_07_02.models.UserAssignedIdentitiesValue] """ @@ -1756,7 +1756,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1880,7 +1880,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2022_07_02.models.AccessLevel @@ -2010,7 +2010,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -2049,7 +2049,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2022_07_02.models.SourceVault @@ -2082,7 +2082,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2022_07_02.models.SourceVault @@ -2374,7 +2374,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2424,7 +2424,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2531,7 +2531,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2798,7 +2798,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2022_07_02.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/operations/_operations.py index aaa8b4f5d143..de7ca5858bb2 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_07_02/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -276,7 +276,7 @@ def build_disks_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +406,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +460,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -491,7 +491,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -532,7 +532,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -570,7 +570,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -608,7 +608,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -639,7 +639,7 @@ def build_disk_accesses_list_private_endpoint_connections_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -673,7 +673,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -738,7 +738,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -769,7 +769,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +823,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -894,7 +894,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -932,7 +932,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -975,7 +975,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1139,7 +1139,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_07_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_07_02.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2022_07_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2022_07_02.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -2409,7 +2257,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2286,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2294,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2322,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -2499,14 +2346,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2517,7 +2356,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2532,18 +2371,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2551,7 +2382,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -2562,19 +2397,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2605,7 +2429,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2615,23 +2439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -2657,7 +2479,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2665,16 +2487,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2694,10 +2515,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -2722,14 +2539,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2549,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2564,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2578,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -2789,19 +2590,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2022_07_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2622,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -2864,7 +2652,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -2883,21 +2670,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2909,13 +2695,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2934,21 +2716,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,11 +2739,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2974,14 +2751,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2777,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3017,17 +2786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -3035,7 +2800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3057,16 +2821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3077,14 +2840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3094,11 +2857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3110,15 +2873,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3140,15 +2898,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3176,11 +2933,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2949,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3206,7 +2961,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,21 +2979,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3251,20 +3004,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -3290,7 +3039,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -3299,16 +3048,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3328,12 +3076,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -3361,14 +3105,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3377,12 +3113,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3136,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3420,12 +3148,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -3441,20 +3169,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3488,7 +3205,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3498,17 +3215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -3524,7 +3239,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -3543,22 +3257,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3570,15 +3283,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -3595,22 +3304,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,14 +3328,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -3640,14 +3344,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3675,7 +3371,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3684,17 +3380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -3708,7 +3400,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3732,17 +3423,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3753,14 +3443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -3770,11 +3460,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3786,10 +3476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -3815,7 +3501,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3841,7 +3527,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3849,16 +3535,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3878,10 +3563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3906,14 +3587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -3925,7 +3598,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3940,18 +3613,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -3963,7 +3628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3975,19 +3640,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -4019,7 +3674,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4029,23 +3684,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4071,7 +3724,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4079,16 +3732,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4108,10 +3760,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4136,14 +3784,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -4155,7 +3795,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4170,18 +3810,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -4193,7 +3825,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -4205,19 +3837,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] @@ -4249,7 +3872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4259,17 +3882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4281,7 +3902,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4300,21 +3920,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4326,13 +3945,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4351,21 +3966,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4375,11 +3989,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4391,14 +4001,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4425,7 +4027,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4434,17 +4036,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4452,7 +4050,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4474,16 +4071,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4494,14 +4090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4511,11 +4107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4527,15 +4123,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4557,15 +4148,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4576,14 +4166,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4593,11 +4183,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4609,8 +4199,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4623,7 +4211,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4645,17 +4232,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4683,11 +4269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,10 +4285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -4745,7 +4327,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4764,23 +4345,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4792,13 +4372,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4814,7 +4390,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,18 +4411,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4858,14 +4432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4875,11 +4449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4891,17 +4465,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4927,7 +4497,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4937,16 +4507,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4529,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4998,14 +4563,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5018,7 +4575,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5037,18 +4594,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5061,7 +4610,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5077,19 +4626,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5122,7 +4660,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5134,17 +4672,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5168,23 +4704,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5194,11 +4729,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -5221,14 +4752,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5257,7 +4780,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -5268,17 +4791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -5302,7 +4821,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5327,7 +4846,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5335,16 +4854,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5364,10 +4882,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -5391,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5409,7 +4915,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5423,18 +4929,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,7 +4940,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -5453,19 +4951,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5496,7 +4983,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5506,20 +4993,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5544,7 +5033,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5552,16 +5041,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5581,10 +5069,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -5609,14 +5093,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5627,7 +5103,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5642,18 +5118,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5661,7 +5129,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -5672,19 +5144,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2022_07_02.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5715,7 +5176,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5725,17 +5186,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -5747,7 +5206,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_07_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -5766,21 +5224,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5792,13 +5249,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5817,21 +5270,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5841,11 +5293,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5857,14 +5305,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5891,7 +5331,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5900,17 +5340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -5918,7 +5354,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5940,16 +5375,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5960,14 +5394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -5977,11 +5411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5993,15 +5427,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_07_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -6023,15 +5452,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6042,14 +5470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -6059,11 +5487,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6075,13 +5503,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6107,7 +5533,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -6115,16 +5541,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6138,13 +5563,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6170,14 +5591,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,7 +5601,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6203,18 +5616,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6225,7 +5630,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -6237,19 +5642,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2022_07_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_07_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6280,7 +5674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6292,17 +5686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -6321,21 +5713,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-07-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6345,11 +5736,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6361,14 +5748,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_compute_management_client.py index 83b8d822e570..fccd5e166fce 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-08-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_configuration.py index 9d2711ed44a4..93a3a28bb429 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-08-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_metadata.json index fec146c54a5a..e5ea236e04c4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_compute_management_client.py index 25aa7aa726e0..c90fbf60a96a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-08-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_configuration.py index accdbc95f79b..8c69c5e84610 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-08-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/operations/_operations.py index 2938490b5c81..783eec620088 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -234,7 +234,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -258,14 +257,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -276,14 +274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -293,11 +291,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -309,8 +307,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -339,7 +335,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -361,16 +356,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -381,14 +375,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -398,11 +392,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -414,8 +408,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -444,7 +436,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] @@ -467,16 +458,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -487,14 +477,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -504,11 +494,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -520,8 +510,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -550,7 +538,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -574,16 +561,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -594,14 +580,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -611,11 +597,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -627,15 +613,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -661,7 +643,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -669,16 +651,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -698,10 +679,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -723,14 +700,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -743,7 +712,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -755,18 +724,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -779,7 +740,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -788,20 +749,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -834,7 +784,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -844,23 +794,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -886,7 +834,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -894,16 +842,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -915,13 +862,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -944,14 +887,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -964,7 +899,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -976,18 +911,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1000,7 +927,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1009,20 +936,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1055,7 +972,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1065,17 +982,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1094,22 +1009,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1119,11 +1033,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1138,14 +1048,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1173,7 +1075,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1182,17 +1084,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1213,7 +1111,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1232,22 +1129,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1259,19 +1155,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -1300,7 +1192,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1308,16 +1200,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1327,11 +1218,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1358,14 +1245,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1376,7 +1255,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1391,18 +1270,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1413,7 +1284,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -1425,20 +1296,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -1468,7 +1328,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1477,23 +1337,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1521,7 +1377,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1530,16 +1386,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1549,11 +1404,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1582,14 +1433,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1600,7 +1443,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1614,21 +1457,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1639,7 +1474,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1651,23 +1486,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1698,7 +1522,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1707,17 +1531,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1729,7 +1549,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1748,21 +1567,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1774,13 +1592,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1788,7 +1602,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1812,16 +1625,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1832,14 +1644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1849,11 +1661,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1865,17 +1677,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1899,15 +1706,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1918,14 +1724,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -1935,11 +1741,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1951,8 +1757,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -1964,7 +1768,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -1988,17 +1791,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2009,14 +1811,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2026,11 +1828,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2042,10 +1844,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2056,7 +1854,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2080,17 +1877,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2101,14 +1897,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2118,11 +1914,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2134,15 +1930,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2173,7 +1965,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2182,16 +1974,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2201,11 +1992,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2237,14 +2024,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2255,7 +2034,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2271,7 +2050,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2279,14 +2058,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2297,7 +2068,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2311,24 +2082,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2359,7 +2119,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2368,23 +2128,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2413,7 +2169,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2421,16 +2177,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2440,11 +2195,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2469,14 +2220,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2487,7 +2230,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2500,18 +2243,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2522,7 +2257,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2532,20 +2267,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2575,7 +2299,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2584,23 +2308,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2629,7 +2349,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2637,16 +2357,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2656,11 +2375,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2685,14 +2400,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2703,7 +2410,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2716,18 +2423,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2738,7 +2437,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2748,20 +2447,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2791,7 +2479,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2800,23 +2488,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2845,7 +2529,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2853,16 +2537,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2872,11 +2555,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -2902,14 +2581,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2920,7 +2591,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2934,18 +2605,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2956,7 +2619,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -2967,20 +2630,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3010,7 +2662,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3019,23 +2671,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3064,7 +2712,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3072,16 +2720,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3091,11 +2738,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3123,14 +2766,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3141,7 +2776,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3157,18 +2792,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3179,7 +2806,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3192,20 +2819,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3235,7 +2851,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3244,23 +2860,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3286,7 +2898,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3294,16 +2906,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3313,11 +2924,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3342,14 +2949,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3360,7 +2959,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3373,18 +2972,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3395,7 +2986,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3405,20 +2996,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3448,7 +3028,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3457,23 +3037,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3502,7 +3078,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3510,16 +3086,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3529,11 +3104,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3559,14 +3130,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3577,7 +3140,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3591,18 +3154,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3613,7 +3168,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3625,20 +3180,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3668,7 +3212,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3677,23 +3221,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3722,7 +3262,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3730,16 +3270,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3749,11 +3288,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -3779,14 +3314,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3797,7 +3324,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3811,18 +3338,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3833,7 +3352,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -3844,20 +3363,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3887,7 +3395,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3896,20 +3404,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -3935,7 +3439,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -3954,7 +3457,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3962,16 +3465,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3983,13 +3485,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4014,7 +3512,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4025,7 +3522,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4038,11 +3535,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4053,7 +3549,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4064,13 +3560,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_08_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_08_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4098,7 +3591,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4106,16 +3599,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4125,17 +3617,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4161,7 +3649,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4169,16 +3657,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4188,11 +3675,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4216,14 +3699,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4234,7 +3709,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4247,18 +3722,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4269,7 +3736,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4280,19 +3747,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4322,7 +3779,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4331,20 +3788,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4369,7 +3822,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4395,7 +3848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4404,16 +3857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4433,10 +3885,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4464,14 +3912,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4485,7 +3925,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4501,18 +3941,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4526,7 +3958,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4539,20 +3971,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4586,7 +4007,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4596,24 +4017,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4639,7 +4058,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4648,16 +4067,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4677,10 +4095,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -4708,14 +4122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4729,7 +4135,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4745,18 +4151,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4770,7 +4168,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -4783,20 +4181,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4830,7 +4218,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4840,17 +4228,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -4869,22 +4255,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4894,11 +4279,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4913,14 +4294,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4948,7 +4321,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4957,17 +4330,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4989,7 +4358,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5008,23 +4376,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5036,13 +4403,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5054,7 +4417,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5078,17 +4440,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5099,14 +4460,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5116,11 +4477,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5132,12 +4493,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5174,21 +4531,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5198,11 +4554,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5214,14 +4566,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5248,7 +4592,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5257,17 +4601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5286,21 +4626,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5310,11 +4649,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5328,14 +4663,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5362,7 +4689,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5371,17 +4698,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5400,21 +4723,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5424,11 +4746,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5442,14 +4760,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5476,7 +4786,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5485,17 +4795,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5507,7 +4813,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5526,21 +4831,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5552,16 +4856,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5587,7 +4887,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5613,7 +4913,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5623,16 +4923,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5652,10 +4951,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -5685,14 +4980,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5707,7 +4994,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5724,18 +5011,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5750,7 +5029,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -5764,20 +5043,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5812,7 +5080,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5822,17 +5090,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -5840,7 +5106,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5866,7 +5132,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5876,16 +5142,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5897,13 +5162,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -5934,14 +5195,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5956,7 +5209,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5973,18 +5226,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5999,7 +5244,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6013,20 +5258,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6061,7 +5296,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6071,17 +5306,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6100,23 +5333,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6126,11 +5358,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6146,14 +5374,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6182,7 +5402,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6191,17 +5411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6226,7 +5442,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6245,7 +5460,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6253,16 +5468,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6274,13 +5488,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6302,7 +5512,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6321,23 +5530,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6349,13 +5557,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6383,7 +5587,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6412,7 +5618,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6421,16 +5627,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6440,11 +5645,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6472,14 +5673,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6491,7 +5684,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6506,18 +5699,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6529,7 +5714,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6541,20 +5728,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6585,7 +5762,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6594,17 +5771,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6623,22 +5796,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,11 +5820,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -6667,14 +5835,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6702,7 +5862,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6711,17 +5871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6740,22 +5896,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6765,11 +5920,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6785,14 +5936,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6820,7 +5963,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6829,24 +5972,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -6872,7 +6011,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6881,16 +6020,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6910,10 +6048,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -6940,14 +6074,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -6961,7 +6087,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6977,18 +6103,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7002,7 +6120,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -7015,19 +6133,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7061,7 +6168,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7071,17 +6178,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7106,23 +6211,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7132,11 +6236,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7159,14 +6259,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7195,7 +6287,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7204,17 +6296,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7238,7 +6326,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7257,23 +6344,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7285,13 +6371,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7305,7 +6387,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7324,22 +6405,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7351,13 +6431,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7386,7 +6462,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7410,7 +6485,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7418,12 +6493,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7434,14 +6508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7451,11 +6525,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7467,10 +6541,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -7494,23 +6564,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7520,11 +6589,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7550,14 +6615,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7586,7 +6643,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7595,17 +6652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7624,22 +6677,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7649,11 +6701,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -7667,14 +6715,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7702,7 +6742,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7711,17 +6751,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7740,22 +6776,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7765,11 +6800,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -7783,14 +6814,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7818,7 +6841,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7827,17 +6850,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7856,22 +6875,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7881,11 +6899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -7900,14 +6914,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7935,7 +6941,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7944,17 +6950,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7980,7 +6982,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7999,23 +7000,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8027,13 +7027,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8052,22 +7048,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8077,11 +7072,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8095,14 +7086,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8130,7 +7113,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8139,17 +7122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8163,7 +7142,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8182,22 +7160,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8207,18 +7184,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -8244,7 +7217,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8253,16 +7226,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8276,13 +7248,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8308,14 +7276,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8329,7 +7289,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8343,18 +7303,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8368,7 +7320,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -8380,19 +7332,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8426,7 +7367,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8438,17 +7379,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -8476,7 +7415,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8502,7 +7441,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8511,16 +7450,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8540,10 +7478,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8570,14 +7504,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8591,7 +7517,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8607,18 +7533,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8632,7 +7550,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8645,20 +7563,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8692,7 +7599,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8702,24 +7609,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8745,7 +7650,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8754,16 +7659,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8775,13 +7679,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8810,14 +7710,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8831,7 +7723,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8847,18 +7739,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8872,7 +7756,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8885,20 +7769,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8932,7 +7805,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8942,17 +7815,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8971,22 +7842,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8996,11 +7866,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9015,14 +7881,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9050,7 +7908,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9059,17 +7917,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9091,7 +7945,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9110,23 +7963,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9138,13 +7990,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9158,7 +8006,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9177,22 +8024,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9204,13 +8050,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -9240,7 +8082,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -9263,16 +8104,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9283,14 +8123,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9300,11 +8140,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9316,15 +8156,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9350,7 +8186,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9358,16 +8194,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9381,13 +8216,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9411,14 +8242,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9431,7 +8254,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9444,18 +8267,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9468,7 +8283,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9479,19 +8294,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9524,7 +8329,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9536,20 +8341,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9574,7 +8381,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9582,16 +8389,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9611,10 +8417,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9637,14 +8439,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9657,7 +8451,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9670,18 +8464,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9691,7 +8477,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9701,19 +8491,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9746,7 +8525,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9756,20 +8535,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9794,7 +8575,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9802,16 +8583,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9823,13 +8603,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9852,14 +8628,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9872,7 +8640,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9884,18 +8652,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9905,7 +8665,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -9914,19 +8678,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9959,7 +8712,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9969,17 +8722,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9998,22 +8749,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10023,11 +8773,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10042,14 +8788,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10077,7 +8815,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10086,17 +8824,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10119,7 +8853,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10138,22 +8871,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10165,13 +8897,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10183,7 +8911,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10202,21 +8929,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10228,13 +8954,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10253,21 +8975,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10277,11 +8998,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10294,14 +9011,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10328,7 +9037,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10337,17 +9046,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -10366,22 +9071,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10391,11 +9095,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10411,14 +9111,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10446,7 +9138,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10455,17 +9147,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10482,7 +9170,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10501,21 +9188,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10525,11 +9211,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -10545,7 +9227,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -10568,17 +9249,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10589,14 +9269,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10606,11 +9286,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10622,10 +9302,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -10641,7 +9317,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -10664,17 +9339,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10685,14 +9359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10702,11 +9376,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10718,8 +9392,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10731,7 +9403,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] @@ -10754,17 +9425,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10775,14 +9445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10792,11 +9462,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10808,10 +9478,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10829,22 +9495,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10854,11 +9519,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -10875,14 +9536,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10910,7 +9563,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10919,17 +9572,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10948,21 +9597,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10972,11 +9620,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -10986,14 +9630,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11020,7 +9656,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11029,17 +9665,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11058,21 +9690,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11082,11 +9713,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11096,14 +9723,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11130,7 +9749,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11139,17 +9758,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11168,21 +9783,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11192,11 +9806,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11206,14 +9816,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11240,7 +9842,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11249,17 +9851,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11278,21 +9876,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11302,11 +9899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11316,14 +9909,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11350,7 +9935,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11359,23 +9944,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11404,7 +9985,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11412,16 +9993,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11431,11 +10011,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11459,14 +10035,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11477,7 +10045,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11490,18 +10058,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11512,7 +10072,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -11522,19 +10082,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11564,7 +10114,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11573,17 +10123,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11605,7 +10151,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11624,22 +10169,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11651,13 +10195,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11676,21 +10216,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11700,11 +10239,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11716,14 +10251,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11750,7 +10277,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11759,17 +10286,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11781,7 +10304,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11800,21 +10322,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11824,11 +10345,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11847,21 +10364,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11875,13 +10391,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -11893,14 +10405,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -11930,7 +10434,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11942,23 +10446,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -11984,7 +10486,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11992,16 +10494,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12015,13 +10516,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12046,14 +10543,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12066,7 +10555,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12079,18 +10568,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12103,7 +10584,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12113,20 +10594,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12159,7 +10629,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12171,20 +10641,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12209,7 +10681,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12217,16 +10689,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12240,13 +10711,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12269,14 +10736,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12289,7 +10748,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12301,18 +10760,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12322,7 +10773,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12331,19 +10786,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12376,7 +10820,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12388,17 +10832,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -12437,7 +10879,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12456,7 +10897,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12464,16 +10905,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12485,13 +10925,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12523,7 +10959,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12542,7 +10977,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12552,16 +10987,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12573,13 +11007,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -12591,7 +11021,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12610,21 +11039,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12636,13 +11064,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -12650,7 +11074,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12669,20 +11092,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12694,13 +11116,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -12714,7 +11132,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12733,22 +11150,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12760,13 +11176,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -12778,7 +11190,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12797,21 +11208,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12823,13 +11233,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -12870,7 +11276,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12889,7 +11294,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -12898,16 +11303,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12919,13 +11323,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12962,7 +11362,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12981,7 +11380,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -12992,16 +11391,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13013,13 +11411,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13034,7 +11428,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13053,22 +11446,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13080,13 +11472,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -13098,7 +11486,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13117,21 +11504,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13143,13 +11529,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13166,7 +11548,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13185,23 +11566,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13213,13 +11593,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13256,7 +11632,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13275,23 +11650,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13303,13 +11677,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -13321,7 +11691,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13340,21 +11709,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13366,13 +11734,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -13400,7 +11764,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13419,7 +11782,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13428,16 +11791,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13449,13 +11811,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -13499,7 +11857,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13510,7 +11867,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13522,11 +11879,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13537,7 +11893,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -13547,12 +11903,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13580,7 +11932,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13588,16 +11940,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13609,13 +11960,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -13638,7 +11985,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13649,7 +11995,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13661,11 +12007,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13676,7 +12021,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -13686,12 +12031,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13719,7 +12060,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13727,16 +12068,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13748,13 +12088,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -13766,7 +12102,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13785,21 +12120,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13809,11 +12143,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -13823,7 +12153,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13842,21 +12171,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13868,13 +12196,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -13885,7 +12209,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet] @@ -13908,16 +12231,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13928,14 +12250,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -13945,11 +12267,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13961,17 +12283,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet] @@ -13994,16 +12311,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14014,14 +12330,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14031,11 +12347,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14047,10 +12363,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -14062,7 +12374,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] @@ -14085,17 +12396,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14106,14 +12416,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -14123,11 +12433,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14139,10 +12449,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -14186,7 +12492,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14197,7 +12502,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14210,11 +12515,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14225,7 +12529,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -14235,12 +12539,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14268,7 +12568,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14276,16 +12576,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14305,10 +12604,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -14331,7 +12626,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14342,7 +12636,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14355,11 +12649,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14370,7 +12663,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -14380,12 +12673,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14413,7 +12703,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14421,16 +12711,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14442,13 +12731,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14460,7 +12745,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14479,21 +12763,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14503,11 +12786,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14527,7 +12806,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14546,22 +12824,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14573,19 +12850,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14609,15 +12881,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14628,14 +12899,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14645,11 +12916,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14661,10 +12932,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14673,7 +12940,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14697,16 +12963,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14717,14 +12982,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14734,11 +12999,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14750,10 +13015,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -14798,7 +13059,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14809,7 +13069,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14823,11 +13083,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14838,7 +13097,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -14850,12 +13109,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14883,7 +13138,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -14891,16 +13146,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14920,10 +13174,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -14945,7 +13195,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14956,7 +13205,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14968,11 +13217,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14983,7 +13231,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -14993,12 +13241,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15026,7 +13270,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15034,16 +13278,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15055,13 +13298,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15073,7 +13312,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15092,21 +13330,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15116,11 +13353,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15142,7 +13375,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15161,22 +13393,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15188,13 +13419,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15205,7 +13432,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup] @@ -15228,16 +13454,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15248,14 +13473,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15265,11 +13490,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15281,16 +13506,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup] @@ -15313,15 +13533,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15332,14 +13551,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15349,11 +13568,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15365,8 +13584,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -15393,7 +13610,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15419,7 +13636,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15428,16 +13645,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15457,10 +13673,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -15485,14 +13697,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15506,7 +13710,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15520,18 +13724,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15545,7 +13741,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -15557,19 +13753,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15603,7 +13788,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15613,24 +13798,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15656,7 +13839,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15665,16 +13848,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15686,13 +13868,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15718,14 +13896,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15739,7 +13909,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15753,18 +13923,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15778,7 +13940,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -15790,19 +13952,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15836,7 +13987,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15846,17 +13997,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -15875,22 +14024,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15900,11 +14048,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15918,14 +14062,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15953,7 +14089,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15962,17 +14098,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -15996,7 +14128,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -16015,23 +14146,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16043,13 +14173,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -16062,7 +14188,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -16085,17 +14210,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16106,14 +14230,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -16123,11 +14247,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16139,10 +14263,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -16160,22 +14280,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16185,11 +14304,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -16206,14 +14321,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16241,7 +14348,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16250,17 +14357,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -16288,7 +14391,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16312,15 +14414,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16331,14 +14432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16348,11 +14449,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16364,8 +14465,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -16375,7 +14474,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16399,16 +14497,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16419,14 +14516,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16436,11 +14533,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16452,10 +14549,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -16477,7 +14570,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16488,7 +14580,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16500,11 +14592,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16515,7 +14606,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -16525,12 +14616,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16558,7 +14645,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16566,16 +14653,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16595,10 +14681,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -16620,7 +14702,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16631,7 +14712,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16643,11 +14724,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16658,7 +14738,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -16668,12 +14748,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16701,7 +14778,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16709,16 +14786,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16730,13 +14806,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -16748,7 +14820,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16767,21 +14838,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16791,11 +14861,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -16807,7 +14873,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16826,21 +14891,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16852,13 +14916,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -16872,7 +14932,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -16891,21 +14950,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16917,13 +14975,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -16947,7 +15001,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -16972,7 +15026,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -16980,16 +15034,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17009,10 +15062,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17034,14 +15083,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17052,7 +15093,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17064,18 +15105,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17083,7 +15116,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -17092,19 +15125,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17135,7 +15157,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17145,20 +15167,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17183,7 +15207,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17191,16 +15215,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17220,10 +15243,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -17245,14 +15264,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17263,7 +15274,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17275,18 +15286,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17294,7 +15297,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -17303,19 +15310,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17346,7 +15342,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17356,17 +15352,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -17385,21 +15379,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17409,11 +15402,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17423,14 +15412,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17457,7 +15438,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17466,17 +15447,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17490,7 +15467,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -17509,22 +15485,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17536,13 +15511,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -17550,7 +15521,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17572,16 +15542,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17592,14 +15561,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17609,11 +15578,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17625,16 +15594,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17656,15 +15620,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17675,14 +15638,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17692,11 +15655,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17708,8 +15671,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -17755,7 +15716,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17766,7 +15726,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17781,11 +15741,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17796,7 +15755,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -17808,12 +15767,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17841,7 +15796,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17849,16 +15804,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17878,10 +15832,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -17904,7 +15854,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17915,7 +15864,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17928,11 +15877,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17943,7 +15891,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -17953,12 +15901,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17986,7 +15931,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17994,16 +15939,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18015,13 +15959,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -18040,21 +15980,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18064,11 +16003,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18081,14 +16016,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18115,7 +16042,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18124,17 +16051,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18156,7 +16079,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18175,22 +16097,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18202,13 +16123,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -18216,7 +16133,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18240,16 +16156,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18260,14 +16175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18277,11 +16192,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18293,17 +16208,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18327,15 +16237,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18346,14 +16255,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18363,11 +16272,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18379,8 +16288,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -18407,7 +16314,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -18433,7 +16340,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -18442,16 +16349,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18463,13 +16369,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -18496,14 +16398,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -18516,7 +16410,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18531,18 +16425,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -18555,7 +16441,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -18568,19 +16454,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -18613,7 +16488,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18623,17 +16498,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18652,22 +16525,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18677,11 +16549,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18695,14 +16563,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18730,7 +16590,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18739,17 +16599,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18772,7 +16628,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -18791,23 +16646,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18819,13 +16673,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -18871,7 +16721,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18882,7 +16731,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18896,11 +16745,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18911,7 +16759,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -18923,12 +16771,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18956,7 +16800,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -18964,16 +16808,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18993,10 +16836,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -19020,7 +16859,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19031,7 +16869,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19045,11 +16883,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19060,7 +16897,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -19071,12 +16908,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19104,7 +16938,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19112,16 +16946,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19133,13 +16966,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -19154,7 +16983,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19173,21 +17001,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19197,11 +17024,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -19225,7 +17048,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19244,22 +17066,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19271,13 +17092,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -19298,7 +17115,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19322,17 +17138,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19343,14 +17158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19360,11 +17175,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19376,10 +17191,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -19393,7 +17204,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19417,16 +17227,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19437,14 +17246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19454,11 +17263,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19470,10 +17279,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -19500,7 +17305,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -19526,7 +17331,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19535,16 +17340,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19564,10 +17368,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -19594,14 +17394,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19615,7 +17407,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19631,18 +17423,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19656,7 +17440,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -19670,19 +17454,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19716,7 +17489,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19726,24 +17499,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -19769,7 +17540,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19778,16 +17549,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19801,13 +17571,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19833,14 +17599,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19854,7 +17612,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19868,18 +17626,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19893,7 +17643,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -19905,19 +17655,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19951,7 +17690,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19961,17 +17700,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -19994,22 +17731,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20019,11 +17755,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20043,14 +17775,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20078,7 +17802,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20087,17 +17811,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20122,7 +17842,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -20141,23 +17860,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20169,13 +17887,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -20188,7 +17902,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_01.models.CapacityReservation] @@ -20211,17 +17924,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20232,14 +17944,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -20249,11 +17961,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20265,10 +17977,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -20291,7 +17999,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20316,23 +18024,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20346,13 +18053,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20374,14 +18077,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20391,7 +18086,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20400,18 +18095,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20421,7 +18108,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20429,19 +18116,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20473,7 +18150,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20486,20 +18163,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20524,23 +18199,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20554,13 +18228,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20581,14 +18251,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20598,7 +18260,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20606,18 +18268,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20627,7 +18281,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20635,19 +18289,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20679,7 +18322,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20692,17 +18335,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20731,7 +18372,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20755,16 +18395,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20775,14 +18414,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20792,11 +18431,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20808,10 +18447,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20820,7 +18455,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20839,21 +18473,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20865,20 +18498,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20904,7 +18533,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20913,16 +18542,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20942,10 +18570,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20972,14 +18596,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -20993,7 +18609,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21009,18 +18625,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21034,7 +18642,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21047,19 +18655,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21093,7 +18690,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21103,24 +18700,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21146,7 +18741,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21155,16 +18750,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21176,13 +18770,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21210,14 +18800,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21231,7 +18813,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21247,18 +18829,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21272,7 +18846,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21285,19 +18859,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21331,7 +18895,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21341,17 +18905,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21370,22 +18932,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21395,11 +18956,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21414,14 +18971,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21449,7 +18998,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21458,17 +19007,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21490,7 +19035,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21509,23 +19053,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21537,13 +19080,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21557,7 +19096,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21581,18 +19119,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21603,14 +19140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21620,11 +19157,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21636,12 +19173,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21667,7 +19200,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21693,7 +19226,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21703,16 +19236,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21732,10 +19264,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21764,14 +19292,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21786,7 +19306,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21803,18 +19323,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21829,7 +19341,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21843,19 +19355,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21890,7 +19391,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21900,17 +19401,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21918,7 +19417,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21944,7 +19443,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21954,16 +19453,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21975,13 +19473,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22011,14 +19505,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22033,7 +19519,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22050,18 +19536,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22076,7 +19554,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22090,19 +19568,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22137,7 +19605,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22147,17 +19615,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22176,23 +19642,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22202,11 +19667,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22222,14 +19683,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22258,7 +19711,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22267,17 +19720,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22302,7 +19751,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22321,7 +19769,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22329,16 +19777,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22350,13 +19797,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22378,7 +19821,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22402,19 +19844,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22425,14 +19866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22442,11 +19883,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22457,7 +19898,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_compute_management_client_enums.py index beeede152cb5..7ea169d1209c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_compute_management_client_enums.py @@ -397,7 +397,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_models_py3.py index 4bedf3b1cf1d..054dcefffa2c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -265,7 +261,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -296,7 +292,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -316,7 +312,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -399,7 +395,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -456,7 +452,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -555,7 +551,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet] @@ -861,7 +857,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -987,7 +983,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1089,7 +1085,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup] @@ -1249,7 +1245,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.CapacityReservation] @@ -1499,7 +1495,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1700,7 +1696,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1883,7 +1879,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2013,7 +2009,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup] @@ -2046,7 +2042,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group.:code:`
    `:code:`
    `Minimum api-version: 2022-03-01. @@ -2266,7 +2262,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -2515,17 +2511,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2844,7 +2829,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3024,7 +3009,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_08_01.models.SubResource @@ -3132,7 +3117,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.Image] @@ -3166,7 +3151,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_08_01.models.SubResource @@ -3308,7 +3293,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3367,7 +3352,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3589,7 +3574,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3622,7 +3607,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -3927,7 +3912,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -3960,7 +3945,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.Usage] @@ -3994,7 +3979,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4348,7 +4333,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4424,7 +4409,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. Possible values are: **Windows** @@ -4574,7 +4559,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5000,7 +4985,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5111,7 +5096,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup] @@ -5171,17 +5156,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5222,7 +5196,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2022_08_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2022_08_01.models.PublicIPAddressSkuTier """ @@ -5241,8 +5215,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2022_08_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2022_08_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5253,7 +5226,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5324,7 +5297,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5557,7 +5530,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6266,7 +6239,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6329,7 +6302,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6392,7 +6365,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6466,7 +6439,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6513,7 +6486,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6546,7 +6519,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.RunCommandDocumentBase] @@ -6582,7 +6555,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6924,7 +6897,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -6934,7 +6907,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -6963,7 +6936,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -6978,7 +6951,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7034,7 +7007,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource] @@ -7297,7 +7270,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7318,67 +7291,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7447,7 +7359,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7590,7 +7502,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -7806,7 +7718,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8298,7 +8210,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8537,7 +8449,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8581,7 +8493,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8896,7 +8808,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_08_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_08_01.models.UserAssignedIdentitiesValue] """ @@ -8928,7 +8840,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2022_08_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_08_01.models.UserAssignedIdentitiesValue] """ @@ -8942,7 +8854,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9005,7 +8917,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9159,7 +9071,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9484,7 +9396,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -9517,10 +9429,12 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineNetworkInterfaceConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -9636,7 +9550,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9656,10 +9570,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -9825,10 +9739,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -9928,10 +9842,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -9986,7 +9900,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10233,7 +10147,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand] @@ -10382,7 +10296,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10642,7 +10556,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -10922,10 +10836,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension] @@ -11185,7 +11099,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_08_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_08_01.models.UserAssignedIdentitiesValue] """ @@ -11219,7 +11133,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_08_01.models.UserAssignedIdentitiesValue] """ @@ -11273,7 +11187,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11300,7 +11214,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(SubResource): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11455,10 +11369,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -11500,7 +11414,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet] @@ -11536,7 +11450,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetSku] @@ -11572,7 +11486,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet] @@ -11605,7 +11519,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -11656,10 +11570,12 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(SubResource): # pylint: disable=too-many-instance-attributes +class VirtualMachineScaleSetNetworkConfiguration( + SubResource +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11771,7 +11687,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -11796,7 +11712,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_08_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -11827,7 +11743,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_08_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -11846,7 +11762,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12141,10 +12057,10 @@ def __init__( self.allow_extension_operations = allow_extension_operations -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12233,10 +12149,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12263,7 +12181,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12271,18 +12189,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype temp_disk: bool """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - } - - def __init__(self, *, temp_disk: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - """ - super().__init__(temp_disk=temp_disk, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -12598,7 +12504,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -12716,7 +12622,9 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=too-many-instance-attributes +class VirtualMachineScaleSetUpdateNetworkConfiguration( + SubResource +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -12826,12 +12734,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_08_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12864,7 +12772,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_08_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13022,7 +12930,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -13082,7 +12990,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -13248,7 +13156,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -13647,7 +13555,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -13670,7 +13578,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -13853,10 +13761,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -14000,7 +13908,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM] @@ -14033,7 +13941,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -14884,14 +14792,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -14936,7 +14844,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -14992,7 +14900,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -15216,7 +15124,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/operations/_operations.py index 195744dd9062..07c5d3281107 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -176,7 +176,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -210,7 +210,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -248,7 +248,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -286,7 +286,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -320,7 +320,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -361,7 +361,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -392,7 +392,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -422,7 +422,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -448,7 +450,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -479,7 +481,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -510,7 +512,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -551,7 +553,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -585,7 +587,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -619,7 +621,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -653,7 +655,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -687,7 +689,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -721,7 +723,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -755,7 +757,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -789,7 +791,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -832,7 +834,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -866,7 +868,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -900,7 +902,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -935,7 +937,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -970,7 +972,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1002,7 +1004,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1042,7 +1044,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1073,7 +1075,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1104,7 +1106,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1137,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1168,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1197,7 +1199,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1238,7 +1240,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1279,7 +1281,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1317,7 +1319,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1359,7 +1361,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1399,7 +1401,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1434,7 +1436,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1466,7 +1468,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1498,7 +1500,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1533,7 +1535,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1573,7 +1575,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1613,7 +1615,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1645,7 +1647,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1691,7 +1693,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1731,7 +1733,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1763,7 +1765,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1795,7 +1797,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1827,7 +1829,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1869,7 +1871,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1901,7 +1903,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1933,7 +1935,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1968,7 +1970,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2003,7 +2005,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2038,7 +2040,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2070,7 +2072,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2110,7 +2112,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2143,7 +2145,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2205,7 +2209,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2349,7 +2353,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2380,7 +2384,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2411,7 +2415,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2444,7 +2448,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2537,7 +2541,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2759,7 +2763,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2799,7 +2803,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2830,7 +2834,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2861,7 +2865,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2892,7 +2896,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2926,7 +2930,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2994,7 +2998,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3042,7 +3046,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3073,7 +3077,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3102,7 +3106,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3134,7 +3138,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3165,7 +3169,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3207,7 +3211,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3257,7 +3261,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3289,7 +3293,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3320,7 +3324,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3353,7 +3357,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3386,7 +3390,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3417,7 +3421,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3463,7 +3467,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3593,7 +3597,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3649,7 +3653,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3680,7 +3684,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3716,7 +3720,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3752,7 +3756,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3785,7 +3789,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -3827,7 +3831,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3853,7 +3859,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3883,7 +3889,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3917,7 +3923,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3951,7 +3957,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4020,7 +4026,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4050,7 +4056,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4074,7 +4082,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4216,7 +4224,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4279,7 +4287,9 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4303,7 +4313,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4463,7 +4473,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4626,7 +4636,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4680,7 +4690,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4716,7 +4726,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4752,7 +4762,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4785,7 +4795,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4825,7 +4835,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4855,7 +4865,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5002,7 +5014,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5038,7 +5050,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5074,7 +5086,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5107,7 +5119,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5147,7 +5159,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5183,7 +5195,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5216,7 +5228,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5257,7 +5269,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5298,7 +5310,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5378,7 +5390,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5411,7 +5423,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5444,7 +5456,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5477,7 +5489,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5504,7 +5518,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5535,7 +5549,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5570,7 +5584,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5605,7 +5619,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5637,7 +5651,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5677,7 +5691,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5710,7 +5724,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5751,7 +5765,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5792,7 +5806,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5830,7 +5844,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5872,7 +5886,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5936,7 +5950,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5960,14 +5973,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5978,14 +5990,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5995,11 +6007,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6011,8 +6023,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6041,7 +6051,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6063,16 +6072,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6083,14 +6091,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6100,11 +6108,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6116,8 +6124,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6146,7 +6152,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6168,16 +6173,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6188,14 +6192,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6205,11 +6209,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6221,8 +6225,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6251,7 +6253,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6275,16 +6276,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6295,14 +6295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6312,11 +6312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6328,15 +6328,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6362,7 +6358,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6370,16 +6366,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6399,10 +6394,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6424,14 +6415,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6444,7 +6427,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6456,18 +6439,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6480,7 +6455,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6489,20 +6464,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6535,7 +6499,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6545,23 +6509,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6587,7 +6549,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6595,16 +6557,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6616,13 +6577,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6645,14 +6602,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6665,7 +6614,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6677,18 +6626,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6701,7 +6642,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6710,20 +6651,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6756,7 +6687,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6766,17 +6697,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6795,22 +6724,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6820,11 +6748,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6839,14 +6763,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6874,7 +6790,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6883,17 +6799,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6914,7 +6826,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6933,22 +6844,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6960,19 +6870,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7001,7 +6907,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7009,16 +6915,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7028,11 +6933,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7059,14 +6960,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7077,7 +6970,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7092,18 +6985,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7114,7 +6999,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7126,20 +7011,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7169,7 +7043,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7178,23 +7052,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7222,7 +7092,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7231,16 +7101,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7250,11 +7119,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7283,14 +7148,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7301,7 +7158,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7315,21 +7172,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7340,7 +7189,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7352,23 +7201,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7399,7 +7237,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7408,17 +7246,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7430,7 +7264,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7449,21 +7282,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7475,13 +7307,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7489,7 +7317,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7513,16 +7340,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7533,14 +7359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7550,11 +7376,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7566,17 +7392,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7600,15 +7421,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7619,14 +7439,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7636,11 +7456,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7652,8 +7472,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7665,7 +7483,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7689,17 +7506,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7710,14 +7526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7727,11 +7543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7743,10 +7559,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7757,7 +7569,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7781,17 +7592,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7802,14 +7612,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7819,11 +7629,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7835,15 +7645,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -7874,7 +7680,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7883,16 +7689,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7902,11 +7707,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -7938,14 +7739,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7956,7 +7749,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -7972,7 +7765,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -7980,14 +7773,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7998,7 +7783,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8012,24 +7797,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8060,7 +7834,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8069,23 +7843,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8114,7 +7884,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8122,16 +7892,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8141,11 +7910,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8170,14 +7935,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8188,7 +7945,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8201,18 +7958,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8223,7 +7972,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8233,20 +7982,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8276,7 +8014,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8285,23 +8023,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8330,7 +8064,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8338,16 +8072,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8357,11 +8090,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8386,14 +8115,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8404,7 +8125,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8417,18 +8138,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8439,7 +8152,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8449,20 +8162,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8492,7 +8194,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8501,23 +8203,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8546,7 +8244,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8554,16 +8252,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8573,11 +8270,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -8603,14 +8296,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8621,7 +8306,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8635,18 +8320,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8657,7 +8334,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8668,20 +8345,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8711,7 +8377,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8720,23 +8386,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8765,7 +8427,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8773,16 +8435,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8792,11 +8453,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -8824,14 +8481,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8842,7 +8491,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8858,18 +8507,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8880,7 +8521,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8893,20 +8534,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8936,7 +8566,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8945,23 +8575,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8987,7 +8613,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8995,16 +8621,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9014,11 +8639,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9043,14 +8664,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9061,7 +8674,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9074,18 +8687,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9096,7 +8701,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9106,20 +8711,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9149,7 +8743,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9158,23 +8752,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9203,7 +8793,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9211,16 +8801,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9230,11 +8819,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9260,14 +8845,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9278,7 +8855,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9292,18 +8869,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9314,7 +8883,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9326,20 +8895,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9369,7 +8927,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9378,23 +8936,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9423,7 +8977,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9431,16 +8985,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9450,11 +9003,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9480,14 +9029,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9498,7 +9039,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9512,18 +9053,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9534,7 +9067,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9545,20 +9078,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9588,7 +9110,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9597,20 +9119,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -9636,7 +9154,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9655,7 +9172,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9663,16 +9180,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9684,13 +9200,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -9715,7 +9227,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9726,7 +9237,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9739,11 +9250,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9754,7 +9264,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -9765,13 +9275,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_08_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_08_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9799,7 +9306,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9807,16 +9314,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9826,17 +9332,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9862,7 +9364,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9870,16 +9372,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9889,11 +9390,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -9917,14 +9414,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9935,7 +9424,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9948,18 +9437,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9970,7 +9451,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -9981,19 +9462,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10023,7 +9494,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10032,20 +9503,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10070,7 +9537,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10096,7 +9563,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10105,16 +9572,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10134,10 +9600,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10165,14 +9627,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10186,7 +9640,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10202,18 +9656,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10227,7 +9673,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10240,20 +9686,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10287,7 +9722,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10297,24 +9732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10340,7 +9773,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10349,16 +9782,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10378,10 +9810,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -10409,14 +9837,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10430,7 +9850,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10446,18 +9866,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10471,7 +9883,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -10484,20 +9896,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10531,7 +9933,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10541,17 +9943,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10570,22 +9970,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10595,11 +9994,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10614,14 +10009,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10649,7 +10036,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10658,17 +10045,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10690,7 +10073,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10709,23 +10091,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10737,13 +10118,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10755,7 +10132,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10779,17 +10155,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10800,14 +10175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10817,11 +10192,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10833,12 +10208,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10875,21 +10246,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10899,11 +10269,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10913,14 +10279,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10947,7 +10305,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10956,17 +10314,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10985,21 +10339,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11009,11 +10362,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11027,14 +10376,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11061,7 +10402,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11070,17 +10411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11099,21 +10436,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11123,11 +10459,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11141,14 +10473,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11175,7 +10499,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11184,17 +10508,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11206,7 +10526,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11225,21 +10544,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11251,16 +10569,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11286,7 +10600,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11312,7 +10626,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11322,16 +10636,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11351,10 +10664,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11384,14 +10693,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11406,7 +10707,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11423,18 +10724,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11449,7 +10742,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -11463,20 +10756,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11511,7 +10793,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11521,17 +10803,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -11539,7 +10819,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11565,7 +10845,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11575,16 +10855,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11596,13 +10875,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11633,14 +10908,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11655,7 +10922,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11672,18 +10939,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11698,7 +10957,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -11712,20 +10971,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11760,7 +11009,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11770,17 +11019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -11799,23 +11046,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11825,11 +11071,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11845,14 +11087,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11881,7 +11115,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11890,17 +11124,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11925,7 +11155,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11944,7 +11173,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11952,16 +11181,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11973,13 +11201,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12001,7 +11225,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12020,23 +11243,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12048,13 +11270,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12082,7 +11300,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12111,7 +11331,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12120,16 +11340,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12139,11 +11358,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12171,14 +11386,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12190,7 +11397,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12205,18 +11412,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12228,7 +11427,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12240,20 +11441,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12284,7 +11475,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12293,17 +11484,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12322,22 +11509,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12347,11 +11533,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -12366,14 +11548,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12401,7 +11575,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12410,17 +11584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12439,22 +11609,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12464,11 +11633,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12484,14 +11649,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12519,7 +11676,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12528,24 +11685,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12571,7 +11724,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12580,16 +11733,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12609,10 +11761,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -12639,14 +11787,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12660,7 +11800,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12676,18 +11816,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12701,7 +11833,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12714,19 +11846,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12760,7 +11881,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12770,17 +11891,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -12805,23 +11924,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12831,11 +11949,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12858,14 +11972,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12894,7 +12000,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12903,17 +12009,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12937,7 +12039,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -12956,23 +12057,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12984,13 +12084,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13004,7 +12100,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13023,22 +12118,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13050,13 +12144,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13085,7 +12175,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13109,7 +12198,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13117,12 +12206,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13133,14 +12221,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13150,11 +12238,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13166,10 +12254,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13193,23 +12277,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13219,11 +12302,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13249,14 +12328,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13285,7 +12356,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13294,17 +12365,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13323,22 +12390,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13348,11 +12414,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -13366,14 +12428,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13401,7 +12455,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13410,17 +12464,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13439,22 +12489,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13464,11 +12513,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -13482,14 +12527,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13517,7 +12554,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13526,17 +12563,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13555,22 +12588,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13580,11 +12612,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -13599,14 +12627,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13634,7 +12654,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13643,17 +12663,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13679,7 +12695,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13698,23 +12713,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13726,13 +12740,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13751,22 +12761,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13776,11 +12785,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -13794,14 +12799,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13829,7 +12826,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13838,17 +12835,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -13862,7 +12855,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13881,22 +12873,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13906,18 +12897,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -13943,7 +12930,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13952,16 +12939,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13975,13 +12961,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14007,14 +12989,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -14027,7 +13001,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14041,18 +13015,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -14065,7 +13031,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14077,19 +13043,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -14122,7 +13077,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14134,17 +13089,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -14172,7 +13125,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14198,7 +13151,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14207,16 +13160,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14236,10 +13188,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -14266,14 +13214,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14287,7 +13227,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14303,18 +13243,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14328,7 +13260,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -14341,20 +13273,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14388,7 +13309,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14398,24 +13319,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14441,7 +13360,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14450,16 +13369,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14471,13 +13389,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14506,14 +13420,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14527,7 +13433,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14543,18 +13449,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14568,7 +13466,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -14581,20 +13479,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14628,7 +13515,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14638,17 +13525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14667,22 +13552,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14692,11 +13576,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14711,14 +13591,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14746,7 +13618,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14755,17 +13627,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14787,7 +13655,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14806,23 +13673,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14834,13 +13700,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14854,7 +13716,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14873,22 +13734,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14900,13 +13760,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -14936,7 +13792,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -14958,16 +13813,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14978,14 +13832,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -14995,11 +13849,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15011,15 +13865,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15045,7 +13895,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15053,16 +13903,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15076,13 +13925,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15106,14 +13951,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15126,7 +13963,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15139,18 +13976,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15163,7 +13992,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15174,19 +14003,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15219,7 +14038,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15231,20 +14050,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15269,7 +14090,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15277,16 +14098,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15306,10 +14126,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15332,14 +14148,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15351,7 +14159,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15364,18 +14172,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15384,7 +14184,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15394,19 +14198,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15438,7 +14231,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15448,20 +14241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15486,7 +14281,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15494,16 +14289,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15515,13 +14309,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15544,14 +14334,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15563,7 +14345,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15575,18 +14357,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15595,7 +14369,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15604,19 +14382,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] @@ -15648,7 +14415,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15658,17 +14425,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15687,22 +14452,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15712,11 +14476,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15731,14 +14491,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15766,7 +14518,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15775,17 +14527,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15808,7 +14556,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -15827,22 +14574,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15854,13 +14600,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -15872,7 +14614,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15891,21 +14632,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15917,13 +14657,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -15942,21 +14678,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15966,11 +14701,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -15981,14 +14712,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16015,7 +14738,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16024,17 +14747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -16053,22 +14772,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16078,11 +14796,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -16098,14 +14812,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16133,7 +14839,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16142,17 +14848,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16169,7 +14871,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16188,21 +14889,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16212,11 +14912,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -16232,7 +14928,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16254,17 +14949,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16275,14 +14969,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16292,11 +14986,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16308,10 +15002,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -16327,7 +15017,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16349,17 +15038,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16370,14 +15058,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16387,11 +15075,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16403,8 +15091,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16416,7 +15102,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16438,17 +15123,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16459,14 +15143,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16476,11 +15160,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16492,10 +15176,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16513,22 +15193,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16538,11 +15217,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16559,14 +15234,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16594,7 +15261,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16603,17 +15270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16632,21 +15295,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16656,11 +15318,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16670,14 +15328,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16704,7 +15354,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16713,17 +15363,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16742,21 +15388,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16766,11 +15411,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16780,14 +15421,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16814,7 +15447,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16823,17 +15456,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16852,21 +15481,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16876,11 +15504,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16890,14 +15514,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16924,7 +15540,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16933,17 +15549,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16962,21 +15574,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16986,11 +15597,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17000,14 +15607,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17034,7 +15633,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17043,23 +15642,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17088,7 +15683,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17096,16 +15691,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17115,11 +15709,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17143,14 +15733,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17161,7 +15743,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17174,18 +15756,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17196,7 +15770,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages the virtual machine which has an ephemeral OS disk back to its initial state. @@ -17206,19 +15780,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17248,7 +15812,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17257,17 +15821,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17289,7 +15849,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17308,22 +15867,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17335,13 +15893,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17360,21 +15914,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17384,11 +15937,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17398,14 +15947,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17432,7 +15973,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17441,17 +15982,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17463,7 +16000,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17482,21 +16018,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17506,11 +16041,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17529,21 +16060,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17557,13 +16087,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17575,14 +16101,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17612,7 +16130,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17624,23 +16142,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17666,7 +16182,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17674,16 +16190,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17697,13 +16212,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17728,14 +16239,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17748,7 +16251,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17761,18 +16264,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17785,7 +16280,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -17795,20 +16290,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17841,7 +16325,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17853,20 +16337,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -17891,7 +16377,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17899,16 +16385,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17922,13 +16407,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -17951,14 +16432,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -17970,7 +16443,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17982,18 +16455,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -18002,7 +16467,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18011,19 +16480,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RunCommandResult] @@ -18055,7 +16513,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18067,17 +16525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -18116,7 +16572,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18135,7 +16590,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18143,16 +16598,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18164,13 +16618,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18202,7 +16652,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18221,7 +16670,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18231,16 +16680,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18252,13 +16700,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18270,7 +16714,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18289,21 +16732,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18315,13 +16757,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -18329,7 +16767,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18348,20 +16785,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18373,13 +16809,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18393,7 +16825,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18412,22 +16843,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18439,13 +16869,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -18455,7 +16881,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18474,21 +16899,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18500,13 +16924,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -18547,7 +16967,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18566,7 +16985,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18575,16 +16994,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18596,13 +17014,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18639,7 +17053,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18658,7 +17071,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18669,16 +17082,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18690,13 +17102,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18711,7 +17119,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18730,22 +17137,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18757,13 +17163,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -18775,7 +17177,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18794,21 +17195,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18820,13 +17220,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18843,7 +17239,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18862,23 +17257,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18890,13 +17284,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -18933,7 +17323,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18952,23 +17341,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18980,13 +17368,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -18998,7 +17382,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19017,21 +17400,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19043,13 +17425,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -19077,7 +17455,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19096,7 +17473,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -19105,16 +17482,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19126,13 +17502,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -19176,7 +17548,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19187,7 +17558,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19199,11 +17570,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19214,7 +17584,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -19224,12 +17594,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19257,7 +17623,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19265,16 +17631,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19286,13 +17651,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -19315,7 +17676,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19326,7 +17686,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19338,11 +17698,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19353,7 +17712,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -19363,12 +17722,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19396,7 +17751,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19404,16 +17759,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19425,13 +17779,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -19443,7 +17793,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19462,21 +17811,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19486,11 +17834,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -19500,7 +17844,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19519,21 +17862,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19545,13 +17887,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -19562,7 +17900,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19584,16 +17921,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19604,14 +17940,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19621,11 +17957,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19637,17 +17973,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19669,16 +18000,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19689,14 +18019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19706,11 +18036,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19722,10 +18052,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -19737,7 +18063,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -19759,17 +18084,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19780,14 +18104,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -19797,11 +18121,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19813,10 +18137,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -19860,7 +18180,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19871,7 +18190,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19884,11 +18203,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19899,7 +18217,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -19909,12 +18227,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19942,7 +18256,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -19950,16 +18264,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19979,10 +18292,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -20005,7 +18314,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20016,7 +18324,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20029,11 +18337,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20044,7 +18351,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -20054,12 +18361,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20087,7 +18391,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20095,16 +18399,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20116,13 +18419,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20134,7 +18433,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20153,21 +18451,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20177,11 +18474,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20201,7 +18494,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20220,22 +18512,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20247,19 +18538,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20283,15 +18569,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20302,14 +18587,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20319,11 +18604,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20335,10 +18620,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -20347,7 +18628,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20371,16 +18651,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20391,14 +18670,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20408,11 +18687,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20424,10 +18703,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -20472,7 +18747,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20483,7 +18757,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20497,11 +18771,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20512,7 +18785,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -20524,12 +18797,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20557,7 +18826,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20565,16 +18834,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20594,10 +18862,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -20619,7 +18883,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20630,7 +18893,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20642,11 +18905,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20657,7 +18919,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -20666,13 +18928,9 @@ def update( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20700,7 +18958,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20708,16 +18966,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20729,13 +18986,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20747,7 +19000,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20766,21 +19018,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20790,11 +19041,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20816,7 +19063,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20835,22 +19081,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20862,13 +19107,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -20877,7 +19118,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20899,16 +19139,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20919,14 +19158,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -20936,11 +19175,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20952,16 +19191,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20983,15 +19217,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21002,14 +19235,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21019,11 +19252,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21035,8 +19268,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -21063,7 +19294,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21089,7 +19320,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21098,16 +19329,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21127,10 +19357,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -21155,14 +19381,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21175,7 +19393,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21189,18 +19407,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21213,7 +19423,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -21225,19 +19435,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21270,7 +19469,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21280,24 +19479,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21323,7 +19520,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21332,16 +19529,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21353,13 +19549,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21385,14 +19577,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21405,7 +19589,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21419,18 +19603,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21443,7 +19619,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -21455,19 +19631,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] @@ -21500,7 +19665,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21510,17 +19675,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -21539,22 +19702,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21564,11 +19726,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21582,14 +19740,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21617,7 +19767,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21626,17 +19776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21660,7 +19806,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -21679,23 +19824,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21707,13 +19851,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -21726,7 +19866,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -21748,17 +19887,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21769,14 +19907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -21786,11 +19924,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21802,10 +19940,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -21823,22 +19957,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21848,11 +19981,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -21869,14 +19998,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21904,7 +20025,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21913,17 +20034,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -21951,7 +20068,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -21975,15 +20091,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21994,14 +20109,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22011,11 +20126,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22027,8 +20142,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -22038,7 +20151,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22062,16 +20174,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22082,14 +20193,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22099,11 +20210,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22115,10 +20226,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -22140,7 +20247,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22151,7 +20257,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22163,11 +20269,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22178,7 +20283,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -22188,12 +20293,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22221,7 +20322,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22229,16 +20330,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22258,10 +20358,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -22283,7 +20379,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22294,7 +20389,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22306,11 +20401,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22321,7 +20415,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -22331,12 +20425,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22364,7 +20455,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22372,16 +20463,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22393,13 +20483,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22411,7 +20497,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22430,21 +20515,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22454,11 +20538,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -22468,7 +20548,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22487,21 +20566,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22513,13 +20591,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -22533,7 +20607,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22552,21 +20625,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22578,13 +20650,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -22608,7 +20676,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22633,7 +20701,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22641,16 +20709,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22670,10 +20737,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -22695,14 +20758,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22713,7 +20768,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22725,18 +20780,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22744,7 +20791,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -22753,19 +20800,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22796,7 +20832,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22806,20 +20842,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22844,7 +20882,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22852,16 +20890,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22881,10 +20918,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -22906,14 +20939,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22924,7 +20949,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22936,18 +20961,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22955,7 +20972,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -22964,19 +20985,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23007,7 +21017,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23017,17 +21027,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -23046,21 +21054,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23070,11 +21077,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23084,14 +21087,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23118,7 +21113,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23127,17 +21122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23151,7 +21142,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -23170,22 +21160,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23197,13 +21186,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -23211,7 +21196,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23233,16 +21217,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23253,14 +21236,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23270,11 +21253,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23286,16 +21269,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23317,15 +21295,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23336,14 +21313,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23353,11 +21330,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23369,8 +21346,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23416,7 +21391,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23427,7 +21401,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23442,11 +21416,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23457,7 +21430,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23469,12 +21442,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23502,7 +21471,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23510,16 +21479,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23539,10 +21507,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23565,7 +21529,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23576,7 +21539,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23589,11 +21552,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23604,7 +21566,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23614,12 +21576,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23647,7 +21606,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23655,16 +21614,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23676,13 +21634,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -23701,21 +21655,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23725,11 +21678,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23742,14 +21691,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23776,7 +21717,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23785,17 +21726,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23817,7 +21754,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23836,22 +21772,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23863,13 +21798,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -23877,7 +21808,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23901,16 +21831,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23921,14 +21850,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -23938,11 +21867,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23954,17 +21883,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23988,15 +21912,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24007,14 +21930,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24024,11 +21947,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24040,8 +21963,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -24068,7 +21989,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -24094,7 +22015,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -24103,16 +22024,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24124,13 +22044,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -24157,14 +22073,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -24177,7 +22085,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24192,18 +22100,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -24216,7 +22116,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -24229,19 +22129,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_01.models.RestorePoint] @@ -24274,7 +22163,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24284,17 +22173,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -24313,22 +22200,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24338,11 +22224,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24356,14 +22238,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24391,7 +22265,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24400,17 +22274,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24433,7 +22303,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24452,23 +22321,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24480,13 +22348,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -24532,7 +22396,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24543,7 +22406,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24557,11 +22420,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24572,7 +22434,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -24584,12 +22446,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24617,7 +22475,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24625,16 +22483,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24654,10 +22511,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -24681,7 +22534,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24692,7 +22544,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24706,11 +22558,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24721,7 +22572,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -24732,12 +22583,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24765,7 +22613,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24773,16 +22621,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24794,13 +22641,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -24815,7 +22658,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -24834,21 +22676,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24858,11 +22699,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -24886,7 +22723,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24905,22 +22741,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24932,13 +22767,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -24959,7 +22790,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -24983,17 +22813,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25004,14 +22833,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25021,11 +22850,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25037,10 +22866,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -25054,7 +22879,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25078,16 +22902,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25098,14 +22921,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25115,11 +22938,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25131,10 +22954,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -25161,7 +22980,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -25187,7 +23006,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25196,16 +23015,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25225,10 +23043,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -25255,14 +23069,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25276,7 +23082,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25292,18 +23098,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25317,7 +23115,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -25331,19 +23129,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25377,7 +23164,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25387,24 +23174,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -25430,7 +23215,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25439,16 +23224,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25462,13 +23246,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25494,14 +23274,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25515,7 +23287,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25529,18 +23301,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25554,7 +23318,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -25566,19 +23330,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25612,7 +23365,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25622,17 +23375,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25655,22 +23406,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25680,11 +23430,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25704,14 +23450,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25739,7 +23477,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25748,17 +23486,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25783,7 +23517,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -25802,23 +23535,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25830,13 +23562,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -25849,7 +23577,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_01.models.CapacityReservation] @@ -25872,17 +23599,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25893,14 +23619,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -25910,11 +23636,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25926,10 +23652,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -25952,7 +23674,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -25977,23 +23699,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26007,13 +23728,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26035,14 +23752,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26052,7 +23761,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26061,18 +23770,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26082,7 +23783,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26090,19 +23791,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26134,7 +23825,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26146,20 +23837,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26184,23 +23873,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26214,13 +23902,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26241,14 +23925,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26258,7 +23934,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26266,18 +23942,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26287,7 +23955,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26295,19 +23963,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_08_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26339,7 +23996,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26351,17 +24008,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26390,7 +24045,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26414,16 +24068,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26434,14 +24087,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26451,11 +24104,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26467,10 +24120,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26479,7 +24128,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26498,21 +24146,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26524,20 +24171,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26563,7 +24206,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26572,16 +24215,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26601,10 +24243,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26631,14 +24269,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26652,7 +24282,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26668,18 +24298,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26693,7 +24315,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -26706,19 +24328,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26752,7 +24363,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26762,24 +24373,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26805,7 +24414,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26814,16 +24423,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26835,13 +24443,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26869,14 +24473,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26890,7 +24486,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26906,18 +24502,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26931,7 +24519,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -26944,19 +24532,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26990,7 +24568,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27000,17 +24578,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27029,22 +24605,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27054,11 +24629,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27073,14 +24644,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27108,7 +24671,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27117,17 +24680,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27149,7 +24708,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27168,23 +24726,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27196,13 +24753,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27216,7 +24769,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27240,18 +24792,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27262,14 +24813,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27279,11 +24830,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27295,12 +24846,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27326,7 +24873,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27352,7 +24899,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27362,16 +24909,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27391,10 +24937,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27423,14 +24965,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27445,7 +24979,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27462,18 +24996,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27488,7 +25014,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27502,19 +25028,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27549,7 +25064,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27559,17 +25074,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27577,7 +25090,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27603,7 +25116,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27613,16 +25126,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27634,13 +25146,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27670,14 +25178,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27692,7 +25192,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27709,18 +25209,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27735,7 +25227,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -27749,19 +25241,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27796,7 +25278,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27806,17 +25288,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -27835,23 +25315,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27861,11 +25340,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27881,14 +25356,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27917,7 +25384,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27926,17 +25393,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27961,7 +25424,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27980,7 +25442,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27988,16 +25450,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28009,13 +25470,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28037,7 +25494,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28061,19 +25517,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28084,14 +25539,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28101,11 +25556,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28116,7 +25571,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_compute_management_client.py index 70dcb9c511ca..ab7b84ef29ae 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -95,7 +97,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -138,7 +158,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-08-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -158,7 +178,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_configuration.py index c7370ae67794..ac16da701e76 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-08-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_metadata.json index 324837b653f0..71817824e491 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "galleries": "GalleriesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_compute_management_client.py index c0abcf0a3457..9d96effce0d1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -96,7 +98,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -139,7 +159,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-08-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -159,7 +181,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_configuration.py index 144674b54f4c..14da25f019ce 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-08-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/operations/_operations.py index 5c0bbc2d2622..2bb5e66bc170 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -162,10 +161,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -190,14 +185,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -209,7 +196,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -224,18 +211,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -244,7 +223,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -255,19 +234,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_08_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_08_03.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -299,7 +267,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -309,20 +277,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -347,7 +317,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -355,16 +325,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,13 +345,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -407,14 +372,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -426,7 +383,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -440,18 +397,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -460,7 +409,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -471,19 +424,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_08_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_08_03.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -515,7 +457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -525,17 +467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -559,7 +499,6 @@ async def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -578,23 +517,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -606,13 +544,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -631,21 +565,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -655,11 +588,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -669,14 +598,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -703,7 +624,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -712,17 +633,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -730,7 +647,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -752,16 +668,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -772,14 +687,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -789,11 +704,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -805,15 +720,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -835,15 +745,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -854,14 +763,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -871,11 +780,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -887,8 +796,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -915,7 +822,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -941,7 +848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -950,16 +857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,10 +888,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -1014,14 +916,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1034,7 +928,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1052,18 +946,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1076,7 +962,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -1091,19 +977,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1136,7 +1011,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1146,24 +1021,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1189,7 +1062,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1198,16 +1071,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1219,13 +1091,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1254,14 +1122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1274,7 +1134,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1291,18 +1151,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1315,7 +1167,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -1330,19 +1182,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1375,7 +1216,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1385,17 +1226,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1410,7 +1249,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1429,22 +1267,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1456,13 +1293,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -1481,22 +1314,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1506,11 +1338,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1525,14 +1353,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1560,7 +1380,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1569,17 +1389,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -1592,7 +1408,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -1615,17 +1430,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1636,14 +1450,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -1653,11 +1467,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,10 +1483,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -1700,7 +1510,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1726,7 +1536,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1736,16 +1546,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1768,10 +1577,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -1805,14 +1610,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1827,7 +1624,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1849,18 +1646,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1875,7 +1664,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -1894,19 +1683,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1941,7 +1720,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1951,17 +1730,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -1969,7 +1746,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1995,7 +1772,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2005,16 +1782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2026,13 +1802,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2067,14 +1839,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2089,7 +1853,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2111,18 +1875,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2137,7 +1893,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -2156,20 +1912,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2204,7 +1949,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2214,17 +1959,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2253,7 +1996,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -2272,7 +2014,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2280,16 +2022,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2301,13 +2042,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -2331,23 +2068,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2357,11 +2093,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2385,14 +2117,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2421,7 +2145,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2430,17 +2154,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -2456,7 +2176,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion] @@ -2479,18 +2198,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2501,14 +2219,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -2518,11 +2236,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2534,10 +2252,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -2564,7 +2278,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2590,7 +2304,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2599,16 +2313,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2631,10 +2344,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -2663,14 +2372,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2684,7 +2385,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2702,18 +2403,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2727,7 +2420,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -2742,19 +2435,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2788,7 +2471,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2798,24 +2481,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2841,7 +2522,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2850,16 +2531,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2871,13 +2551,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2907,14 +2583,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2928,7 +2596,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2946,18 +2614,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2971,7 +2631,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -2986,20 +2646,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3033,7 +2682,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3043,17 +2692,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3069,7 +2716,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3088,22 +2734,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3115,13 +2760,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -3140,22 +2781,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3165,11 +2805,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3185,14 +2821,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3220,7 +2848,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3229,17 +2857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -3252,7 +2876,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryApplication] @@ -3275,17 +2898,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3296,14 +2918,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -3313,11 +2935,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3329,10 +2951,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -3360,7 +2978,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3386,7 +3004,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3396,16 +3014,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3428,10 +3045,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -3466,14 +3079,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3488,7 +3093,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3510,18 +3115,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3536,7 +3133,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -3555,21 +3152,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3604,7 +3190,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3614,17 +3200,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -3632,7 +3216,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3658,7 +3242,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3668,16 +3252,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3689,13 +3272,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3731,14 +3310,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3753,7 +3324,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3775,18 +3346,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3801,7 +3364,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -3820,20 +3383,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3868,7 +3421,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3878,17 +3431,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3917,7 +3468,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3936,7 +3486,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3944,16 +3494,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3965,13 +3514,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3995,23 +3540,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4021,11 +3565,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4049,14 +3589,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4085,7 +3617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4094,17 +3626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -4120,7 +3648,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4144,18 +3671,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4166,14 +3692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -4183,11 +3709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4199,10 +3725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -4228,7 +3750,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -4254,7 +3776,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4262,16 +3784,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4291,10 +3812,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -4316,14 +3833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4336,7 +3845,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4348,18 +3857,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4372,7 +3873,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -4382,19 +3883,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_08_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_08_03.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4427,7 +3917,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4437,17 +3927,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -4481,7 +3969,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.SharedGallery] @@ -4504,17 +3991,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4525,14 +4011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -4542,11 +4028,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4558,10 +4044,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -4570,7 +4052,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,21 +4070,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4615,13 +4095,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -4662,7 +4138,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImage] @@ -4685,18 +4160,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4707,14 +4181,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -4724,11 +4198,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4740,10 +4214,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -4757,7 +4227,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4776,22 +4245,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4803,13 +4271,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -4854,7 +4318,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -4878,19 +4341,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4901,14 +4363,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -4918,11 +4380,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4934,10 +4396,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -4961,7 +4419,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4980,23 +4437,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5008,13 +4464,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -5045,7 +4497,6 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5064,21 +4515,20 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5090,13 +4540,9 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -5131,7 +4577,6 @@ async def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5150,22 +4595,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4621,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4635,6 @@ def list( :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -5219,17 +4658,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -5257,11 +4695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,10 +4711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -5320,7 +4754,6 @@ async def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5339,23 +4772,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5367,13 +4799,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5387,7 +4815,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -5411,18 +4838,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5433,14 +4859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -5450,11 +4876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5465,7 +4891,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/models/_models_py3.py index 60941e824f47..56fcc6878bc4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/models/_models_py3.py @@ -400,7 +400,7 @@ def __init__( class CommunityGalleryImageList(_serialization.Model): """The List Community Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImage] @@ -527,7 +527,7 @@ def __init__( class CommunityGalleryImageVersionList(_serialization.Model): """The List Community Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImageVersion] @@ -629,7 +629,7 @@ def __init__( class CommunityGalleryMetadata(_serialization.Model): """The metadata of community gallery. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher_uri: The publisher URI of this community gallery. :vartype publisher_uri: str @@ -713,7 +713,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -838,7 +838,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -887,7 +887,7 @@ class Gallery(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -981,7 +981,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1091,7 +1091,7 @@ def __init__( class GalleryApplicationCustomAction(_serialization.Model): """A custom action that can be performed with a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1149,7 +1149,7 @@ class GalleryApplicationCustomActionParameter(_serialization.Model): """The definition of a parameter that can be passed to a custom action of a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1217,7 +1217,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.GalleryApplication] @@ -1399,7 +1399,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1480,7 +1480,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion] @@ -1613,12 +1613,12 @@ def __init__( class GalleryApplicationVersionPublishingProfile( GalleryArtifactPublishingProfileBase -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1796,18 +1796,6 @@ class GalleryApplicationVersionSafetyProfile(GalleryArtifactSafetyProfileBase): :vartype allow_deletion_of_replicated_locations: bool """ - _attribute_map = { - "allow_deletion_of_replicated_locations": {"key": "allowDeletionOfReplicatedLocations", "type": "bool"}, - } - - def __init__(self, *, allow_deletion_of_replicated_locations: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword allow_deletion_of_replicated_locations: Indicates whether or not removing this Gallery - Image Version from replicated regions is allowed. - :paramtype allow_deletion_of_replicated_locations: bool - """ - super().__init__(allow_deletion_of_replicated_locations=allow_deletion_of_replicated_locations, **kwargs) - class GalleryApplicationVersionUpdate(UpdateResourceDefinition): """Specifies information about the gallery Application Version that you want to update. @@ -1886,7 +1874,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2022_08_03.models.ManagedArtifact @@ -2015,7 +2003,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -2170,7 +2158,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2369,7 +2357,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -2409,7 +2397,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2611,7 +2599,7 @@ class GalleryImageVersion(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2705,7 +2693,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion] @@ -2771,71 +2759,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) list[~azure.mgmt.compute.v2022_08_03.models.GalleryTargetExtendedLocation] """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - "target_extended_locations": {"key": "targetExtendedLocations", "type": "[GalleryTargetExtendedLocation]"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - target_extended_locations: Optional[List["_models.GalleryTargetExtendedLocation"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2022_08_03.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2022_08_03.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2022_08_03.models.ReplicationMode - :keyword target_extended_locations: The target extended locations where the Image Version is - going to be replicated to. This property is updatable. - :paramtype target_extended_locations: - list[~azure.mgmt.compute.v2022_08_03.models.GalleryTargetExtendedLocation] - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - target_extended_locations=target_extended_locations, - **kwargs - ) - class GalleryImageVersionSafetyProfile(GalleryArtifactSafetyProfileBase): """This is the safety profile of the Gallery Image Version. @@ -3042,7 +2965,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.Gallery] @@ -3087,32 +3010,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2022_08_03.models.GalleryDiskImageSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryDiskImageSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryDiskImageSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_08_03.models.HostCaching - :keyword source: The source for the disk image. - :paramtype source: ~azure.mgmt.compute.v2022_08_03.models.GalleryDiskImageSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryTargetExtendedLocation(_serialization.Model): """GalleryTargetExtendedLocation. @@ -3382,7 +3279,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3864,7 +3761,7 @@ class SharedGalleryDataDiskImage(SharedGalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_size_gb: This property indicates the size of the VHD to be created. :vartype disk_size_gb: int @@ -4053,7 +3950,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImage] @@ -4169,7 +4066,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImageVersion] @@ -4239,7 +4136,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2022_08_03.models.SharedGallery] @@ -4282,25 +4179,6 @@ class SharedGalleryOSDiskImage(SharedGalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryHostCaching """ - _validation = { - "disk_size_gb": {"readonly": True}, - } - - _attribute_map = { - "disk_size_gb": {"key": "diskSizeGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__( - self, *, host_caching: Optional[Union[str, "_models.SharedGalleryHostCaching"]] = None, **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryHostCaching - """ - super().__init__(host_caching=host_caching, **kwargs) - class SharingProfile(_serialization.Model): """Profile for gallery sharing to subscription or tenant. @@ -4420,7 +4298,7 @@ def __init__(self, *, summary: Optional[List["_models.RegionalSharingStatus"]] = class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. Possible values are: **Add,** **Remove,** **Reset.**. Required. Known values are: @@ -4559,7 +4437,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -4708,7 +4586,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -4792,7 +4670,7 @@ def __init__( class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/operations/_operations.py index 429fd2e54f4e..794aa40a6bd4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_08_03/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -179,7 +179,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -233,7 +233,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -367,7 +367,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -398,7 +398,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -439,7 +439,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -522,7 +522,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -560,7 +560,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +592,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +627,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -694,7 +694,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -726,7 +726,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -757,7 +757,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -800,7 +800,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -843,7 +843,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -887,7 +887,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -927,7 +927,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -959,7 +959,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1130,7 +1130,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1170,7 +1170,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1239,7 +1239,7 @@ def build_community_galleries_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_get_request( +def build_community_gallery_images_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1271,7 +1271,7 @@ def build_community_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_list_request( +def build_community_gallery_images_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1302,7 +1302,7 @@ def build_community_gallery_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_get_request( +def build_community_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, @@ -1340,7 +1340,7 @@ def build_community_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_list_request( +def build_community_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1393,7 +1393,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1418,7 +1418,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1426,16 +1426,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1458,10 +1457,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -1486,14 +1481,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1504,7 +1491,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1519,18 +1506,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1538,7 +1517,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -1549,19 +1528,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_08_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_08_03.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1592,7 +1560,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1602,20 +1570,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1640,7 +1610,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1648,16 +1618,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,13 +1638,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -1700,14 +1665,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,7 +1675,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1732,18 +1689,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1751,7 +1700,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -1762,19 +1715,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2022_08_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2022_08_03.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1805,7 +1747,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1815,17 +1757,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -1849,7 +1789,6 @@ def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -1868,23 +1807,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1896,13 +1834,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -1921,21 +1855,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1945,11 +1878,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1959,14 +1888,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1993,7 +1914,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2002,17 +1923,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -2020,7 +1937,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2042,16 +1958,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2062,14 +1977,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2079,11 +1994,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2095,15 +2010,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2125,15 +2035,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2144,14 +2053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2161,11 +2070,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2177,8 +2086,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -2205,7 +2112,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2231,7 +2138,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2240,16 +2147,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,10 +2178,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -2304,14 +2206,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2324,7 +2218,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2342,18 +2236,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2366,7 +2252,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -2381,19 +2267,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2426,7 +2301,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,24 +2311,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2479,7 +2352,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2488,16 +2361,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2509,13 +2381,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2544,14 +2412,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2564,7 +2424,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2581,18 +2441,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2605,7 +2457,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -2620,19 +2472,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] @@ -2665,7 +2506,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2675,17 +2516,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -2700,7 +2539,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2719,22 +2557,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2746,13 +2583,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -2771,22 +2604,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2796,11 +2628,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -2815,14 +2643,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2850,7 +2670,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2859,17 +2679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -2882,7 +2698,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2904,17 +2719,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2925,14 +2739,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -2942,11 +2756,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,10 +2772,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -2989,7 +2799,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3015,7 +2825,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3025,16 +2835,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3057,10 +2866,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -3094,14 +2899,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3116,7 +2913,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3138,18 +2935,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3164,7 +2953,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -3183,19 +2972,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3230,7 +3009,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3240,17 +3019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -3258,7 +3035,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3284,7 +3061,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3294,16 +3071,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3315,13 +3091,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3356,14 +3128,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3378,7 +3142,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3164,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3426,7 +3182,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -3445,20 +3201,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3493,7 +3238,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3503,17 +3248,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3542,7 +3285,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3561,7 +3303,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3569,16 +3311,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3590,13 +3331,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3620,23 +3357,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3646,11 +3382,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -3674,14 +3406,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3710,7 +3434,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3719,17 +3443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -3745,7 +3465,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryImageVersion] @@ -3768,18 +3487,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3790,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -3807,11 +3525,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3823,10 +3541,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -3853,7 +3567,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3879,7 +3593,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3888,16 +3602,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3920,10 +3633,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -3952,14 +3661,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3973,7 +3674,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3991,18 +3692,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4016,7 +3709,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -4031,19 +3724,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4077,7 +3760,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4087,24 +3770,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -4130,7 +3811,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4139,16 +3820,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4160,13 +3840,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4196,14 +3872,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4217,7 +3885,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4235,18 +3903,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4260,7 +3920,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -4275,20 +3935,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4322,7 +3971,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4332,17 +3981,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4358,7 +4005,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -4377,22 +4023,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4404,13 +4049,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -4429,22 +4070,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4454,11 +4094,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4474,14 +4110,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4509,7 +4137,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4518,17 +4146,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -4541,7 +4165,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -4563,17 +4186,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4584,14 +4206,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -4601,11 +4223,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4617,10 +4239,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -4648,7 +4266,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4674,7 +4292,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4684,16 +4302,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4716,10 +4333,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -4754,14 +4367,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4776,7 +4381,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4798,18 +4403,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4824,7 +4421,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -4843,21 +4440,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4892,7 +4478,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4902,17 +4488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -4920,7 +4504,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4946,7 +4530,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4956,16 +4540,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4977,13 +4560,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5019,14 +4598,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5041,7 +4612,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5063,18 +4634,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5089,7 +4652,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -5108,20 +4671,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5156,7 +4709,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5166,17 +4719,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5205,7 +4756,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_08_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5224,7 +4774,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -5232,16 +4782,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5253,13 +4802,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -5283,23 +4828,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5309,11 +4853,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5337,14 +4877,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5373,7 +4905,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5382,17 +4914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -5408,7 +4936,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5432,18 +4959,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5454,14 +4980,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -5471,11 +4997,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5487,10 +5013,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -5516,7 +5038,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -5542,7 +5064,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5550,16 +5072,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5579,10 +5100,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -5604,14 +5121,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.SharingUpdate] @@ -5623,7 +5132,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5635,18 +5144,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.SharingUpdate] @@ -5658,7 +5159,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -5668,19 +5169,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2022_08_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2022_08_03.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_08_03.models.SharingUpdate] @@ -5712,7 +5202,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5722,17 +5212,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -5766,7 +5254,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5788,17 +5275,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5809,14 +5295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -5826,11 +5312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5842,10 +5328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -5854,7 +5336,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5873,21 +5354,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5899,13 +5379,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -5946,7 +5422,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5968,18 +5443,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5990,14 +5464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -6007,11 +5481,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6023,10 +5497,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -6040,7 +5510,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6059,22 +5528,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6086,13 +5554,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -6137,7 +5601,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2022_08_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -6161,19 +5624,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6184,14 +5646,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -6201,11 +5663,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6217,10 +5679,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -6244,7 +5702,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6263,23 +5720,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6291,13 +5747,9 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -6328,7 +5780,6 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -6347,21 +5798,20 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6373,13 +5823,9 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -6414,7 +5860,6 @@ def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6433,22 +5878,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6460,13 +5904,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterable["_models.CommunityGalleryImage"]: @@ -6476,7 +5916,6 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -6500,17 +5939,16 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6521,14 +5959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -6538,11 +5976,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6554,10 +5992,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -6601,7 +6035,6 @@ def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_08_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6620,23 +6053,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-08-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,13 +6080,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6668,7 +6096,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -6692,18 +6119,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6714,14 +6140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -6731,11 +6157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6746,7 +6172,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_compute_management_client.py index ec10ce97531d..de9588f66344 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -69,7 +71,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -91,7 +111,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-09-04" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -111,7 +131,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_configuration.py index ccdecf48b881..24dda0302a53 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-09-04") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_metadata.json index c8aa4b7bad23..49ebd8bf2ba8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "cloud_service_role_instances": "CloudServiceRoleInstancesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_compute_management_client.py index 4cb7010de9b1..70b2f93719b3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -69,7 +71,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -91,7 +111,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-09-04" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -111,7 +133,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_configuration.py index 51650e170769..31275a357b8d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-09-04") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/operations/_operations.py index e4df8c2ed79b..92d12b6e451b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -105,22 +105,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -130,11 +129,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -148,14 +143,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -183,7 +170,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -192,17 +179,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -225,7 +208,6 @@ async def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_09_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -244,23 +226,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -272,13 +253,9 @@ async def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -292,7 +269,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -311,22 +287,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -338,13 +313,9 @@ async def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -366,7 +337,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_09_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.RoleInstance] @@ -389,18 +359,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -411,14 +380,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -428,11 +397,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -444,10 +413,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -465,22 +430,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -490,11 +454,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -509,14 +469,6 @@ async def begin_restart( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -544,7 +496,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -553,17 +505,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -582,22 +530,21 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -607,11 +554,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage( @@ -626,14 +569,6 @@ async def begin_reimage( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -661,7 +596,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -670,17 +605,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -699,22 +630,21 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -724,11 +654,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_rebuild( @@ -744,14 +670,6 @@ async def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -779,7 +697,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -788,17 +706,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_remote_desktop_file( @@ -812,8 +726,7 @@ async def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Async iterator of the response bytes or the result of cls(response) + :return: AsyncIterator[bytes] or the result of cls(response) :rtype: AsyncIterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -831,22 +744,21 @@ async def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -862,10 +774,6 @@ async def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -899,7 +807,6 @@ async def get( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -918,22 +825,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -945,13 +851,9 @@ async def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -964,7 +866,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudServiceRole] @@ -987,17 +888,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1008,14 +908,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1025,11 +925,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1041,10 +941,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -1070,7 +966,7 @@ async def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1099,7 +995,7 @@ async def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1107,16 +1003,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1136,10 +1031,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload async def begin_create_or_update( self, @@ -1162,14 +1053,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1181,7 +1064,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1194,18 +1077,10 @@ async def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1217,7 +1092,7 @@ async def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -1227,20 +1102,9 @@ async def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudService or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1272,7 +1136,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1282,23 +1146,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -1327,7 +1189,7 @@ async def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -1335,16 +1197,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1356,13 +1217,9 @@ async def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1385,14 +1242,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1404,7 +1253,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1416,18 +1265,10 @@ async def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1439,7 +1280,7 @@ async def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[_models.CloudService]: """Update a cloud service. @@ -1448,20 +1289,9 @@ async def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1493,7 +1323,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1503,17 +1333,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1532,21 +1360,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1556,11 +1383,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1572,14 +1395,6 @@ async def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1606,7 +1421,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1615,17 +1430,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -1635,7 +1446,6 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -1654,21 +1464,20 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1680,13 +1489,9 @@ async def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -1698,7 +1503,6 @@ async def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1717,21 +1521,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1743,13 +1546,9 @@ async def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: @@ -1757,7 +1556,6 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1780,15 +1578,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1799,14 +1596,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1816,11 +1613,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1832,8 +1629,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -1842,7 +1637,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -1865,16 +1659,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1885,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -1902,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1918,10 +1711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1939,21 +1728,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1963,11 +1751,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -1979,14 +1763,6 @@ async def begin_start( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2013,7 +1789,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2022,17 +1798,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2051,21 +1823,20 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2075,11 +1846,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -2092,14 +1859,6 @@ async def begin_power_off( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2126,7 +1885,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2135,23 +1894,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2180,7 +1935,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2188,16 +1943,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2207,11 +1961,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2234,14 +1984,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2252,7 +1994,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2264,18 +2006,10 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2286,7 +2020,7 @@ async def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -2296,19 +2030,8 @@ async def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2338,7 +2061,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2347,23 +2070,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2392,7 +2111,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2400,16 +2119,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2419,11 +2137,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -2447,14 +2161,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2465,7 +2171,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2478,18 +2184,10 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2500,7 +2198,7 @@ async def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -2511,19 +2209,8 @@ async def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2553,7 +2240,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2562,23 +2249,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2607,7 +2290,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2615,16 +2298,15 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2634,11 +2316,7 @@ async def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_rebuild( @@ -2663,14 +2341,6 @@ async def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2681,7 +2351,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2695,18 +2365,10 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2717,7 +2379,7 @@ async def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -2729,19 +2391,8 @@ async def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2771,7 +2422,7 @@ async def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2780,23 +2431,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2825,7 +2472,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2833,16 +2480,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2852,11 +2498,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -2879,14 +2521,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2897,7 +2531,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2909,18 +2543,10 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2931,7 +2557,7 @@ async def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes role instances in a cloud service. @@ -2941,19 +2567,8 @@ async def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2983,7 +2598,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2992,17 +2607,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3030,7 +2641,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3059,7 +2670,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3068,16 +2679,15 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3087,11 +2697,7 @@ async def _walk_update_domain_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_walk_update_domain( @@ -3119,14 +2725,6 @@ async def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3138,7 +2736,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3154,18 +2752,10 @@ async def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3177,7 +2767,7 @@ async def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Updates the role instances in the specified update domain. @@ -3190,20 +2780,9 @@ async def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3234,7 +2813,7 @@ async def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3243,17 +2822,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_update_domain( @@ -3271,7 +2846,6 @@ async def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -3290,22 +2864,21 @@ async def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3317,13 +2890,9 @@ async def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -3335,7 +2904,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.UpdateDomain] @@ -3358,17 +2926,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3379,14 +2946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -3396,11 +2963,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3412,10 +2979,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -3446,7 +3009,6 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3465,21 +3027,20 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3491,13 +3052,9 @@ async def get_os_version(self, location: str, os_version_name: str, **kwargs: An deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSVersion"]: @@ -3507,7 +3064,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.OSVersion] @@ -3530,16 +3086,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3550,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -3567,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3583,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace_async async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -3596,7 +3147,6 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -3615,21 +3165,20 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3641,13 +3190,9 @@ async def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_models.OSFamily"]: @@ -3657,7 +3202,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_09_04.models.OSFamily] @@ -3680,16 +3224,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3700,14 +3243,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -3717,11 +3260,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3732,7 +3275,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/models/_models_py3.py index 63fb6842b265..29d0bc6c7ead 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/models/_models_py3.py @@ -115,7 +115,7 @@ class CloudService(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -388,7 +388,7 @@ def __init__(self, *, role_instance: Optional["_models.InstanceViewStatusesSumma class CloudServiceListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -705,7 +705,7 @@ def __init__( class CloudServiceRoleListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.CloudServiceRole] @@ -1108,7 +1108,7 @@ def __init__(self, **kwargs: Any) -> None: class LoadBalancerConfiguration(_serialization.Model): """Describes the load balancer configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1155,7 +1155,7 @@ def __init__( class LoadBalancerConfigurationProperties(_serialization.Model): """Describes the properties of the load balancer configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar frontend_ip_configurations: Specifies the frontend IP to be used for the load balancer. Only IPv4 frontend IP address is supported. Each load balancer configuration must have exactly @@ -1193,7 +1193,7 @@ class LoadBalancerFrontendIpConfiguration(_serialization.Model): """Specifies the frontend IP to be used for the load balancer. Only IPv4 frontend IP address is supported. Each load balancer configuration must have exactly one frontend IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the resource that is unique within the set of frontend IP configurations used by the load balancer. This name can be used to access the resource. @@ -1231,7 +1231,7 @@ def __init__( self.properties = properties -class LoadBalancerFrontendIpConfigurationProperties(_serialization.Model): +class LoadBalancerFrontendIpConfigurationProperties(_serialization.Model): # pylint: disable=name-too-long """Describes a cloud service IP Configuration. :ivar public_ip_address: The reference to the public ip address resource. @@ -1318,7 +1318,7 @@ def __init__(self, *, properties: Optional["_models.OSFamilyProperties"] = None, class OSFamilyListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.OSFamily] @@ -1430,7 +1430,7 @@ def __init__(self, *, properties: Optional["_models.OSVersionProperties"] = None class OSVersionListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.OSVersion] @@ -1552,7 +1552,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1752,7 +1752,7 @@ def __init__( class RoleInstanceListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.RoleInstance] @@ -1842,7 +1842,7 @@ def __init__( class RoleInstances(_serialization.Model): """Specifies a list of role instances from the cloud service. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar role_instances: List of cloud service role instance names. Value of '*' will signify all role instances of the cloud service. Required. @@ -2039,7 +2039,7 @@ def __init__(self, **kwargs: Any) -> None: class UpdateDomainListResult(_serialization.Model): """The list operation result. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of resources. Required. :vartype value: list[~azure.mgmt.compute.v2022_09_04.models.UpdateDomain] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/operations/_operations.py index 22c7f29856de..39ca5690c602 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_09_04/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -39,7 +39,7 @@ _SERIALIZER.client_side_validation = False -def build_cloud_service_role_instances_delete_request( +def build_cloud_service_role_instances_delete_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -71,7 +71,7 @@ def build_cloud_service_role_instances_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_request( +def build_cloud_service_role_instances_get_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, @@ -111,7 +111,7 @@ def build_cloud_service_role_instances_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_instance_view_request( +def build_cloud_service_role_instances_get_instance_view_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -143,7 +143,7 @@ def build_cloud_service_role_instances_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_list_request( +def build_cloud_service_role_instances_list_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, @@ -181,7 +181,7 @@ def build_cloud_service_role_instances_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_restart_request( +def build_cloud_service_role_instances_restart_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -213,7 +213,7 @@ def build_cloud_service_role_instances_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_reimage_request( +def build_cloud_service_role_instances_reimage_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -245,7 +245,7 @@ def build_cloud_service_role_instances_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_rebuild_request( +def build_cloud_service_role_instances_rebuild_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -277,7 +277,7 @@ def build_cloud_service_role_instances_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_role_instances_get_remote_desktop_file_request( +def build_cloud_service_role_instances_get_remote_desktop_file_request( # pylint: disable=name-too-long role_instance_name: str, resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -372,7 +372,7 @@ def build_cloud_service_roles_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_create_or_update_request( +def build_cloud_services_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -502,7 +502,7 @@ def build_cloud_services_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_get_instance_view_request( +def build_cloud_services_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -749,7 +749,7 @@ def build_cloud_services_rebuild_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_delete_instances_request( +def build_cloud_services_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -783,7 +783,7 @@ def build_cloud_services_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_walk_update_domain_request( +def build_cloud_services_update_domain_walk_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -818,7 +818,7 @@ def build_cloud_services_update_domain_walk_update_domain_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_get_update_domain_request( +def build_cloud_services_update_domain_get_update_domain_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, update_domain: int, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -850,7 +850,7 @@ def build_cloud_services_update_domain_get_update_domain_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_services_update_domain_list_update_domains_request( +def build_cloud_services_update_domain_list_update_domains_request( # pylint: disable=name-too-long resource_group_name: str, cloud_service_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -881,7 +881,7 @@ def build_cloud_services_update_domain_list_update_domains_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_version_request( +def build_cloud_service_operating_systems_get_os_version_request( # pylint: disable=name-too-long location: str, os_version_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -912,7 +912,7 @@ def build_cloud_service_operating_systems_get_os_version_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_versions_request( +def build_cloud_service_operating_systems_list_os_versions_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -942,7 +942,7 @@ def build_cloud_service_operating_systems_list_os_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_get_os_family_request( +def build_cloud_service_operating_systems_get_os_family_request( # pylint: disable=name-too-long location: str, os_family_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -973,7 +973,7 @@ def build_cloud_service_operating_systems_get_os_family_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_cloud_service_operating_systems_list_os_families_request( +def build_cloud_service_operating_systems_list_os_families_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1040,22 +1040,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_delete_request( + _request = build_cloud_service_role_instances_delete_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1065,11 +1064,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -1083,14 +1078,6 @@ def begin_delete( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1118,7 +1105,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1127,17 +1114,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -1160,7 +1143,6 @@ def get( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_09_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstance or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.RoleInstance :raises ~azure.core.exceptions.HttpResponseError: @@ -1179,23 +1161,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.RoleInstance] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_request( + _request = build_cloud_service_role_instances_get_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1207,13 +1188,9 @@ def get( deserialized = self._deserialize("RoleInstance", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -1227,7 +1204,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RoleInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.RoleInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1246,22 +1222,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.RoleInstanceView] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_instance_view_request( + _request = build_cloud_service_role_instances_get_instance_view_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1273,13 +1248,9 @@ def get_instance_view( deserialized = self._deserialize("RoleInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1301,7 +1272,6 @@ def list( :keyword expand: The expand expression to apply to the operation. 'UserData' is not supported for cloud services. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_09_04.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RoleInstance or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.RoleInstance] :raises ~azure.core.exceptions.HttpResponseError: @@ -1323,18 +1293,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_role_instances_list_request( + _request = build_cloud_service_role_instances_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1345,14 +1314,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RoleInstanceListResult", pipeline_response) @@ -1362,11 +1331,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1378,10 +1347,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -1399,22 +1364,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_restart_request( + _request = build_cloud_service_role_instances_restart_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1424,11 +1388,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -1443,14 +1403,6 @@ def begin_restart( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1478,7 +1430,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1487,17 +1439,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1516,22 +1464,21 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_reimage_request( + _request = build_cloud_service_role_instances_reimage_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1541,11 +1488,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage( @@ -1560,14 +1503,6 @@ def begin_reimage( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1595,7 +1530,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1604,17 +1539,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, role_instance_name: str, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -1633,22 +1564,21 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_rebuild_request( + _request = build_cloud_service_role_instances_rebuild_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1658,11 +1588,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_rebuild( @@ -1678,14 +1604,6 @@ def begin_rebuild( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1713,7 +1631,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1722,17 +1640,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_remote_desktop_file( @@ -1746,8 +1660,7 @@ def get_remote_desktop_file( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :return: Iterator of the response bytes or the result of cls(response) + :return: Iterator[bytes] or the result of cls(response) :rtype: Iterator[bytes] :raises ~azure.core.exceptions.HttpResponseError: """ @@ -1765,22 +1678,21 @@ def get_remote_desktop_file( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[Iterator[bytes]] = kwargs.pop("cls", None) - request = build_cloud_service_role_instances_get_remote_desktop_file_request( + _request = build_cloud_service_role_instances_get_remote_desktop_file_request( role_instance_name=role_instance_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_remote_desktop_file.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = True pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1796,10 +1708,6 @@ def get_remote_desktop_file( return deserialized # type: ignore - get_remote_desktop_file.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roleInstances/{roleInstanceName}/remoteDesktopFile" - } - class CloudServiceRolesOperations: """ @@ -1833,7 +1741,6 @@ def get( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceRole or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceRole :raises ~azure.core.exceptions.HttpResponseError: @@ -1852,22 +1759,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudServiceRole] = kwargs.pop("cls", None) - request = build_cloud_service_roles_get_request( + _request = build_cloud_service_roles_get_request( role_name=role_name, resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1879,13 +1785,9 @@ def get( deserialized = self._deserialize("CloudServiceRole", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles/{roleName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -1898,7 +1800,6 @@ def list( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudServiceRole or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudServiceRole] :raises ~azure.core.exceptions.HttpResponseError: @@ -1920,17 +1821,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_roles_list_request( + _request = build_cloud_service_roles_list_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1941,14 +1841,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceRoleListResult", pipeline_response) @@ -1958,11 +1858,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1974,10 +1874,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/roles" - } - class CloudServicesOperations: # pylint: disable=too-many-public-methods """ @@ -2003,7 +1899,7 @@ def _create_or_update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -2032,7 +1928,7 @@ def _create_or_update_initial( else: _json = None - request = build_cloud_services_create_or_update_request( + _request = build_cloud_services_create_or_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2040,16 +1936,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2069,10 +1964,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } - @overload def begin_create_or_update( self, @@ -2095,14 +1986,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2114,7 +1997,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2127,18 +2010,10 @@ def begin_create_or_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2150,7 +2025,7 @@ def begin_create_or_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudService, IO]] = None, + parameters: Optional[Union[_models.CloudService, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Create or update a cloud service. Please note some properties can be set only during cloud @@ -2160,20 +2035,9 @@ def begin_create_or_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudService type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudService or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. + :param parameters: The cloud service object. Is either a CloudService type or a IO[bytes] type. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudService or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2205,7 +2069,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2215,23 +2079,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> _models.CloudService: error_map = { @@ -2260,7 +2122,7 @@ def _update_initial( else: _json = None - request = build_cloud_services_update_request( + _request = build_cloud_services_update_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -2268,16 +2130,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2289,13 +2150,9 @@ def _update_initial( deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2318,14 +2175,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2337,7 +2186,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2349,18 +2198,10 @@ def begin_update( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: The cloud service object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2372,7 +2213,7 @@ def begin_update( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.CloudServiceUpdate, IO]] = None, + parameters: Optional[Union[_models.CloudServiceUpdate, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[_models.CloudService]: """Update a cloud service. @@ -2381,20 +2222,9 @@ def begin_update( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO type. - Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The cloud service object. Is either a CloudServiceUpdate type or a IO[bytes] + type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceUpdate or IO[bytes] :return: An instance of LROPoller that returns either CloudService or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_09_04.models.CloudService] @@ -2426,7 +2256,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,17 +2266,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CloudService].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[_models.CloudService]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2465,21 +2293,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_delete_request( + _request = build_cloud_services_delete_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2489,11 +2316,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2503,14 +2326,6 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2537,7 +2352,7 @@ def begin_delete(self, resource_group_name: str, cloud_service_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2546,17 +2361,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> _models.CloudService: @@ -2566,7 +2377,6 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudService or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudService :raises ~azure.core.exceptions.HttpResponseError: @@ -2585,21 +2395,20 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudService] = kwargs.pop("cls", None) - request = build_cloud_services_get_request( + _request = build_cloud_services_get_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2611,13 +2420,9 @@ def get(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) deserialized = self._deserialize("CloudService", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -2629,7 +2434,6 @@ def get_instance_view( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CloudServiceInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.CloudServiceInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -2648,21 +2452,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.CloudServiceInstanceView] = kwargs.pop("cls", None) - request = build_cloud_services_get_instance_view_request( + _request = build_cloud_services_get_instance_view_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2674,13 +2477,9 @@ def get_instance_view( deserialized = self._deserialize("CloudServiceInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: @@ -2688,7 +2487,6 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: group. Use nextLink property in the response to get the next page of Cloud Services. Do this till nextLink is null to fetch all the Cloud Services. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -2710,15 +2508,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.CloudService"]: def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_all_request( + _request = build_cloud_services_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2729,14 +2526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2746,11 +2543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2762,8 +2559,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/cloudServices"} - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.CloudService"]: """Gets a list of all cloud services under a resource group. Use nextLink property in the response @@ -2772,7 +2567,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo :param resource_group_name: Name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CloudService or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.CloudService] :raises ~azure.core.exceptions.HttpResponseError: @@ -2794,16 +2588,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Clo def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_list_request( + _request = build_cloud_services_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2814,14 +2607,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CloudServiceListResult", pipeline_response) @@ -2831,11 +2624,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2847,10 +2640,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices" - } - def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any ) -> None: @@ -2868,21 +2657,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_start_request( + _request = build_cloud_services_start_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2892,11 +2680,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2906,14 +2690,6 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2940,7 +2716,7 @@ def begin_start(self, resource_group_name: str, cloud_service_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2949,17 +2725,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, **kwargs: Any @@ -2978,21 +2750,20 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_cloud_services_power_off_request( + _request = build_cloud_services_power_off_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3002,11 +2773,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **kwargs: Any) -> LROPoller[None]: @@ -3017,14 +2784,6 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3051,7 +2810,7 @@ def begin_power_off(self, resource_group_name: str, cloud_service_name: str, **k def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3060,23 +2819,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3105,7 +2860,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_restart_request( + _request = build_cloud_services_restart_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3113,16 +2868,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3132,11 +2886,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -3159,14 +2909,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3177,7 +2919,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3189,18 +2931,10 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3211,7 +2945,7 @@ def begin_restart( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more role instances in a cloud service. @@ -3221,19 +2955,8 @@ def begin_restart( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3263,7 +2986,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3272,23 +2995,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3317,7 +3036,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_reimage_request( + _request = build_cloud_services_reimage_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3325,16 +3044,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3344,11 +3062,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -3372,14 +3086,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3390,7 +3096,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3403,18 +3109,10 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3425,7 +3123,7 @@ def begin_reimage( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimage asynchronous operation reinstalls the operating system on instances of web roles or @@ -3436,19 +3134,8 @@ def begin_reimage( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3478,7 +3165,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3487,23 +3174,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _rebuild_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3532,7 +3215,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_rebuild_request( + _request = build_cloud_services_rebuild_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3540,16 +3223,15 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._rebuild_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3559,11 +3241,7 @@ def _rebuild_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _rebuild_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_rebuild( @@ -3588,14 +3266,6 @@ def begin_rebuild( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3606,7 +3276,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3620,18 +3290,10 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3642,7 +3304,7 @@ def begin_rebuild( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Rebuild Role Instances reinstalls the operating system on instances of web roles or worker @@ -3654,19 +3316,8 @@ def begin_rebuild( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3696,7 +3347,7 @@ def begin_rebuild( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3705,23 +3356,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_rebuild.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/rebuild" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3750,7 +3397,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_cloud_services_delete_instances_request( + _request = build_cloud_services_delete_instances_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, @@ -3758,16 +3405,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3777,11 +3423,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -3804,14 +3446,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3822,7 +3456,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3834,18 +3468,10 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3856,7 +3482,7 @@ def begin_delete_instances( self, resource_group_name: str, cloud_service_name: str, - parameters: Optional[Union[_models.RoleInstances, IO]] = None, + parameters: Optional[Union[_models.RoleInstances, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deletes role instances in a cloud service. @@ -3866,19 +3492,8 @@ def begin_delete_instances( :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str :param parameters: List of cloud service role instance names. Is either a RoleInstances type or - a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.RoleInstances or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3908,7 +3523,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3917,17 +3532,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class CloudServicesUpdateDomainOperations: @@ -3955,7 +3566,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3984,7 +3595,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen else: _json = None - request = build_cloud_services_update_domain_walk_update_domain_request( + _request = build_cloud_services_update_domain_walk_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, @@ -3993,16 +3604,15 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen content_type=content_type, json=_json, content=_content, - template_url=self._walk_update_domain_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4012,11 +3622,7 @@ def _walk_update_domain_initial( # pylint: disable=inconsistent-return-statemen raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _walk_update_domain_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_walk_update_domain( @@ -4044,14 +3650,6 @@ def begin_walk_update_domain( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4063,7 +3661,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4079,18 +3677,10 @@ def begin_walk_update_domain( second has an ID of 1, and so on. Required. :type update_domain: int :param parameters: The update domain object. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4102,7 +3692,7 @@ def begin_walk_update_domain( resource_group_name: str, cloud_service_name: str, update_domain: int, - parameters: Optional[Union[_models.UpdateDomain, IO]] = None, + parameters: Optional[Union[_models.UpdateDomain, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Updates the role instances in the specified update domain. @@ -4115,20 +3705,9 @@ def begin_walk_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :param parameters: The update domain object. Is either a UpdateDomain type or a IO type. + :param parameters: The update domain object. Is either a UpdateDomain type or a IO[bytes] type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :type parameters: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4159,7 +3738,7 @@ def begin_walk_update_domain( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4168,17 +3747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_walk_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_update_domain( @@ -4196,7 +3771,6 @@ def get_update_domain( domains are identified with a zero-based index: the first update domain has an ID of 0, the second has an ID of 1, and so on. Required. :type update_domain: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: UpdateDomain or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.UpdateDomain :raises ~azure.core.exceptions.HttpResponseError: @@ -4215,22 +3789,21 @@ def get_update_domain( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.UpdateDomain] = kwargs.pop("cls", None) - request = build_cloud_services_update_domain_get_update_domain_request( + _request = build_cloud_services_update_domain_get_update_domain_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, update_domain=update_domain, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_update_domain.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4242,13 +3815,9 @@ def get_update_domain( deserialized = self._deserialize("UpdateDomain", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_update_domain.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains/{updateDomain}" - } + return deserialized # type: ignore @distributed_trace def list_update_domains( @@ -4260,7 +3829,6 @@ def list_update_domains( :type resource_group_name: str :param cloud_service_name: Name of the cloud service. Required. :type cloud_service_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpdateDomain or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.UpdateDomain] :raises ~azure.core.exceptions.HttpResponseError: @@ -4282,17 +3850,16 @@ def list_update_domains( def prepare_request(next_link=None): if not next_link: - request = build_cloud_services_update_domain_list_update_domains_request( + _request = build_cloud_services_update_domain_list_update_domains_request( resource_group_name=resource_group_name, cloud_service_name=cloud_service_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_update_domains.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4303,14 +3870,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("UpdateDomainListResult", pipeline_response) @@ -4320,11 +3887,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4336,10 +3903,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_update_domains.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/cloudServices/{cloudServiceName}/updateDomains" - } - class CloudServiceOperatingSystemsOperations: """ @@ -4370,7 +3933,6 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> :type location: str :param os_version_name: Name of the OS version. Required. :type os_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.OSVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4389,21 +3951,20 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.OSVersion] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_version_request( + _request = build_cloud_service_operating_systems_get_os_version_request( location=location, os_version_name=os_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_version.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4415,13 +3976,9 @@ def get_os_version(self, location: str, os_version_name: str, **kwargs: Any) -> deserialized = self._deserialize("OSVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_version.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions/{osVersionName}" - } + return deserialized # type: ignore @distributed_trace def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OSVersion"]: @@ -4431,7 +3988,6 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS versions pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.OSVersion] :raises ~azure.core.exceptions.HttpResponseError: @@ -4453,16 +4009,15 @@ def list_os_versions(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_versions_request( + _request = build_cloud_service_operating_systems_list_os_versions_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4473,14 +4028,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSVersionListResult", pipeline_response) @@ -4490,11 +4045,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4506,10 +4061,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_os_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsVersions" - } - @distributed_trace def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _models.OSFamily: """Gets properties of a guest operating system family that can be specified in the XML service @@ -4519,7 +4070,6 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m :type location: str :param os_family_name: Name of the OS family. Required. :type os_family_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: OSFamily or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_09_04.models.OSFamily :raises ~azure.core.exceptions.HttpResponseError: @@ -4538,21 +4088,20 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-09-04")) cls: ClsType[_models.OSFamily] = kwargs.pop("cls", None) - request = build_cloud_service_operating_systems_get_os_family_request( + _request = build_cloud_service_operating_systems_get_os_family_request( location=location, os_family_name=os_family_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_family.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4564,13 +4113,9 @@ def get_os_family(self, location: str, os_family_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("OSFamily", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_os_family.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies/{osFamilyName}" - } + return deserialized # type: ignore @distributed_trace def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OSFamily"]: @@ -4580,7 +4125,6 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS :param location: Name of the location that the OS families pertain to. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either OSFamily or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_09_04.models.OSFamily] :raises ~azure.core.exceptions.HttpResponseError: @@ -4602,16 +4146,15 @@ def list_os_families(self, location: str, **kwargs: Any) -> Iterable["_models.OS def prepare_request(next_link=None): if not next_link: - request = build_cloud_service_operating_systems_list_os_families_request( + _request = build_cloud_service_operating_systems_list_os_families_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_os_families.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4622,14 +4165,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("OSFamilyListResult", pipeline_response) @@ -4639,11 +4182,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4654,7 +4197,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list_os_families.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/cloudServiceOsFamilies" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_compute_management_client.py index 91651b663641..e807bf560721 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_configuration.py index 1bb2f3e4274b..bfae89d71b9e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_metadata.json index 86e2adc1e430..5452e9979377 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_compute_management_client.py index 0233b6959150..0993517b6206 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2022-11-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_configuration.py index 7499e65f993d..6eee2e47ad0a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2022-11-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/operations/_operations.py index b26f5beef7d7..00245b882a6d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -234,7 +234,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -258,14 +257,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -276,14 +274,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -293,11 +291,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -309,8 +307,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -339,7 +335,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -361,16 +356,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -381,14 +375,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -398,11 +392,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -414,8 +408,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -444,7 +436,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] @@ -467,16 +458,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -487,14 +477,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -504,11 +494,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -520,8 +510,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -550,7 +538,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -574,16 +561,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -594,14 +580,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -611,11 +597,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -627,15 +613,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -661,7 +643,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -669,16 +651,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -698,10 +679,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -723,14 +700,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -743,7 +712,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -755,18 +724,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -779,7 +740,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -788,20 +749,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -834,7 +784,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -844,23 +794,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -886,7 +834,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -894,16 +842,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -915,13 +862,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -944,14 +887,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -964,7 +899,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -976,18 +911,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1000,7 +927,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1009,20 +936,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1055,7 +972,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1065,17 +982,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1094,22 +1009,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1119,11 +1033,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1138,14 +1048,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1173,7 +1075,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1182,17 +1084,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1213,7 +1111,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1232,22 +1129,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1259,19 +1155,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -1300,7 +1192,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1308,16 +1200,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1327,11 +1218,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1358,14 +1245,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1376,7 +1255,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -1391,18 +1270,10 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1413,7 +1284,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -1425,20 +1296,9 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -1468,7 +1328,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1477,23 +1337,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1521,7 +1377,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1530,16 +1386,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1549,11 +1404,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1582,14 +1433,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1600,7 +1443,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1614,21 +1457,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1639,7 +1474,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1651,23 +1486,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1698,7 +1522,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1707,17 +1531,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1729,7 +1549,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1748,21 +1567,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1774,13 +1592,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1788,7 +1602,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1812,16 +1625,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1832,14 +1644,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1849,11 +1661,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1865,17 +1677,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1899,15 +1706,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1918,14 +1724,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -1935,11 +1741,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1951,8 +1757,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -1964,7 +1768,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -1988,17 +1791,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2009,14 +1811,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2026,11 +1828,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2042,10 +1844,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2056,7 +1854,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2080,17 +1877,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2101,14 +1897,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2118,11 +1914,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2134,15 +1930,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2173,7 +1965,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2182,16 +1974,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2201,11 +1992,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2237,14 +2024,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2255,7 +2034,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2271,7 +2050,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2279,14 +2058,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2297,7 +2068,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2311,24 +2082,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2359,7 +2119,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2368,23 +2128,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2413,7 +2169,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2421,16 +2177,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2440,11 +2195,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2469,14 +2220,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2487,7 +2230,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2500,18 +2243,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2522,7 +2257,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2532,20 +2267,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2575,7 +2299,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2584,23 +2308,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2629,7 +2349,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2637,16 +2357,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2656,11 +2375,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2685,14 +2400,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2703,7 +2410,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2716,18 +2423,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2738,7 +2437,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2748,20 +2447,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2791,7 +2479,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2800,23 +2488,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2845,7 +2529,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2853,16 +2537,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2872,11 +2555,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -2902,14 +2581,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2920,7 +2591,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2934,18 +2605,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2956,7 +2619,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -2967,20 +2630,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3010,7 +2662,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3019,23 +2671,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3064,7 +2712,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3072,16 +2720,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3091,11 +2738,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3123,14 +2766,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3141,7 +2776,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3157,18 +2792,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3179,7 +2806,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3192,20 +2819,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3235,7 +2851,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3244,23 +2860,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3286,7 +2898,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3294,16 +2906,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3313,11 +2924,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3342,14 +2949,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3360,7 +2959,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3373,18 +2972,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3395,7 +2986,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3405,20 +2996,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3448,7 +3028,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3457,23 +3037,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3502,7 +3078,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3510,16 +3086,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3529,11 +3104,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3559,14 +3130,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3577,7 +3140,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3591,18 +3154,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3613,7 +3168,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3625,20 +3180,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3668,7 +3212,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3677,23 +3221,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3722,7 +3262,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3730,16 +3270,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3749,11 +3288,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -3779,14 +3314,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3797,7 +3324,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3811,18 +3338,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3833,7 +3352,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -3844,20 +3363,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3887,7 +3395,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3896,20 +3404,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -3935,7 +3439,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -3954,7 +3457,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3962,16 +3465,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3983,13 +3485,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4014,7 +3512,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4025,7 +3522,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4038,11 +3535,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4053,7 +3549,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4064,13 +3560,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4098,7 +3591,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4106,16 +3599,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4125,17 +3617,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4161,7 +3649,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4169,16 +3657,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4188,11 +3675,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4216,14 +3699,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4234,7 +3709,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4247,18 +3722,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4269,7 +3736,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4280,19 +3747,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4322,7 +3779,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4331,20 +3788,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4369,7 +3822,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4395,7 +3848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4404,16 +3857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4433,10 +3885,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4464,14 +3912,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4485,7 +3925,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4501,18 +3941,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4526,7 +3958,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4539,20 +3971,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4586,7 +4007,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4596,24 +4017,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4639,7 +4058,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4648,16 +4067,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4677,10 +4095,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -4708,14 +4122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4729,7 +4135,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4745,18 +4151,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4770,7 +4168,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -4783,20 +4181,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4830,7 +4218,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4840,17 +4228,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -4869,22 +4255,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4894,11 +4279,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4913,14 +4294,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4948,7 +4321,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4957,17 +4330,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -4989,7 +4358,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5008,23 +4376,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5036,13 +4403,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5054,7 +4417,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5078,17 +4440,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5099,14 +4460,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5116,11 +4477,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5132,12 +4493,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5174,21 +4531,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5198,11 +4554,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5214,14 +4566,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5248,7 +4592,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5257,17 +4601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5286,21 +4626,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5310,11 +4649,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5328,14 +4663,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5362,7 +4689,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5371,17 +4698,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5400,21 +4723,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5424,11 +4746,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5442,14 +4760,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5476,7 +4786,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5485,17 +4795,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5507,7 +4813,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5526,21 +4831,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5552,16 +4856,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5587,7 +4887,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5613,7 +4913,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5623,16 +4923,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5652,10 +4951,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -5685,14 +4980,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5707,7 +4994,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5724,18 +5011,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5750,7 +5029,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -5764,20 +5043,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5812,7 +5080,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5822,17 +5090,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -5840,7 +5106,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5866,7 +5132,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5876,16 +5142,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5897,13 +5162,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -5934,14 +5195,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5956,7 +5209,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5973,18 +5226,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5999,7 +5244,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6013,20 +5258,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6061,7 +5296,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6071,17 +5306,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6100,23 +5333,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6126,11 +5358,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6146,14 +5374,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6182,7 +5402,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6191,17 +5411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6226,7 +5442,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6245,7 +5460,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6253,16 +5468,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6274,13 +5488,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6302,7 +5512,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6321,23 +5530,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6349,13 +5557,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6383,7 +5587,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6412,7 +5618,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6421,16 +5627,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6440,11 +5645,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6472,14 +5673,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6491,7 +5684,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6506,18 +5699,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6529,7 +5714,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6541,20 +5728,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6585,7 +5762,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6594,17 +5771,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6623,22 +5796,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,11 +5820,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -6667,14 +5835,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6702,7 +5862,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6711,17 +5871,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6740,22 +5896,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6765,11 +5920,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6785,14 +5936,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6820,7 +5963,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6829,24 +5972,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -6872,7 +6011,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6881,16 +6020,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6910,10 +6048,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -6940,14 +6074,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -6961,7 +6087,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6977,18 +6103,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7002,7 +6120,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -7015,19 +6133,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7061,7 +6168,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7071,17 +6178,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7106,23 +6211,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7132,11 +6236,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7159,14 +6259,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7195,7 +6287,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7204,17 +6296,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7238,7 +6326,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7257,23 +6344,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7285,13 +6371,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7305,7 +6387,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7324,22 +6405,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7351,13 +6431,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7386,7 +6462,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7410,7 +6485,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7418,12 +6493,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7434,14 +6508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7451,11 +6525,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7467,10 +6541,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -7494,23 +6564,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7520,11 +6589,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7550,14 +6615,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7586,7 +6643,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7595,17 +6652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7624,22 +6677,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7649,11 +6701,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -7667,14 +6715,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7702,7 +6742,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7711,17 +6751,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7740,22 +6776,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7765,11 +6800,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -7783,14 +6814,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7818,7 +6841,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7827,17 +6850,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7856,22 +6875,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7881,11 +6899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -7900,14 +6914,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7935,7 +6941,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7944,17 +6950,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -7980,7 +6982,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -7999,23 +7000,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8027,13 +7027,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8052,22 +7048,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8077,11 +7072,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8095,14 +7086,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8130,7 +7113,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8139,17 +7122,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8163,7 +7142,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8182,22 +7160,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8207,18 +7184,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -8244,7 +7217,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8253,16 +7226,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8276,13 +7248,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8308,14 +7276,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8329,7 +7289,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8343,18 +7303,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8368,7 +7320,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -8380,19 +7332,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8426,7 +7367,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8438,17 +7379,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -8476,7 +7415,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8502,7 +7441,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8511,16 +7450,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8540,10 +7478,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8570,14 +7504,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8591,7 +7517,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8607,18 +7533,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8632,7 +7550,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8645,20 +7563,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8692,7 +7599,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8702,24 +7609,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8745,7 +7650,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8754,16 +7659,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8775,13 +7679,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8810,14 +7710,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8831,7 +7723,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8847,18 +7739,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8872,7 +7756,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -8885,20 +7769,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8932,7 +7805,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8942,17 +7815,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -8971,22 +7842,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8996,11 +7866,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9015,14 +7881,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9050,7 +7908,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9059,17 +7917,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9091,7 +7945,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9110,23 +7963,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9138,13 +7990,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9158,7 +8006,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9177,22 +8024,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9204,13 +8050,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -9240,7 +8082,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -9263,16 +8104,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9283,14 +8123,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9300,11 +8140,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9316,15 +8156,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9350,7 +8186,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9358,16 +8194,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9381,13 +8216,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9411,14 +8242,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9431,7 +8254,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9444,18 +8267,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9468,7 +8283,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9479,19 +8294,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9524,7 +8329,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9536,20 +8341,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9574,7 +8381,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9582,16 +8389,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9611,10 +8417,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9637,14 +8439,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9657,7 +8451,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9670,18 +8464,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9691,7 +8477,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9701,19 +8491,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9746,7 +8525,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9756,20 +8535,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9794,7 +8575,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9802,16 +8583,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9823,13 +8603,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9852,14 +8628,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9872,7 +8640,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9884,18 +8652,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9905,7 +8665,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -9914,19 +8678,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9959,7 +8712,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9969,17 +8722,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -9998,22 +8749,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10023,11 +8773,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10042,14 +8788,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10077,7 +8815,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10086,17 +8824,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10119,7 +8853,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10138,22 +8871,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10165,13 +8897,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10183,7 +8911,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10202,21 +8929,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10228,13 +8954,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10253,21 +8975,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10277,11 +8998,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10294,14 +9011,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10328,7 +9037,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10337,17 +9046,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -10366,22 +9071,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10391,11 +9095,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10411,14 +9111,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10446,7 +9138,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10455,17 +9147,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10482,7 +9170,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10501,21 +9188,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10525,11 +9211,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -10545,7 +9227,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -10568,17 +9249,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10589,14 +9269,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10606,11 +9286,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10622,10 +9302,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -10641,7 +9317,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -10664,17 +9339,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10685,14 +9359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10702,11 +9376,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10718,8 +9392,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10731,7 +9403,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] @@ -10754,17 +9425,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10775,14 +9445,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10792,11 +9462,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10808,10 +9478,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10829,22 +9495,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10854,11 +9519,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -10875,14 +9536,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10910,7 +9563,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10919,17 +9572,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10948,21 +9597,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10972,11 +9620,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -10986,14 +9630,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11020,7 +9656,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11029,17 +9665,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11058,21 +9690,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11082,11 +9713,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11096,14 +9723,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11130,7 +9749,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11139,17 +9758,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11168,21 +9783,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11192,11 +9806,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11206,14 +9816,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11240,7 +9842,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11249,17 +9851,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11278,21 +9876,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11302,11 +9899,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11316,14 +9909,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11350,7 +9935,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11359,23 +9944,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11404,7 +9985,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11412,16 +9993,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11431,11 +10011,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11464,14 +10040,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11482,7 +10050,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11500,18 +10068,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11522,7 +10082,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -11537,19 +10097,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11579,7 +10129,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11588,17 +10138,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11620,7 +10166,6 @@ async def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11639,22 +10184,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11666,13 +10210,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11691,21 +10231,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11715,11 +10254,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11731,14 +10266,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11765,7 +10292,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11774,17 +10301,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11796,7 +10319,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11815,21 +10337,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11839,11 +10360,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11862,21 +10379,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11890,13 +10406,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -11908,14 +10420,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -11945,7 +10449,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11957,23 +10461,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -11999,7 +10501,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12007,16 +10509,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12030,13 +10531,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12061,14 +10558,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12081,7 +10570,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12094,18 +10583,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12118,7 +10599,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12128,20 +10609,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12174,7 +10644,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12186,20 +10656,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12224,7 +10696,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12232,16 +10704,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12255,13 +10726,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12284,14 +10751,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12304,7 +10763,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12316,18 +10775,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12337,7 +10788,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12346,19 +10801,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12391,7 +10835,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12403,17 +10847,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -12452,7 +10894,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12471,7 +10912,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12479,16 +10920,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12500,13 +10940,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12538,7 +10974,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12557,7 +10992,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12567,16 +11002,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12588,13 +11022,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -12606,7 +11036,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12625,21 +11054,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12651,13 +11079,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -12665,7 +11089,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12684,20 +11107,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12709,13 +11131,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -12729,7 +11147,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12748,22 +11165,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12775,13 +11191,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -12793,7 +11205,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12812,21 +11223,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12838,13 +11248,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -12885,7 +11291,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12904,7 +11309,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -12913,16 +11318,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12934,13 +11338,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12977,7 +11377,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12996,7 +11395,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -13007,16 +11406,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13028,13 +11426,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13049,7 +11443,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13068,22 +11461,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13095,13 +11487,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -13113,7 +11501,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13132,21 +11519,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13158,13 +11544,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13181,7 +11563,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13200,23 +11581,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13228,13 +11608,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13271,7 +11647,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13290,23 +11665,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13318,13 +11692,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -13336,7 +11706,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13355,21 +11724,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13381,13 +11749,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -13415,7 +11779,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13434,7 +11797,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13443,16 +11806,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13464,13 +11826,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -13514,7 +11872,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13525,7 +11882,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13537,11 +11894,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13552,7 +11908,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -13562,12 +11918,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13595,7 +11947,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13603,16 +11955,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13624,13 +11975,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -13653,7 +12000,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13664,7 +12010,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13676,11 +12022,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13691,7 +12036,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -13701,12 +12046,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13734,7 +12075,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13742,16 +12083,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13763,13 +12103,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -13781,7 +12117,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13800,21 +12135,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13824,11 +12158,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -13838,7 +12168,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13857,21 +12186,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13883,13 +12211,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -13900,7 +12224,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet] @@ -13923,16 +12246,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13943,14 +12265,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -13960,11 +12282,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13976,17 +12298,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet] @@ -14009,16 +12326,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14029,14 +12345,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14046,11 +12362,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14062,10 +12378,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -14077,7 +12389,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] @@ -14100,17 +12411,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14121,14 +12431,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -14138,11 +12448,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14154,10 +12464,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -14201,7 +12507,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14212,7 +12517,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14225,11 +12530,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14240,7 +12544,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -14250,12 +12554,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14283,7 +12583,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14291,16 +12591,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14320,10 +12619,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -14346,7 +12641,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14357,7 +12651,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14370,11 +12664,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14385,7 +12678,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -14395,12 +12688,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14428,7 +12718,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14436,16 +12726,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14457,13 +12746,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14475,7 +12760,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14494,21 +12778,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14518,11 +12801,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14542,7 +12821,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14561,22 +12839,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14588,19 +12865,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14624,15 +12896,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14643,14 +12914,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14660,11 +12931,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14676,10 +12947,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14688,7 +12955,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14712,16 +12978,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14732,14 +12997,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14749,11 +13014,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14765,10 +13030,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -14813,7 +13074,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14824,7 +13084,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14838,11 +13098,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14853,7 +13112,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -14865,12 +13124,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14898,7 +13153,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -14906,16 +13161,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14935,10 +13189,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -14960,7 +13210,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14971,7 +13220,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14983,11 +13232,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14998,7 +13246,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -15008,12 +13256,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15041,7 +13285,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15049,16 +13293,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15070,13 +13313,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15088,7 +13327,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15107,21 +13345,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15131,11 +13368,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15157,7 +13390,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15176,22 +13408,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15203,13 +13434,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15220,7 +13447,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup] @@ -15243,16 +13469,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15263,14 +13488,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15280,11 +13505,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15296,16 +13521,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup] @@ -15328,15 +13548,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15347,14 +13566,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15364,11 +13583,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15380,8 +13599,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -15408,7 +13625,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15434,7 +13651,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15443,16 +13660,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15472,10 +13688,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -15500,14 +13712,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15521,7 +13725,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15535,18 +13739,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15560,7 +13756,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -15572,19 +13768,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15618,7 +13803,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15628,24 +13813,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15671,7 +13854,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15680,16 +13863,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15701,13 +13883,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15733,14 +13911,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15754,7 +13924,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15768,18 +13938,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15793,7 +13955,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -15805,19 +13967,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15851,7 +14002,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15861,17 +14012,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -15890,22 +14039,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15915,11 +14063,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -15933,14 +14077,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15968,7 +14104,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -15977,17 +14113,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16011,7 +14143,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -16030,23 +14161,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16058,13 +14188,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -16077,7 +14203,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -16100,17 +14225,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16121,14 +14245,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -16138,11 +14262,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16154,10 +14278,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -16175,22 +14295,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16200,11 +14319,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -16221,14 +14336,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16256,7 +14363,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16265,17 +14372,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -16303,7 +14406,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16327,15 +14429,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16346,14 +14447,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16363,11 +14464,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16379,8 +14480,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -16390,7 +14489,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16414,16 +14512,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16434,14 +14531,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16451,11 +14548,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16467,10 +14564,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -16492,7 +14585,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16503,7 +14595,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16515,11 +14607,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16530,7 +14621,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -16540,12 +14631,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16573,7 +14660,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16581,16 +14668,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16610,10 +14696,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -16635,7 +14717,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16646,7 +14727,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16658,11 +14739,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16673,7 +14753,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -16683,12 +14763,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16716,7 +14793,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16724,16 +14801,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16745,13 +14821,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -16763,7 +14835,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16782,21 +14853,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16806,11 +14876,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -16822,7 +14888,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16841,21 +14906,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16867,13 +14931,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -16887,7 +14947,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -16906,21 +14965,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16932,13 +14990,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -16962,7 +15016,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -16987,7 +15041,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -16995,16 +15049,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17024,10 +15077,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17049,14 +15098,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17067,7 +15108,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17079,18 +15120,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17098,7 +15131,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -17107,19 +15140,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17150,7 +15172,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17160,20 +15182,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17198,7 +15222,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17206,16 +15230,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17235,10 +15258,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -17260,14 +15279,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17278,7 +15289,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17290,18 +15301,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17309,7 +15312,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -17318,19 +15325,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17361,7 +15357,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17371,17 +15367,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -17400,21 +15394,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17424,11 +15417,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17438,14 +15427,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17472,7 +15453,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17481,17 +15462,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17505,7 +15482,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -17524,22 +15500,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17551,13 +15526,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -17565,7 +15536,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17587,16 +15557,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17607,14 +15576,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17624,11 +15593,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17640,16 +15609,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17671,15 +15635,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17690,14 +15653,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17707,11 +15670,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17723,8 +15686,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -17770,7 +15731,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17781,7 +15741,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17796,11 +15756,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17811,7 +15770,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -17823,12 +15782,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17856,7 +15811,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -17864,16 +15819,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17893,10 +15847,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -17919,7 +15869,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17930,7 +15879,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17943,11 +15892,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -17958,7 +15906,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -17968,12 +15916,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18001,7 +15946,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -18009,16 +15954,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18030,13 +15974,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -18055,21 +15995,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18079,11 +16018,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18096,14 +16031,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18130,7 +16057,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18139,17 +16066,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18171,7 +16094,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18190,22 +16112,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18217,13 +16138,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -18231,7 +16148,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18255,16 +16171,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18275,14 +16190,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18292,11 +16207,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18308,17 +16223,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18342,15 +16252,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18361,14 +16270,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18378,11 +16287,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18394,8 +16303,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -18422,7 +16329,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -18448,7 +16355,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -18457,16 +16364,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18478,13 +16384,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -18511,14 +16413,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -18531,7 +16425,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18546,18 +16440,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -18570,7 +16456,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -18583,19 +16469,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -18628,7 +16503,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18638,17 +16513,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18667,22 +16540,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18692,11 +16564,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18710,14 +16578,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18745,7 +16605,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18754,17 +16614,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18787,7 +16643,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -18806,23 +16661,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18834,13 +16688,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -18886,7 +16736,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18897,7 +16746,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18911,11 +16760,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18926,7 +16774,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -18938,12 +16786,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -18971,7 +16815,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -18979,16 +16823,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19008,10 +16851,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -19035,7 +16874,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19046,7 +16884,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19060,11 +16898,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19075,7 +16912,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -19086,12 +16923,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19119,7 +16953,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19127,16 +16961,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19148,13 +16981,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -19169,7 +16998,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19188,21 +17016,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19212,11 +17039,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -19240,7 +17063,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19259,22 +17081,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19286,13 +17107,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -19313,7 +17130,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19337,17 +17153,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19358,14 +17173,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19375,11 +17190,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19391,10 +17206,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -19408,7 +17219,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19432,16 +17242,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19452,14 +17261,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19469,11 +17278,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19485,10 +17294,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -19515,7 +17320,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -19541,7 +17346,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19550,16 +17355,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19579,10 +17383,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -19609,14 +17409,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19630,7 +17422,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19646,18 +17438,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19671,7 +17455,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -19685,19 +17469,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19731,7 +17504,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19741,24 +17514,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -19784,7 +17555,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19793,16 +17564,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19816,13 +17586,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -19848,14 +17614,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19869,7 +17627,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19883,18 +17641,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19908,7 +17658,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -19920,19 +17670,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19966,7 +17705,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19976,17 +17715,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20009,22 +17746,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20034,11 +17770,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20058,14 +17790,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20093,7 +17817,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20102,17 +17826,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20137,7 +17857,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -20156,23 +17875,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20184,13 +17902,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -20203,7 +17917,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2022_11_01.models.CapacityReservation] @@ -20226,17 +17939,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20247,14 +17959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -20264,11 +17976,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20280,10 +17992,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -20306,7 +18014,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20331,23 +18039,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20361,13 +18068,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20389,14 +18092,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20406,7 +18101,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20415,18 +18110,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20436,7 +18123,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20444,19 +18131,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20488,7 +18165,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20501,20 +18178,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20539,23 +18214,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20569,13 +18243,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20596,14 +18266,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20613,7 +18275,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20621,18 +18283,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20642,7 +18296,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20650,19 +18304,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20694,7 +18337,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20707,17 +18350,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -20746,7 +18387,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -20770,16 +18410,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20790,14 +18429,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -20807,11 +18446,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20823,10 +18462,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -20835,7 +18470,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -20854,21 +18488,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20880,20 +18513,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -20919,7 +18548,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -20928,16 +18557,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20957,10 +18585,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -20987,14 +18611,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21008,7 +18624,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21024,18 +18640,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21049,7 +18657,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21062,19 +18670,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21108,7 +18705,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21118,24 +18715,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21161,7 +18756,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21170,16 +18765,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21191,13 +18785,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21225,14 +18815,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21246,7 +18828,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21262,18 +18844,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21287,7 +18861,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21300,19 +18874,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21346,7 +18910,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21356,17 +18920,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21385,22 +18947,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21410,11 +18971,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21429,14 +18986,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21464,7 +19013,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21473,17 +19022,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21505,7 +19050,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21524,23 +19068,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21552,13 +19095,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21572,7 +19111,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21596,18 +19134,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21618,14 +19155,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21635,11 +19172,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21651,12 +19188,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21682,7 +19215,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21708,7 +19241,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21718,16 +19251,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21747,10 +19279,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21779,14 +19307,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21801,7 +19321,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21818,18 +19338,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21844,7 +19356,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -21858,19 +19370,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21905,7 +19406,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21915,17 +19416,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -21933,7 +19432,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21959,7 +19458,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21969,16 +19468,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21990,13 +19488,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22026,14 +19520,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22048,7 +19534,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22065,18 +19551,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22091,7 +19569,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22105,19 +19583,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22152,7 +19620,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22162,17 +19630,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22191,23 +19657,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22217,11 +19682,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22237,14 +19698,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22273,7 +19726,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22282,17 +19735,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22317,7 +19766,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22336,7 +19784,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22344,16 +19792,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22365,13 +19812,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22393,7 +19836,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22417,19 +19859,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22440,14 +19881,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22457,11 +19898,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22472,7 +19913,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_compute_management_client_enums.py index fb414760f08b..f4aec20759fc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_compute_management_client_enums.py @@ -427,7 +427,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_models_py3.py index 52fa10033684..d7432d3468e3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -302,7 +298,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -333,7 +329,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. :code:`
    `:code:`
    ` If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -353,7 +349,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -436,7 +432,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -592,7 +588,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet] @@ -905,7 +901,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1031,7 +1027,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1133,7 +1129,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup] @@ -1293,7 +1289,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.CapacityReservation] @@ -1543,7 +1539,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1756,7 +1752,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1939,7 +1935,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2069,7 +2065,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup] @@ -2102,7 +2098,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group.:code:`
    `:code:`
    `Minimum api-version: 2022-03-01. @@ -2322,7 +2318,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -2580,17 +2576,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2911,7 +2896,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3091,7 +3076,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_11_01.models.SubResource @@ -3244,7 +3229,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.Image] @@ -3278,7 +3263,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2022_11_01.models.SubResource @@ -3420,7 +3405,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3479,7 +3464,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3701,7 +3686,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3734,7 +3719,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4040,7 +4025,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -4073,7 +4058,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.Usage] @@ -4107,7 +4092,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4461,7 +4446,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4537,7 +4522,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu :code:`
    `:code:`
    ` For more information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. :code:`
    `:code:`
    ` Possible @@ -4701,7 +4686,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5250,7 +5235,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5361,7 +5346,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup] @@ -5421,17 +5406,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5472,7 +5446,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2022_11_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2022_11_01.models.PublicIPAddressSkuTier """ @@ -5491,8 +5465,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2022_11_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2022_11_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5503,7 +5476,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5574,7 +5547,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5807,7 +5780,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6545,7 +6518,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6608,7 +6581,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6671,7 +6644,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6745,7 +6718,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6792,7 +6765,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6825,7 +6798,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.RunCommandDocumentBase] @@ -6861,7 +6834,7 @@ def __init__( class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7090,7 +7063,7 @@ class ServiceArtifactReference(_serialization.Model): machines in the scale set when using 'latest' image version. Minimum api-version: 2022-11-01. :ivar id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :vartype id: str """ @@ -7101,7 +7074,7 @@ class ServiceArtifactReference(_serialization.Model): def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :paramtype id: str """ super().__init__(**kwargs) @@ -7242,7 +7215,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7252,7 +7225,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7281,7 +7254,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7296,7 +7269,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7352,7 +7325,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource] @@ -7621,7 +7594,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7642,67 +7615,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7771,7 +7683,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -7914,7 +7826,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8132,7 +8044,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8624,7 +8536,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8863,7 +8775,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -8907,7 +8819,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9222,7 +9134,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_11_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_11_01.models.UserAssignedIdentitiesValue] """ @@ -9254,7 +9166,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2022_11_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_11_01.models.UserAssignedIdentitiesValue] """ @@ -9268,7 +9180,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9331,7 +9243,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9494,7 +9406,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9819,7 +9731,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -9852,10 +9764,12 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineNetworkInterfaceConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -9971,7 +9885,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -9991,10 +9905,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -10160,10 +10074,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10263,10 +10177,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -10344,7 +10258,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10591,7 +10505,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand] @@ -10740,7 +10654,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11009,7 +10923,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11289,10 +11203,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension] @@ -11552,7 +11466,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2022_11_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_11_01.models.UserAssignedIdentitiesValue] """ @@ -11586,7 +11500,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2022_11_01.models.UserAssignedIdentitiesValue] """ @@ -11640,7 +11554,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11667,7 +11581,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(_serialization.Model): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -11816,10 +11730,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -11861,7 +11775,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet] @@ -11897,7 +11811,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetSku] @@ -11933,7 +11847,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet] @@ -11966,7 +11880,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -12017,10 +11931,10 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): +class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network configuration name. Required. :vartype name: str @@ -12126,7 +12040,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12151,7 +12065,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_11_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12182,7 +12096,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_11_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12201,7 +12115,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12505,10 +12419,10 @@ def __init__( self.require_guest_provision_signal = require_guest_provision_signal -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12597,10 +12511,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12627,7 +12543,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12641,33 +12557,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype os_profile: ~azure.mgmt.compute.v2022_11_01.models.OSProfileProvisioningData """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - "exact_version": {"key": "exactVersion", "type": "str"}, - "os_profile": {"key": "osProfile", "type": "OSProfileProvisioningData"}, - } - - def __init__( - self, - *, - temp_disk: Optional[bool] = None, - exact_version: Optional[str] = None, - os_profile: Optional["_models.OSProfileProvisioningData"] = None, - **kwargs: Any - ) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - :keyword exact_version: Specifies in decimal number, the version the OS disk should be reimaged - to. If exact version is not provided, the OS disk is reimaged to the existing version of OS - Disk. - :paramtype exact_version: str - :keyword os_profile: Specifies information required for reimaging the non-ephemeral OS disk. - :paramtype os_profile: ~azure.mgmt.compute.v2022_11_01.models.OSProfileProvisioningData - """ - super().__init__(temp_disk=temp_disk, exact_version=exact_version, os_profile=os_profile, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -13003,7 +12892,7 @@ def __init__( self.proximity_placement_group = proximity_placement_group -class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): +class VirtualMachineScaleSetUpdateIPConfiguration(SubResource): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -13121,7 +13010,9 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(SubResource): # pylint: disable=too-many-instance-attributes +class VirtualMachineScaleSetUpdateNetworkConfiguration( + SubResource +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar id: Resource Id. @@ -13231,12 +13122,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2022_11_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -13269,7 +13160,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2022_11_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13427,7 +13318,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -13487,7 +13378,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -13653,7 +13544,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14052,7 +13943,7 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -14075,7 +13966,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -14258,10 +14149,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -14327,7 +14218,7 @@ class VirtualMachineScaleSetVMInstanceView(_serialization.Model): # pylint: dis :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar os_name: The Operating System running on the hybrid machine. :vartype os_name: str @@ -14413,7 +14304,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword os_name: The Operating System running on the hybrid machine. :paramtype os_name: str @@ -14445,7 +14336,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM] @@ -14478,7 +14369,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -15344,14 +15235,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -15396,7 +15287,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -15452,7 +15343,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -15681,7 +15572,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/operations/_operations.py index ccd398648fe3..0d1ae59185ec 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2022_11_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -176,7 +176,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -210,7 +210,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -248,7 +248,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -286,7 +286,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -320,7 +320,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -361,7 +361,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -392,7 +392,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -422,7 +422,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -448,7 +450,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -479,7 +481,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -510,7 +512,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -551,7 +553,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -585,7 +587,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -619,7 +621,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -653,7 +655,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -687,7 +689,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -721,7 +723,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -755,7 +757,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -789,7 +791,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -832,7 +834,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -866,7 +868,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -900,7 +902,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -935,7 +937,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -970,7 +972,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1002,7 +1004,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1042,7 +1044,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1073,7 +1075,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1104,7 +1106,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1135,7 +1137,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1166,7 +1168,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1197,7 +1199,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1238,7 +1240,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1279,7 +1281,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1317,7 +1319,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1359,7 +1361,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1399,7 +1401,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1434,7 +1436,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1466,7 +1468,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1498,7 +1500,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1533,7 +1535,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1573,7 +1575,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1613,7 +1615,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1645,7 +1647,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1691,7 +1693,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1731,7 +1733,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1763,7 +1765,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1795,7 +1797,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1827,7 +1829,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1869,7 +1871,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1901,7 +1903,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1933,7 +1935,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1968,7 +1970,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2003,7 +2005,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2038,7 +2040,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2070,7 +2072,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2110,7 +2112,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2143,7 +2145,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2205,7 +2209,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2349,7 +2353,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2380,7 +2384,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2411,7 +2415,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2444,7 +2448,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2537,7 +2541,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2759,7 +2763,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2799,7 +2803,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2830,7 +2834,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2861,7 +2865,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2892,7 +2896,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2926,7 +2930,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2994,7 +2998,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3042,7 +3046,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3073,7 +3077,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3102,7 +3106,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3134,7 +3138,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3165,7 +3169,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3207,7 +3211,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3257,7 +3261,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3289,7 +3293,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3320,7 +3324,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3353,7 +3357,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3386,7 +3390,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3417,7 +3421,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3463,7 +3467,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3593,7 +3597,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3649,7 +3653,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3680,7 +3684,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3716,7 +3720,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3752,7 +3756,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3785,7 +3789,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -3827,7 +3831,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3853,7 +3859,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3883,7 +3889,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3917,7 +3923,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3951,7 +3957,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4020,7 +4026,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4050,7 +4056,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4074,7 +4082,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4216,7 +4224,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4279,7 +4287,9 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4303,7 +4313,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4463,7 +4473,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4626,7 +4636,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4680,7 +4690,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4716,7 +4726,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4752,7 +4762,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4785,7 +4795,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4825,7 +4835,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4855,7 +4865,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5002,7 +5014,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5038,7 +5050,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5074,7 +5086,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5107,7 +5119,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5147,7 +5159,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5183,7 +5195,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5216,7 +5228,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5257,7 +5269,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5298,7 +5310,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5378,7 +5390,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5411,7 +5423,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5444,7 +5456,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5477,7 +5489,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5504,7 +5518,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5535,7 +5549,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5570,7 +5584,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5605,7 +5619,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5637,7 +5651,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5677,7 +5691,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5710,7 +5724,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5751,7 +5765,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5792,7 +5806,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5830,7 +5844,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5872,7 +5886,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5936,7 +5950,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -5960,14 +5973,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5978,14 +5990,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -5995,11 +6007,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6011,8 +6023,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6041,7 +6051,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6063,16 +6072,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6083,14 +6091,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6100,11 +6108,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6116,8 +6124,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6146,7 +6152,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6168,16 +6173,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6188,14 +6192,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6205,11 +6209,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6221,8 +6225,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6251,7 +6253,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6275,16 +6276,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6295,14 +6295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6312,11 +6312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6328,15 +6328,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6362,7 +6358,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6370,16 +6366,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6399,10 +6394,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6424,14 +6415,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6444,7 +6427,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6456,18 +6439,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6480,7 +6455,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6489,20 +6464,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6535,7 +6499,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6545,23 +6509,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6587,7 +6549,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6595,16 +6557,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6616,13 +6577,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6645,14 +6602,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6665,7 +6614,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6677,18 +6626,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6701,7 +6642,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6710,20 +6651,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6756,7 +6687,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6766,17 +6697,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6795,22 +6724,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6820,11 +6748,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6839,14 +6763,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6874,7 +6790,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6883,17 +6799,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6914,7 +6826,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -6933,22 +6844,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6960,19 +6870,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -7001,7 +6907,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7009,16 +6915,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7028,11 +6933,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7059,14 +6960,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7077,7 +6970,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -7092,18 +6985,10 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7114,7 +6999,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and @@ -7126,20 +7011,9 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7169,7 +7043,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7178,23 +7052,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7222,7 +7092,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7231,16 +7101,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7250,11 +7119,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7283,14 +7148,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7301,7 +7158,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7315,21 +7172,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7340,7 +7189,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7352,23 +7201,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7399,7 +7237,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7408,17 +7246,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7430,7 +7264,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7449,21 +7282,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7475,13 +7307,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7489,7 +7317,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7513,16 +7340,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7533,14 +7359,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7550,11 +7376,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7566,17 +7392,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7600,15 +7421,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7619,14 +7439,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7636,11 +7456,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7652,8 +7472,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7665,7 +7483,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7689,17 +7506,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7710,14 +7526,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7727,11 +7543,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7743,10 +7559,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7757,7 +7569,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7781,17 +7592,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7802,14 +7612,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7819,11 +7629,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7835,15 +7645,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -7874,7 +7680,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7883,16 +7689,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7902,11 +7707,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -7938,14 +7739,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7956,7 +7749,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -7972,7 +7765,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -7980,14 +7773,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7998,7 +7783,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8012,24 +7797,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8060,7 +7834,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8069,23 +7843,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8114,7 +7884,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8122,16 +7892,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8141,11 +7910,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8170,14 +7935,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8188,7 +7945,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8201,18 +7958,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8223,7 +7972,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8233,20 +7982,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8276,7 +8014,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8285,23 +8023,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8330,7 +8064,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8338,16 +8072,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8357,11 +8090,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8386,14 +8115,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8404,7 +8125,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8417,18 +8138,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8439,7 +8152,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8449,20 +8162,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8492,7 +8194,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8501,23 +8203,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8546,7 +8244,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8554,16 +8252,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8573,11 +8270,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -8603,14 +8296,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8621,7 +8306,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8635,18 +8320,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8657,7 +8334,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8668,20 +8345,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8711,7 +8377,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8720,23 +8386,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8765,7 +8427,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8773,16 +8435,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8792,11 +8453,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -8824,14 +8481,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8842,7 +8491,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8858,18 +8507,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8880,7 +8521,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -8893,20 +8534,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8936,7 +8566,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8945,23 +8575,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -8987,7 +8613,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8995,16 +8621,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9014,11 +8639,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9043,14 +8664,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9061,7 +8674,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9074,18 +8687,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9096,7 +8701,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9106,20 +8711,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9149,7 +8743,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9158,23 +8752,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9203,7 +8793,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9211,16 +8801,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9230,11 +8819,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9260,14 +8845,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9278,7 +8855,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9292,18 +8869,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9314,7 +8883,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9326,20 +8895,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9369,7 +8927,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9378,23 +8936,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9423,7 +8977,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9431,16 +8985,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9450,11 +9003,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9480,14 +9029,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9498,7 +9039,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9512,18 +9053,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9534,7 +9067,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9545,20 +9078,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9588,7 +9110,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9597,20 +9119,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -9636,7 +9154,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9655,7 +9172,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9663,16 +9180,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9684,13 +9200,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -9715,7 +9227,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9726,7 +9237,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9739,11 +9250,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9754,7 +9264,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -9765,13 +9275,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2022_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2022_11_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9799,7 +9306,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9807,16 +9314,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9826,17 +9332,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9862,7 +9364,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9870,16 +9372,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9889,11 +9390,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -9917,14 +9414,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9935,7 +9424,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9948,18 +9437,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9970,7 +9451,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -9981,19 +9462,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10023,7 +9494,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10032,20 +9503,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10070,7 +9537,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10096,7 +9563,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10105,16 +9572,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10134,10 +9600,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10165,14 +9627,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10186,7 +9640,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10202,18 +9656,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10227,7 +9673,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10240,20 +9686,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10287,7 +9722,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10297,24 +9732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10340,7 +9773,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10349,16 +9782,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10378,10 +9810,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -10409,14 +9837,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10430,7 +9850,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10446,18 +9866,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10471,7 +9883,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -10484,20 +9896,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10531,7 +9933,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10541,17 +9943,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10570,22 +9970,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10595,11 +9994,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10614,14 +10009,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10649,7 +10036,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10658,17 +10045,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10690,7 +10073,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10709,23 +10091,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10737,13 +10118,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10755,7 +10132,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10779,17 +10155,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10800,14 +10175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -10817,11 +10192,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10833,12 +10208,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10875,21 +10246,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10899,11 +10269,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -10913,14 +10279,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10947,7 +10305,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10956,17 +10314,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -10985,21 +10339,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11009,11 +10362,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11027,14 +10376,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11061,7 +10402,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11070,17 +10411,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11099,21 +10436,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11123,11 +10459,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11141,14 +10473,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11175,7 +10499,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11184,17 +10508,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11206,7 +10526,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11225,21 +10544,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11251,16 +10569,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11286,7 +10600,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11312,7 +10626,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11322,16 +10636,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11351,10 +10664,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11384,14 +10693,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11406,7 +10707,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11423,18 +10724,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11449,7 +10742,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -11463,20 +10756,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11511,7 +10793,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11521,17 +10803,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -11539,7 +10819,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11565,7 +10845,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11575,16 +10855,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11596,13 +10875,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11633,14 +10908,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11655,7 +10922,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11672,18 +10939,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11698,7 +10957,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -11712,20 +10971,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11760,7 +11009,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11770,17 +11019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -11799,23 +11046,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11825,11 +11071,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11845,14 +11087,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11881,7 +11115,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11890,17 +11124,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11925,7 +11155,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11944,7 +11173,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11952,16 +11181,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11973,13 +11201,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12001,7 +11225,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12020,23 +11243,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12048,13 +11270,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12082,7 +11300,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12111,7 +11331,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12120,16 +11340,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12139,11 +11358,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12171,14 +11386,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12190,7 +11397,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12205,18 +11412,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12228,7 +11427,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12240,20 +11441,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12284,7 +11475,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12293,17 +11484,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12322,22 +11509,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12347,11 +11533,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -12366,14 +11548,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12401,7 +11575,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12410,17 +11584,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12439,22 +11609,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12464,11 +11633,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12484,14 +11649,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12519,7 +11676,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12528,24 +11685,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12571,7 +11724,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12580,16 +11733,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12609,10 +11761,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -12639,14 +11787,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12660,7 +11800,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12676,18 +11816,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12701,7 +11833,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12714,19 +11846,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12760,7 +11881,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12770,17 +11891,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -12805,23 +11924,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12831,11 +11949,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12858,14 +11972,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12894,7 +12000,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12903,17 +12009,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12937,7 +12039,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -12956,23 +12057,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12984,13 +12084,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13004,7 +12100,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13023,22 +12118,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13050,13 +12144,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13085,7 +12175,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13109,7 +12198,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13117,12 +12206,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13133,14 +12221,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13150,11 +12238,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13166,10 +12254,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13193,23 +12277,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13219,11 +12302,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13249,14 +12328,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13285,7 +12356,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13294,17 +12365,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13323,22 +12390,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13348,11 +12414,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -13366,14 +12428,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13401,7 +12455,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13410,17 +12464,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13439,22 +12489,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13464,11 +12513,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -13482,14 +12527,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13517,7 +12554,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13526,17 +12563,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13555,22 +12588,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13580,11 +12612,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -13599,14 +12627,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13634,7 +12654,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13643,17 +12663,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13679,7 +12695,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13698,23 +12713,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13726,13 +12740,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13751,22 +12761,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13776,11 +12785,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -13794,14 +12799,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13829,7 +12826,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13838,17 +12835,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -13862,7 +12855,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13881,22 +12873,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13906,18 +12897,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -13943,7 +12930,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13952,16 +12939,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13975,13 +12961,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14007,14 +12989,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -14027,7 +13001,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14041,18 +13015,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -14065,7 +13031,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14077,19 +13043,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -14122,7 +13077,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14134,17 +13089,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -14172,7 +13125,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14198,7 +13151,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14207,16 +13160,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14236,10 +13188,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -14266,14 +13214,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14287,7 +13227,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14303,18 +13243,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14328,7 +13260,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -14341,20 +13273,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14388,7 +13309,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14398,24 +13319,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14441,7 +13360,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14450,16 +13369,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14471,13 +13389,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14506,14 +13420,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14527,7 +13433,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14543,18 +13449,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14568,7 +13466,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -14581,20 +13479,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14628,7 +13515,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14638,17 +13525,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14667,22 +13552,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14692,11 +13576,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14711,14 +13591,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14746,7 +13618,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14755,17 +13627,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -14787,7 +13655,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -14806,23 +13673,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14834,13 +13700,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -14854,7 +13716,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14873,22 +13734,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14900,13 +13760,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -14936,7 +13792,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -14958,16 +13813,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14978,14 +13832,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -14995,11 +13849,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15011,15 +13865,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15045,7 +13895,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15053,16 +13903,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15076,13 +13925,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15106,14 +13951,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15126,7 +13963,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15139,18 +13976,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15163,7 +13992,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15174,19 +14003,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15219,7 +14038,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15231,20 +14050,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15269,7 +14090,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15277,16 +14098,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15306,10 +14126,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15332,14 +14148,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15351,7 +14159,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15364,18 +14172,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15384,7 +14184,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15394,19 +14198,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15438,7 +14231,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15448,20 +14241,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15486,7 +14281,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15494,16 +14289,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15515,13 +14309,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15544,14 +14334,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15563,7 +14345,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15575,18 +14357,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15595,7 +14369,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15604,19 +14382,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] @@ -15648,7 +14415,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15658,17 +14425,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15687,22 +14452,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15712,11 +14476,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15731,14 +14491,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15766,7 +14518,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15775,17 +14527,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15808,7 +14556,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -15827,22 +14574,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15854,13 +14600,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -15872,7 +14614,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -15891,21 +14632,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15917,13 +14657,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -15942,21 +14678,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15966,11 +14701,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -15981,14 +14712,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16015,7 +14738,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16024,17 +14747,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -16053,22 +14772,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16078,11 +14796,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -16098,14 +14812,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16133,7 +14839,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16142,17 +14848,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16169,7 +14871,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16188,21 +14889,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16212,11 +14912,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -16232,7 +14928,6 @@ def list( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16254,17 +14949,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16275,14 +14969,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16292,11 +14986,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16308,10 +15002,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, *, status_only: Optional[str] = None, filter: Optional[str] = None, **kwargs: Any @@ -16327,7 +15017,6 @@ def list_all( /subscriptions/{subId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmssName}'. Default value is None. :paramtype filter: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16349,17 +15038,16 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16370,14 +15058,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16387,11 +15075,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16403,8 +15091,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16416,7 +15102,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16438,17 +15123,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16459,14 +15143,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16476,11 +15160,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16492,10 +15176,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16513,22 +15193,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16538,11 +15217,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16559,14 +15234,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16594,7 +15261,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16603,17 +15270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16632,21 +15295,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16656,11 +15318,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16670,14 +15328,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16704,7 +15354,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16713,17 +15363,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16742,21 +15388,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16766,11 +15411,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16780,14 +15421,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16814,7 +15447,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16823,17 +15456,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16852,21 +15481,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16876,11 +15504,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16890,14 +15514,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16924,7 +15540,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16933,17 +15549,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16962,21 +15574,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16986,11 +15597,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17000,14 +15607,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17034,7 +15633,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17043,23 +15642,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17088,7 +15683,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17096,16 +15691,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17115,11 +15709,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17148,14 +15738,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17166,7 +15748,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17184,18 +15766,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17206,7 +15780,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -17221,19 +15795,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17263,7 +15827,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17272,17 +15836,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17304,7 +15864,6 @@ def retrieve_boot_diagnostics_data( URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17323,22 +15882,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17350,13 +15908,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17375,21 +15929,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17399,11 +15952,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17413,14 +15962,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17447,7 +15988,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17456,17 +15997,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17478,7 +16015,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17497,21 +16033,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17521,11 +16056,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17544,21 +16075,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17572,13 +16102,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17590,14 +16116,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17627,7 +16145,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17639,23 +16157,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17681,7 +16197,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17689,16 +16205,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17712,13 +16227,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17743,14 +16254,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17763,7 +16266,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17776,18 +16279,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17800,7 +16295,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -17810,20 +16305,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -17856,7 +16340,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17868,20 +16352,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -17906,7 +16392,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17914,16 +16400,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17937,13 +16422,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -17966,14 +16447,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -17985,7 +16458,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17997,18 +16470,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -18017,7 +16482,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18026,19 +16495,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RunCommandResult] @@ -18070,7 +16528,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18082,17 +16540,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -18131,7 +16587,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18150,7 +16605,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18158,16 +16613,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18179,13 +16633,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18217,7 +16667,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18236,7 +16685,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18246,16 +16695,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18267,13 +16715,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18285,7 +16729,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18304,21 +16747,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18330,13 +16772,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -18344,7 +16782,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18363,20 +16800,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18388,13 +16824,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18408,7 +16840,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18427,22 +16858,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18454,13 +16884,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -18470,7 +16896,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18489,21 +16914,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18515,13 +16939,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -18562,7 +16982,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18581,7 +17000,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18590,16 +17009,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18611,13 +17029,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18654,7 +17068,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18673,7 +17086,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18684,16 +17097,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18705,13 +17117,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18726,7 +17134,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18745,22 +17152,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18772,13 +17178,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -18790,7 +17192,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18809,21 +17210,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18835,13 +17235,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18858,7 +17254,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18877,23 +17272,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18905,13 +17299,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -18948,7 +17338,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18967,23 +17356,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18995,13 +17383,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -19013,7 +17397,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19032,21 +17415,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19058,13 +17440,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -19092,7 +17470,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19111,7 +17488,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -19120,16 +17497,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19141,13 +17517,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -19191,7 +17563,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19202,7 +17573,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19214,11 +17585,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19229,7 +17599,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -19239,12 +17609,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19272,7 +17638,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19280,16 +17646,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19301,13 +17666,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -19330,7 +17691,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19341,7 +17701,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19353,11 +17713,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19368,7 +17727,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -19378,12 +17737,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19411,7 +17766,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19419,16 +17774,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19440,13 +17794,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -19458,7 +17808,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19477,21 +17826,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19501,11 +17849,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -19515,7 +17859,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19534,21 +17877,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19560,13 +17902,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -19577,7 +17915,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19599,16 +17936,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19619,14 +17955,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19636,11 +17972,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19652,17 +17988,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19684,16 +18015,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19704,14 +18034,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19721,11 +18051,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19737,10 +18067,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -19752,7 +18078,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -19774,17 +18099,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19795,14 +18119,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -19812,11 +18136,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19828,10 +18152,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -19875,7 +18195,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19886,7 +18205,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19899,11 +18218,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19914,7 +18232,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -19924,12 +18242,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19957,7 +18271,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -19965,16 +18279,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19994,10 +18307,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -20020,7 +18329,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20031,7 +18339,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20044,11 +18352,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20059,7 +18366,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -20069,12 +18376,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20102,7 +18406,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20110,16 +18414,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20131,13 +18434,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20149,7 +18448,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20168,21 +18466,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20192,11 +18489,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20216,7 +18509,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20235,22 +18527,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20262,19 +18553,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20298,15 +18584,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20317,14 +18602,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20334,11 +18619,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20350,10 +18635,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -20362,7 +18643,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20386,16 +18666,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20406,14 +18685,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20423,11 +18702,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20439,10 +18718,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -20487,7 +18762,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20498,7 +18772,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20512,11 +18786,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20527,7 +18800,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -20539,12 +18812,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20572,7 +18841,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20580,16 +18849,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20609,10 +18877,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -20634,7 +18898,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20645,7 +18908,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20657,11 +18920,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20672,7 +18934,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -20681,13 +18943,9 @@ def update( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20715,7 +18973,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20723,16 +18981,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20744,13 +19001,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20762,7 +19015,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20781,21 +19033,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20805,11 +19056,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20831,7 +19078,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20850,22 +19096,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20877,13 +19122,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -20892,7 +19133,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20914,16 +19154,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20934,14 +19173,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -20951,11 +19190,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20967,16 +19206,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -20998,15 +19232,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21017,14 +19250,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21034,11 +19267,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21050,8 +19283,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -21078,7 +19309,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21104,7 +19335,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21113,16 +19344,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21142,10 +19372,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -21170,14 +19396,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21190,7 +19408,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21204,18 +19422,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21228,7 +19438,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -21240,19 +19450,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21285,7 +19484,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21295,24 +19494,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21338,7 +19535,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21347,16 +19544,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21368,13 +19564,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21400,14 +19592,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21420,7 +19604,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21434,18 +19618,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21458,7 +19634,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update an dedicated host . @@ -21470,19 +19646,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] @@ -21515,7 +19680,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21525,17 +19690,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -21554,22 +19717,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21579,11 +19741,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21597,14 +19755,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21632,7 +19782,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21641,17 +19791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21675,7 +19821,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -21694,23 +19839,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21722,13 +19866,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -21741,7 +19881,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -21763,17 +19902,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21784,14 +19922,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -21801,11 +19939,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21817,10 +19955,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -21838,22 +19972,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21863,11 +19996,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -21884,14 +20013,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21919,7 +20040,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21928,17 +20049,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SshPublicKeysOperations: @@ -21966,7 +20083,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -21990,15 +20106,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22009,14 +20124,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22026,11 +20141,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22042,8 +20157,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -22053,7 +20166,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22077,16 +20189,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22097,14 +20208,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22114,11 +20225,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22130,10 +20241,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -22155,7 +20262,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22166,7 +20272,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22178,11 +20284,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22193,7 +20298,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -22203,12 +20308,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22236,7 +20337,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22244,16 +20345,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22273,10 +20373,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -22298,7 +20394,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22309,7 +20404,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22321,11 +20416,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22336,7 +20430,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -22346,12 +20440,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22379,7 +20470,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22387,16 +20478,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22408,13 +20498,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22426,7 +20512,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22445,21 +20530,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22469,11 +20553,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -22483,7 +20563,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22502,21 +20581,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22528,13 +20606,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -22548,7 +20622,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22567,21 +20640,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22593,13 +20665,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -22623,7 +20691,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22648,7 +20716,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22656,16 +20724,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22685,10 +20752,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -22710,14 +20773,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22728,7 +20783,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22740,18 +20795,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22759,7 +20806,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -22768,19 +20815,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22811,7 +20847,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22821,20 +20857,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22859,7 +20897,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22867,16 +20905,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22896,10 +20933,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -22921,14 +20954,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22939,7 +20964,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22951,18 +20976,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -22970,7 +20987,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -22979,19 +21000,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23022,7 +21032,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23032,17 +21042,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -23061,21 +21069,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23085,11 +21092,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23099,14 +21102,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23133,7 +21128,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23142,17 +21137,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23166,7 +21157,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -23185,22 +21175,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23212,13 +21201,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -23226,7 +21211,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23248,16 +21232,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23268,14 +21251,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23285,11 +21268,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23301,16 +21284,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23332,15 +21310,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23351,14 +21328,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23368,11 +21345,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23384,8 +21361,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23431,7 +21406,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23442,7 +21416,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23457,11 +21431,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23472,7 +21445,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23484,12 +21457,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23517,7 +21486,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23525,16 +21494,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23554,10 +21522,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23580,7 +21544,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23591,7 +21554,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23604,11 +21567,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23619,7 +21581,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23629,12 +21591,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23662,7 +21621,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23670,16 +21629,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23691,13 +21649,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -23716,21 +21670,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23740,11 +21693,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23757,14 +21706,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23791,7 +21732,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23800,17 +21741,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23832,7 +21769,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23851,22 +21787,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23878,13 +21813,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -23892,7 +21823,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -23916,16 +21846,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23936,14 +21865,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -23953,11 +21882,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23969,17 +21898,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -24003,15 +21927,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24022,14 +21945,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24039,11 +21962,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24055,8 +21978,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -24083,7 +22004,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -24109,7 +22030,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -24118,16 +22039,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24139,13 +22059,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -24172,14 +22088,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -24192,7 +22100,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24207,18 +22115,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -24231,7 +22131,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -24244,19 +22144,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2022_11_01.models.RestorePoint] @@ -24289,7 +22178,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24299,17 +22188,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -24328,22 +22215,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24353,11 +22239,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24371,14 +22253,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24406,7 +22280,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24415,17 +22289,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24448,7 +22318,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24467,23 +22336,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24495,13 +22363,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -24547,7 +22411,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24558,7 +22421,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24572,11 +22435,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24587,7 +22449,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -24599,12 +22461,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24632,7 +22490,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24640,16 +22498,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24669,10 +22526,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -24696,7 +22549,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24707,7 +22559,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24721,11 +22573,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24736,7 +22587,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -24747,12 +22598,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24780,7 +22628,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24788,16 +22636,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24809,13 +22656,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -24830,7 +22673,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -24849,21 +22691,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24873,11 +22714,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -24901,7 +22738,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24920,22 +22756,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24947,13 +22782,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -24974,7 +22805,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -24998,17 +22828,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25019,14 +22848,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25036,11 +22865,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25052,10 +22881,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -25069,7 +22894,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25093,16 +22917,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25113,14 +22936,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25130,11 +22953,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25146,10 +22969,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -25176,7 +22995,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -25202,7 +23021,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25211,16 +23030,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25240,10 +23058,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -25270,14 +23084,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25291,7 +23097,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25307,18 +23113,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25332,7 +23130,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -25346,19 +23144,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25392,7 +23179,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25402,24 +23189,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -25445,7 +23230,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25454,16 +23239,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25477,13 +23261,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25509,14 +23289,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25530,7 +23302,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25544,18 +23316,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25569,7 +23333,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -25581,19 +23345,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25627,7 +23380,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25637,17 +23390,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -25670,22 +23421,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25695,11 +23445,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25719,14 +23465,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25754,7 +23492,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25763,17 +23501,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25798,7 +23532,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2022_11_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -25817,23 +23550,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25845,13 +23577,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -25864,7 +23592,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2022_11_01.models.CapacityReservation] @@ -25887,17 +23614,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25908,14 +23634,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -25925,11 +23651,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25941,10 +23667,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -25967,7 +23689,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -25992,23 +23714,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26022,13 +23743,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26050,14 +23767,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26067,7 +23776,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26076,18 +23785,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26097,7 +23798,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26105,19 +23806,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26149,7 +23840,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26161,20 +23852,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26199,23 +23888,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26229,13 +23917,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26256,14 +23940,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26273,7 +23949,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26281,18 +23957,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26302,7 +23970,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26310,19 +23978,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2022_11_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26354,7 +24011,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26366,17 +24023,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26405,7 +24060,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26429,16 +24083,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26449,14 +24102,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26466,11 +24119,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26482,10 +24135,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26494,7 +24143,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26513,21 +24161,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26539,20 +24186,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26578,7 +24221,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26587,16 +24230,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26616,10 +24258,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26646,14 +24284,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26667,7 +24297,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26683,18 +24313,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26708,7 +24330,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -26721,19 +24343,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26767,7 +24378,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26777,24 +24388,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26820,7 +24429,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26829,16 +24438,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26850,13 +24458,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -26884,14 +24488,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26905,7 +24501,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26921,18 +24517,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -26946,7 +24534,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -26959,19 +24547,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27005,7 +24583,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27015,17 +24593,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27044,22 +24620,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27069,11 +24644,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27088,14 +24659,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27123,7 +24686,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27132,17 +24695,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27164,7 +24723,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27183,23 +24741,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27211,13 +24768,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27231,7 +24784,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27255,18 +24807,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27277,14 +24828,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27294,11 +24845,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27310,12 +24861,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27341,7 +24888,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27367,7 +24914,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27377,16 +24924,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27406,10 +24952,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27438,14 +24980,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27460,7 +24994,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27477,18 +25011,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27503,7 +25029,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27517,19 +25043,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27564,7 +25079,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27574,17 +25089,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27592,7 +25105,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27618,7 +25131,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27628,16 +25141,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27649,13 +25161,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27685,14 +25193,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27707,7 +25207,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27724,18 +25224,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27750,7 +25242,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -27764,19 +25256,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27811,7 +25293,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27821,17 +25303,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -27850,23 +25330,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27876,11 +25355,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27896,14 +25371,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27932,7 +25399,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27941,17 +25408,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27976,7 +25439,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2022_11_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27995,7 +25457,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2022-11-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28003,16 +25465,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28024,13 +25485,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28052,7 +25509,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28076,19 +25532,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28099,14 +25554,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28116,11 +25571,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28131,7 +25586,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_compute_management_client.py index d03f734c731e..014c7f7b89a9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-01-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_configuration.py index 3589dec9553c..b2b73c7a3e5e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-01-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_metadata.json index 074fbf92ad8d..bf4d310ebade 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_compute_management_client.py index 1314fded918e..c84bdd2301c3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-01-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_configuration.py index 33fe3d094031..059850de7f96 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-01-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/operations/_operations.py index 3433783e437e..2bc50f8f4d79 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_01_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_01_02.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_01_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_01_02.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -1238,7 +1086,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1115,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1123,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1151,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -1328,14 +1175,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1347,7 +1186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1362,18 +1201,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1382,7 +1213,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -1393,19 +1228,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1437,7 +1261,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1447,23 +1271,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -1489,7 +1311,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1497,16 +1319,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1526,10 +1347,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -1554,14 +1371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1573,7 +1382,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1588,18 +1397,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1611,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -1623,19 +1424,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1667,7 +1457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1677,17 +1467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -1699,7 +1487,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,21 +1505,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1744,13 +1530,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -1769,21 +1551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,11 +1574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1811,14 +1588,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1845,7 +1614,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1854,17 +1623,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -1872,7 +1637,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1895,16 +1659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1915,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -1932,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1948,15 +1711,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -1979,15 +1737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1998,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -2015,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2031,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2045,7 +1800,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2064,21 +1818,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2090,20 +1843,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -2129,7 +1878,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -2138,16 +1887,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2167,12 +1915,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -2200,14 +1944,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2216,12 +1952,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2239,18 +1975,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2259,12 +1987,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -2280,20 +2008,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2327,7 +2044,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2337,17 +2054,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -2363,7 +2078,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -2382,22 +2096,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2409,15 +2122,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -2434,22 +2143,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2459,14 +2167,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -2479,14 +2183,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2514,7 +2210,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2523,17 +2219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -2547,7 +2239,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2571,17 +2262,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2592,14 +2282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -2609,11 +2299,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2625,10 +2315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -2654,7 +2340,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2680,7 +2366,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2374,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2717,10 +2402,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2745,14 +2426,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2765,7 +2438,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2780,18 +2453,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2804,7 +2469,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2816,19 +2481,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2861,7 +2516,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2871,23 +2526,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2913,7 +2566,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2921,16 +2574,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2950,10 +2602,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2978,14 +2626,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2998,7 +2638,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3013,18 +2653,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3037,7 +2669,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3049,19 +2681,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3094,7 +2717,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3104,17 +2727,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3128,7 +2749,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3147,21 +2767,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2792,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3198,21 +2813,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,11 +2836,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3240,14 +2850,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3274,7 +2876,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3283,17 +2885,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3303,7 +2901,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -3326,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3346,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3363,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3379,15 +2975,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -3410,15 +3001,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3429,14 +3019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3446,11 +3036,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3462,8 +3052,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3476,7 +3064,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3498,17 +3085,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3519,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3536,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3552,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -3598,7 +3180,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -3617,23 +3198,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3645,13 +3225,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -3667,7 +3243,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskRestorePoint] @@ -3690,18 +3265,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3712,14 +3286,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -3729,11 +3303,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3745,17 +3319,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3781,7 +3351,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -3791,16 +3361,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3383,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -3852,14 +3417,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -3873,7 +3430,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3892,18 +3449,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -3917,7 +3466,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -3933,19 +3482,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -3979,7 +3517,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3991,17 +3529,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -4025,23 +3561,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4051,11 +3586,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -4078,14 +3609,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4114,7 +3637,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -4125,17 +3648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -4159,7 +3678,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4184,7 +3703,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4192,16 +3711,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4221,10 +3739,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -4248,14 +3762,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4267,7 +3773,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4281,18 +3787,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4301,7 +3799,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -4312,19 +3810,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4356,7 +3843,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4366,20 +3853,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4404,7 +3893,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4412,16 +3901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4441,10 +3929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -4469,14 +3953,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4488,7 +3964,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4503,18 +3979,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4523,7 +3991,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -4534,19 +4006,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4578,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4588,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -4610,7 +4069,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -4629,21 +4087,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4655,13 +4112,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4680,21 +4133,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4704,11 +4156,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4720,14 +4168,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4754,7 +4194,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4763,17 +4203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -4781,7 +4217,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4804,16 +4239,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4824,14 +4258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4841,11 +4275,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4857,15 +4291,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_01_02.models.Snapshot] @@ -4888,15 +4317,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4907,14 +4335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4924,11 +4352,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4368,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4972,7 +4398,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4980,16 +4406,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4428,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5035,14 +4456,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -5054,7 +4467,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5069,18 +4482,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -5092,7 +4497,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -5104,19 +4509,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] @@ -5148,7 +4542,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5160,17 +4554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5189,21 +4581,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5213,11 +4604,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5231,14 +4618,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_compute_management_client_enums.py index 601c7c2b0d72..6cf07807069a 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_compute_management_client_enums.py @@ -32,7 +32,7 @@ class CopyCompletionErrorReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): COPY_SOURCE_NOT_FOUND = "CopySourceNotFound" """Indicates that the source snapshot was deleted while the background copy of the resource - #: created via CopyStart operation was in progress.""" + created via CopyStart operation was in progress.""" class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -40,8 +40,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -55,31 +55,28 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + disk and upload using write token in both disk and VM guest state""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -100,10 +97,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -113,17 +110,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -156,19 +153,19 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" PREMIUM_V2_LRS = "PremiumV2_LRS" """Premium SSD v2 locally redundant storage. Best for production and performance-sensitive - #: workloads that consistently require low latency and high IOPS and throughput.""" + workloads that consistently require low latency and high IOPS and throughput.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -176,13 +173,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -247,12 +244,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_models_py3.py index 02d7f00a3a37..1ccca57f49e9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/models/_models_py3.py @@ -135,7 +135,7 @@ class CopyCompletionError(_serialization.Model): """Indicates the error details if the background copy of a resource created via the CopyStart operation fails. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar error_code: Indicates the error code if the background copy of a resource created via the CopyStart operation fails. Required. "CopySourceNotFound" @@ -176,11 +176,11 @@ class CreationData(_serialization.Model): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", and "Import". + "ImportSecure", and "UploadPreparedSecure". :vartype create_option: str or ~azure.mgmt.compute.v2023_01_02.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -253,7 +253,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", and "Import". + "CopyStart", "ImportSecure", and "UploadPreparedSecure". :paramtype create_option: str or ~azure.mgmt.compute.v2023_01_02.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -304,7 +304,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -353,7 +353,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -715,7 +715,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -792,7 +792,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] @@ -848,7 +848,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -969,7 +969,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -1080,7 +1080,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2023_01_02.models.Disk] @@ -1305,7 +1305,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2023_01_02.models.DiskRestorePoint] @@ -1708,7 +1708,7 @@ class EncryptionSetIdentity(_serialization.Model): :vartype tenant_id: str :ivar user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_01_02.models.UserAssignedIdentitiesValue] """ @@ -1742,7 +1742,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_01_02.models.UserAssignedIdentitiesValue] """ @@ -1756,7 +1756,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1880,7 +1880,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2023_01_02.models.AccessLevel @@ -2019,7 +2019,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -2058,7 +2058,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_01_02.models.SourceVault @@ -2091,7 +2091,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_01_02.models.SourceVault @@ -2383,7 +2383,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2433,7 +2433,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2540,7 +2540,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2807,7 +2807,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2023_01_02.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/operations/_operations.py index 3d21314da3a4..bfcf8b32944e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_01_02/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -276,7 +276,7 @@ def build_disks_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +406,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +460,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -491,7 +491,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -532,7 +532,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -570,7 +570,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -608,7 +608,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -639,7 +639,7 @@ def build_disk_accesses_list_private_endpoint_connections_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -673,7 +673,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -738,7 +738,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -769,7 +769,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +823,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -894,7 +894,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -932,7 +932,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -975,7 +975,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1139,7 +1139,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_01_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_01_02.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_01_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_01_02.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -2409,7 +2257,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2286,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2294,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2322,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -2499,14 +2346,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2517,7 +2356,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2532,18 +2371,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2551,7 +2382,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -2562,19 +2397,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2605,7 +2429,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2615,23 +2439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -2657,7 +2479,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2665,16 +2487,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2694,10 +2515,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -2722,14 +2539,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2549,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2564,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2578,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -2789,19 +2590,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_01_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2622,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -2864,7 +2652,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -2883,21 +2670,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2909,13 +2695,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2934,21 +2716,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,11 +2739,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2974,14 +2751,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2777,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3017,17 +2786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -3035,7 +2800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3057,16 +2821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3077,14 +2840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3094,11 +2857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3110,15 +2873,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3140,15 +2898,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3176,11 +2933,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2949,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3206,7 +2961,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,21 +2979,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3251,20 +3004,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -3290,7 +3039,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -3299,16 +3048,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3328,12 +3076,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -3361,14 +3105,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3377,12 +3113,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3136,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3420,12 +3148,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -3441,20 +3169,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3488,7 +3205,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3498,17 +3215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -3524,7 +3239,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -3543,22 +3257,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3570,15 +3283,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -3595,22 +3304,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,14 +3328,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -3640,14 +3344,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3675,7 +3371,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3684,17 +3380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -3708,7 +3400,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3732,17 +3423,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3753,14 +3443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -3770,11 +3460,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3786,10 +3476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -3815,7 +3501,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3841,7 +3527,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3849,16 +3535,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3878,10 +3563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3906,14 +3587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -3925,7 +3598,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3940,18 +3613,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -3963,7 +3628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3975,19 +3640,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -4019,7 +3674,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4029,23 +3684,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4071,7 +3724,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4079,16 +3732,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4108,10 +3760,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4136,14 +3784,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -4155,7 +3795,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4170,18 +3810,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -4193,7 +3825,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -4205,19 +3837,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] @@ -4249,7 +3872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4259,17 +3882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4281,7 +3902,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4300,21 +3920,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4326,13 +3945,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4351,21 +3966,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4375,11 +3989,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4391,14 +4001,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4425,7 +4027,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4434,17 +4036,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4452,7 +4050,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4474,16 +4071,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4494,14 +4090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4511,11 +4107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4527,15 +4123,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4557,15 +4148,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4576,14 +4166,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4593,11 +4183,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4609,8 +4199,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4623,7 +4211,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4645,17 +4232,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4683,11 +4269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,10 +4285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -4745,7 +4327,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4764,23 +4345,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4792,13 +4372,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4814,7 +4390,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,18 +4411,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4858,14 +4432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4875,11 +4449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4891,17 +4465,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4927,7 +4497,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4937,16 +4507,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4529,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4998,14 +4563,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5018,7 +4575,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5037,18 +4594,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5061,7 +4610,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5077,19 +4626,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5122,7 +4660,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5134,17 +4672,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5168,23 +4704,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5194,11 +4729,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -5221,14 +4752,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5257,7 +4780,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -5268,17 +4791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -5302,7 +4821,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5327,7 +4846,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5335,16 +4854,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5364,10 +4882,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -5391,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5409,7 +4915,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5423,18 +4929,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,7 +4940,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -5453,19 +4951,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5496,7 +4983,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5506,20 +4993,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5544,7 +5033,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5552,16 +5041,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5581,10 +5069,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -5609,14 +5093,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5627,7 +5103,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5642,18 +5118,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5661,7 +5129,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -5672,19 +5144,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_01_02.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5715,7 +5176,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5725,17 +5186,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -5747,7 +5206,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_01_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -5766,21 +5224,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5792,13 +5249,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5817,21 +5270,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5841,11 +5293,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5857,14 +5305,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5891,7 +5331,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5900,17 +5340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -5918,7 +5354,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5940,16 +5375,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5960,14 +5394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -5977,11 +5411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5993,15 +5427,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_01_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -6023,15 +5452,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6042,14 +5470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -6059,11 +5487,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6075,13 +5503,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6107,7 +5533,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -6115,16 +5541,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6138,13 +5563,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6170,14 +5591,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,7 +5601,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6203,18 +5616,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6225,7 +5630,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -6237,19 +5642,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_01_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_01_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6280,7 +5674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6292,17 +5686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -6321,21 +5713,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6345,11 +5736,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6361,14 +5748,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_compute_management_client.py index 4e972c8cf3a4..fea411a15e5e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_configuration.py index fecdc9b06892..1b21d309858d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_metadata.json index 01ec05caa16b..c4d85b1664dd 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_compute_management_client.py index 944df03be8bc..7d374375d78d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-03-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_configuration.py index 2f7a13164c2f..46b42aab28b3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-03-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/operations/_operations.py index 865f6bbaa7f3..3b9f02cb1278 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -236,7 +236,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -260,14 +259,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -278,14 +276,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -295,11 +293,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -311,8 +309,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -341,7 +337,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -363,16 +358,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -383,14 +377,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -400,11 +394,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -416,8 +410,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -446,7 +438,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] @@ -469,16 +460,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -489,14 +479,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -506,11 +496,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -522,8 +512,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -552,7 +540,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -576,16 +563,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -596,14 +582,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -613,11 +599,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -629,15 +615,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -663,7 +645,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -671,16 +653,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -700,10 +681,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -725,14 +702,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -745,7 +714,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -757,18 +726,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -781,7 +742,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -790,20 +751,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -836,7 +786,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -846,23 +796,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -888,7 +836,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -896,16 +844,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -917,13 +864,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -946,14 +889,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -966,7 +901,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -978,18 +913,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1002,7 +929,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1011,20 +938,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1057,7 +974,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1067,17 +984,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1096,22 +1011,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1121,11 +1035,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1140,14 +1050,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1175,7 +1077,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1184,17 +1086,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1215,7 +1113,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1234,22 +1131,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1261,19 +1157,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1304,7 +1196,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1313,16 +1205,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1332,11 +1223,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1368,14 +1255,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1386,7 +1265,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -1402,7 +1281,7 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -1410,14 +1289,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1428,7 +1299,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1442,24 +1313,13 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1490,7 +1350,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1499,23 +1359,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1543,7 +1399,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1552,16 +1408,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1571,11 +1426,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1604,14 +1455,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1622,7 +1465,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1636,21 +1479,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1661,7 +1496,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1673,23 +1508,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1720,7 +1544,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1729,17 +1553,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1751,7 +1571,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1770,21 +1589,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1796,13 +1614,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1810,7 +1624,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1834,16 +1647,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1854,14 +1666,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1871,11 +1683,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1887,17 +1699,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1921,15 +1728,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1940,14 +1746,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -1957,11 +1763,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1973,8 +1779,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -1986,7 +1790,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -2010,17 +1813,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2031,14 +1833,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2048,11 +1850,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2064,10 +1866,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2078,7 +1876,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2102,17 +1899,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2123,14 +1919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2140,11 +1936,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2156,15 +1952,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2195,7 +1987,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2204,16 +1996,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2223,11 +2014,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2259,14 +2046,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2277,7 +2056,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2293,7 +2072,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2301,14 +2080,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2319,7 +2090,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2333,24 +2104,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2381,7 +2141,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2390,23 +2150,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2435,7 +2191,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2443,16 +2199,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2462,11 +2217,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2491,14 +2242,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2509,7 +2252,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2522,18 +2265,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2544,7 +2279,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2554,20 +2289,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2597,7 +2321,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2606,23 +2330,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2651,7 +2371,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2379,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2678,11 +2397,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2707,14 +2422,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2725,7 +2432,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2738,18 +2445,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2760,7 +2459,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2770,20 +2469,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2813,7 +2501,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2822,17 +2510,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2851,21 +2535,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2879,11 +2562,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async async def begin_reapply( @@ -2896,14 +2575,6 @@ async def begin_reapply( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2930,7 +2601,7 @@ async def begin_reapply( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2941,23 +2612,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2986,7 +2653,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2994,16 +2661,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3013,11 +2679,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -3043,14 +2705,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3061,7 +2715,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3075,18 +2729,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3097,7 +2743,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -3108,20 +2754,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3151,7 +2786,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3160,23 +2795,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3205,7 +2836,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3213,16 +2844,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3232,11 +2862,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3264,14 +2890,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3282,7 +2900,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3298,18 +2916,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3320,7 +2930,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3333,20 +2943,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3376,7 +2975,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3385,23 +2984,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3427,7 +3022,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3435,16 +3030,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3454,11 +3048,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3483,14 +3073,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3501,7 +3083,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3514,18 +3096,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3536,7 +3110,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3546,20 +3120,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3589,7 +3152,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3598,23 +3161,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3643,7 +3202,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3651,16 +3210,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3670,11 +3228,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3700,14 +3254,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3718,7 +3264,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3732,18 +3278,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3754,7 +3292,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3766,20 +3304,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3809,7 +3336,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3818,23 +3345,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3863,7 +3386,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3871,16 +3394,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3890,11 +3412,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -3920,14 +3438,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3938,7 +3448,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3952,18 +3462,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3974,7 +3476,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -3985,20 +3487,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4028,7 +3519,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4037,20 +3528,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -4076,7 +3563,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -4095,7 +3581,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4103,16 +3589,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4124,13 +3609,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4155,7 +3636,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4166,7 +3646,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4179,11 +3659,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4194,7 +3673,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4205,13 +3684,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4239,7 +3715,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4247,16 +3723,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4266,17 +3741,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4302,7 +3773,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4310,16 +3781,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4329,11 +3799,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4357,14 +3823,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4375,7 +3833,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4388,18 +3846,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4410,7 +3860,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4421,19 +3871,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4463,7 +3903,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4472,20 +3912,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4510,7 +3946,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4536,7 +3972,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4545,16 +3981,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4574,10 +4009,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4605,14 +4036,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4626,7 +4049,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4642,18 +4065,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4667,7 +4082,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4680,20 +4095,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4727,7 +4131,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4737,24 +4141,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4780,7 +4182,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4789,16 +4191,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4818,10 +4219,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -4849,14 +4246,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4870,7 +4259,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4886,18 +4275,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4911,7 +4292,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -4924,20 +4305,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4971,7 +4342,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4981,17 +4352,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -5010,22 +4379,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5035,11 +4403,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -5054,14 +4418,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5089,7 +4445,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5098,17 +4454,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -5130,7 +4482,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5149,23 +4500,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4527,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4541,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5219,17 +4564,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4584,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5257,11 +4601,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,12 +4617,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5315,21 +4655,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5339,11 +4678,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5355,14 +4690,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5389,7 +4716,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5398,17 +4725,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5427,21 +4750,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5451,11 +4773,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5469,14 +4787,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5503,7 +4813,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5512,17 +4822,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5541,21 +4847,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5565,11 +4870,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5583,14 +4884,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5617,7 +4910,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5626,17 +4919,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5648,7 +4937,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5667,21 +4955,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5693,16 +4980,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5728,7 +5011,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5754,7 +5037,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5764,16 +5047,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5793,10 +5075,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -5826,14 +5104,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5848,7 +5118,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5865,18 +5135,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5891,7 +5153,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -5905,20 +5167,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5953,7 +5204,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5963,17 +5214,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -5981,7 +5230,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -6007,7 +5256,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6017,16 +5266,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6038,13 +5286,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6075,14 +5319,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6097,7 +5333,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6114,18 +5350,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6140,7 +5368,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6154,20 +5382,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6202,7 +5420,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6212,17 +5430,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6241,23 +5457,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6267,11 +5482,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6287,14 +5498,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6323,7 +5526,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6332,17 +5535,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6367,7 +5566,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6386,7 +5584,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6394,16 +5592,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6415,13 +5612,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6443,7 +5636,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6462,23 +5654,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6490,13 +5681,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6524,7 +5711,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6553,7 +5742,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6562,16 +5751,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6581,11 +5769,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6613,14 +5797,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6632,7 +5808,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6647,18 +5823,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6670,7 +5838,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6682,20 +5852,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6726,7 +5886,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6735,17 +5895,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6764,22 +5920,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6789,11 +5944,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -6808,14 +5959,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6843,7 +5986,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6852,17 +5995,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6881,22 +6020,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6906,11 +6044,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6926,14 +6060,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6961,7 +6087,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6970,24 +6096,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -7013,7 +6135,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -7022,16 +6144,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7051,10 +6172,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -7081,14 +6198,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7102,7 +6211,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7118,18 +6227,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7143,7 +6244,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -7156,19 +6257,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7202,7 +6292,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7212,17 +6302,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7247,23 +6335,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7273,11 +6360,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7300,14 +6383,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7336,7 +6411,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7345,17 +6420,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7379,7 +6450,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7398,23 +6468,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7426,13 +6495,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7446,7 +6511,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7465,22 +6529,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7492,13 +6555,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7527,7 +6586,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7551,7 +6609,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7559,12 +6617,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7575,14 +6632,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7592,11 +6649,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7608,10 +6665,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -7635,23 +6688,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7661,11 +6713,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7691,14 +6739,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7727,7 +6767,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7736,17 +6776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7765,22 +6801,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7790,11 +6825,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -7808,14 +6839,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7843,7 +6866,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7852,17 +6875,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7881,22 +6900,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7906,11 +6924,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -7924,14 +6938,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7959,7 +6965,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7968,17 +6974,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7997,22 +6999,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8022,11 +7023,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -8041,14 +7038,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8076,7 +7065,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8085,17 +7074,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -8120,7 +7105,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8139,23 +7123,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8167,13 +7150,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8192,22 +7171,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8217,11 +7195,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8235,14 +7209,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8270,7 +7236,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8279,17 +7245,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8303,7 +7265,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8322,22 +7283,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8347,18 +7307,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -8384,7 +7340,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8393,16 +7349,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8416,13 +7371,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8448,14 +7399,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8469,7 +7412,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8483,18 +7426,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8508,7 +7443,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -8520,19 +7455,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8566,7 +7490,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8578,17 +7502,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -8616,7 +7538,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8642,7 +7564,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8651,16 +7573,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8680,10 +7601,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8710,14 +7627,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8731,7 +7640,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8747,18 +7656,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8772,7 +7673,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8785,20 +7686,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8832,7 +7722,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8842,24 +7732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8885,7 +7773,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8894,16 +7782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8915,13 +7802,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8950,14 +7833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8971,7 +7846,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8987,18 +7862,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9012,7 +7879,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -9025,20 +7892,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9072,7 +7928,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9082,17 +7938,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -9111,22 +7965,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9136,11 +7989,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9155,14 +8004,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9190,7 +8031,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9199,17 +8040,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9231,7 +8068,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9250,23 +8086,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9278,13 +8113,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9298,7 +8129,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9317,22 +8147,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9344,13 +8173,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -9380,7 +8205,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -9403,16 +8227,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9423,14 +8246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9440,11 +8263,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9456,15 +8279,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9490,7 +8309,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9498,16 +8317,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9521,13 +8339,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9551,14 +8365,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9571,7 +8377,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9584,18 +8390,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9608,7 +8406,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9619,19 +8417,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9664,7 +8452,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9676,20 +8464,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9714,7 +8504,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9722,16 +8512,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9751,10 +8540,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9777,14 +8562,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9797,7 +8574,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9810,18 +8587,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9831,7 +8600,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9841,19 +8614,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9886,7 +8648,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9896,20 +8658,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9934,7 +8698,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9942,16 +8706,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9963,13 +8726,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9992,14 +8751,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10012,7 +8763,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10024,18 +8775,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10045,7 +8788,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -10054,19 +8801,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10099,7 +8835,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10109,17 +8845,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -10138,22 +8872,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10163,11 +8896,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10182,14 +8911,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10217,7 +8938,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10226,17 +8947,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10259,7 +8976,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10278,22 +8994,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10305,13 +9020,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10323,7 +9034,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10342,21 +9052,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10368,13 +9077,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10393,21 +9098,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10417,11 +9121,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10434,14 +9134,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10468,7 +9160,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10477,17 +9169,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -10506,22 +9194,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10531,11 +9218,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10551,14 +9234,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10586,7 +9261,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10595,17 +9270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10622,7 +9293,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10641,21 +9311,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10665,11 +9334,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -10694,7 +9359,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -10717,18 +9381,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10739,14 +9402,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10756,11 +9419,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10772,10 +9435,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -10800,7 +9459,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -10823,18 +9481,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10845,14 +9502,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10862,11 +9519,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10878,8 +9535,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10891,7 +9546,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] @@ -10914,17 +9568,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10935,14 +9588,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10952,11 +9605,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10968,10 +9621,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10989,22 +9638,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11014,11 +9662,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11035,14 +9679,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11070,7 +9706,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11079,17 +9715,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11108,21 +9740,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11132,11 +9763,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11146,14 +9773,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11180,7 +9799,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11189,17 +9808,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11218,21 +9833,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11242,11 +9856,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11256,14 +9866,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11290,7 +9892,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11299,17 +9901,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11328,21 +9926,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11352,11 +9949,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11366,14 +9959,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11400,7 +9985,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11409,17 +9994,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11438,21 +10019,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11462,11 +10042,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11476,14 +10052,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11510,7 +10078,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11519,23 +10087,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11564,7 +10128,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11572,16 +10136,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11591,11 +10154,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11624,14 +10183,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11642,7 +10193,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11660,18 +10211,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11682,7 +10225,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -11697,19 +10240,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11739,7 +10272,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11748,17 +10281,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11779,7 +10308,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11798,22 +10326,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11825,13 +10352,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11850,21 +10373,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11874,11 +10396,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11890,14 +10408,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11924,7 +10434,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11933,17 +10443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11955,7 +10461,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11974,21 +10479,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11998,11 +10502,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12021,21 +10521,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12049,13 +10548,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -12067,14 +10562,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -12104,7 +10591,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12116,23 +10603,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -12158,7 +10643,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12166,16 +10651,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12189,13 +10673,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12220,14 +10700,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12240,7 +10712,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12253,18 +10725,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12277,7 +10741,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12287,20 +10751,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12333,7 +10786,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12345,20 +10798,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12383,7 +10838,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12391,16 +10846,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12414,13 +10868,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12443,14 +10893,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12463,7 +10905,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12475,18 +10917,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12496,7 +10930,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12505,19 +10943,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12550,7 +10977,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12562,17 +10989,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -12611,7 +11036,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12630,7 +11054,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12638,16 +11062,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12659,13 +11082,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12697,7 +11116,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12716,7 +11134,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12726,16 +11144,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12747,13 +11164,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -12765,7 +11178,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12784,21 +11196,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12810,13 +11221,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -12824,7 +11231,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12843,20 +11249,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12868,13 +11273,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -12888,7 +11289,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12907,22 +11307,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12934,13 +11333,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -12952,7 +11347,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12971,21 +11365,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12997,13 +11390,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -13044,7 +11433,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13063,7 +11451,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -13072,16 +11460,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13093,13 +11480,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -13136,7 +11519,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13155,7 +11537,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -13166,16 +11548,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13187,13 +11568,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13208,7 +11585,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13227,22 +11603,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13254,13 +11629,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -13272,7 +11643,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13291,21 +11661,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13317,13 +11686,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13340,7 +11705,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13359,23 +11723,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13387,13 +11750,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13430,7 +11789,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13449,23 +11807,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13477,13 +11834,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -13495,7 +11848,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13514,21 +11866,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13540,13 +11891,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -13574,7 +11921,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13593,7 +11939,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13602,16 +11948,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13623,13 +11968,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -13673,7 +12014,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13684,7 +12024,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13696,11 +12036,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13711,7 +12050,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -13721,12 +12060,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13754,7 +12089,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13762,16 +12097,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13783,13 +12117,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -13812,7 +12142,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13823,7 +12152,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13835,11 +12164,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13850,7 +12178,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -13860,12 +12188,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13893,7 +12217,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13901,16 +12225,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13922,13 +12245,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -13940,7 +12259,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13959,21 +12277,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13983,11 +12300,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -13997,7 +12310,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14016,21 +12328,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14042,13 +12353,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -14059,7 +12366,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet] @@ -14082,16 +12388,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14102,14 +12407,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14119,11 +12424,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14135,17 +12440,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet] @@ -14168,16 +12468,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14188,14 +12487,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14205,11 +12504,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14221,10 +12520,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -14236,7 +12531,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] @@ -14259,17 +12553,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14280,14 +12573,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -14297,11 +12590,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14313,10 +12606,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -14360,7 +12649,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14371,7 +12659,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14384,11 +12672,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14399,7 +12686,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -14409,12 +12696,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14442,7 +12725,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14450,16 +12733,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14479,10 +12761,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -14505,7 +12783,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14516,7 +12793,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14529,11 +12806,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14544,7 +12820,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -14554,12 +12830,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14587,7 +12860,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14595,16 +12868,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14616,13 +12888,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14634,7 +12902,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14653,21 +12920,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14677,11 +12943,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14701,7 +12963,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14720,22 +12981,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14747,19 +13007,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14783,15 +13038,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14802,14 +13056,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14819,11 +13073,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14835,10 +13089,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14847,7 +13097,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14871,16 +13120,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14891,14 +13139,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14908,11 +13156,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14924,10 +13172,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -14972,7 +13216,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14983,7 +13226,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14997,11 +13240,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15012,7 +13254,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -15024,12 +13266,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15057,7 +13295,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15065,16 +13303,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15094,10 +13331,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -15119,7 +13352,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15130,7 +13362,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15142,11 +13374,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15157,7 +13388,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -15167,12 +13398,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15200,7 +13427,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15208,16 +13435,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15229,13 +13455,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15247,7 +13469,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15266,21 +13487,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15290,11 +13510,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15316,7 +13532,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15335,22 +13550,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15362,13 +13576,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15379,7 +13589,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup] @@ -15402,16 +13611,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15422,14 +13630,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15439,11 +13647,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15455,16 +13663,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup] @@ -15487,15 +13690,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15506,14 +13708,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15523,11 +13725,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15539,8 +13741,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -15567,7 +13767,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15593,7 +13793,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15602,16 +13802,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15631,10 +13830,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -15659,14 +13854,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15680,7 +13867,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15694,18 +13881,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15719,7 +13898,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -15731,19 +13910,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15777,7 +13945,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15787,24 +13955,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15830,7 +13996,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15839,16 +14005,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15860,13 +14025,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15892,14 +14053,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15913,7 +14066,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15927,18 +14080,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15952,7 +14097,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -15964,19 +14109,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16010,7 +14144,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16020,17 +14154,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -16049,22 +14181,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16074,11 +14205,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16092,14 +14219,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16127,7 +14246,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16136,17 +14255,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16170,7 +14285,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -16189,23 +14303,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16217,13 +14330,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -16236,7 +14345,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -16259,17 +14367,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16280,14 +14387,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -16297,11 +14404,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16313,10 +14420,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -16334,22 +14437,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16359,11 +14461,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -16380,14 +14478,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16415,7 +14505,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16424,17 +14514,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -16450,7 +14536,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -16472,18 +14557,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16494,14 +14578,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -16511,11 +14595,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16527,10 +14611,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -16557,7 +14637,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16581,15 +14660,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16600,14 +14678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16617,11 +14695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16633,8 +14711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -16644,7 +14720,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16668,16 +14743,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16688,14 +14762,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16705,11 +14779,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16721,10 +14795,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -16746,7 +14816,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16757,7 +14826,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16769,11 +14838,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16784,7 +14852,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -16794,12 +14862,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16827,7 +14891,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16835,16 +14899,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16864,10 +14927,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -16889,7 +14948,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16900,7 +14958,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16912,11 +14970,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16927,7 +14984,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -16937,12 +14994,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16970,7 +15024,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16978,16 +15032,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16999,13 +15052,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -17017,7 +15066,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17036,21 +15084,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17060,11 +15107,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -17076,7 +15119,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -17095,21 +15137,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17121,13 +15162,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -17141,7 +15178,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17160,21 +15196,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17186,13 +15221,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -17216,7 +15247,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17241,7 +15272,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17249,16 +15280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17278,10 +15308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17303,14 +15329,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17321,7 +15339,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17333,18 +15351,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17352,7 +15362,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -17361,19 +15371,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17404,7 +15403,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17414,20 +15413,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17452,7 +15453,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17460,16 +15461,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17489,10 +15489,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -17514,14 +15510,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17532,7 +15520,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17544,18 +15532,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17563,7 +15543,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -17572,19 +15556,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17615,7 +15588,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17625,17 +15598,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -17654,21 +15625,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17678,11 +15648,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17692,14 +15658,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17726,7 +15684,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17735,17 +15693,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17759,7 +15713,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -17778,22 +15731,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17805,13 +15757,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -17819,7 +15767,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17841,16 +15788,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17861,14 +15807,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17878,11 +15824,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17894,16 +15840,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17925,15 +15866,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17944,14 +15884,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17961,11 +15901,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17977,8 +15917,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -18024,7 +15962,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18035,7 +15972,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18050,11 +15987,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18065,7 +16001,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -18077,12 +16013,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18110,7 +16042,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -18118,16 +16050,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18147,10 +16078,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -18173,7 +16100,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18184,7 +16110,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18197,11 +16123,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18212,7 +16137,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -18222,12 +16147,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18255,7 +16177,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -18263,16 +16185,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18284,13 +16205,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -18309,21 +16226,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18333,11 +16249,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18350,14 +16262,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18384,7 +16288,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18393,17 +16297,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18425,7 +16325,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18444,22 +16343,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18471,13 +16369,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -18485,7 +16379,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18509,16 +16402,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18529,14 +16421,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18546,11 +16438,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18562,17 +16454,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18596,15 +16483,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18615,14 +16501,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18632,11 +16518,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18648,8 +16534,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -18676,7 +16560,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -18702,7 +16586,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -18711,16 +16595,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18732,13 +16615,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -18765,14 +16644,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -18785,7 +16656,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18800,18 +16671,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -18824,7 +16687,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -18837,19 +16700,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -18882,7 +16734,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18892,17 +16744,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18921,22 +16771,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18946,11 +16795,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18964,14 +16809,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18999,7 +16836,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19008,17 +16845,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -19041,7 +16874,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -19060,23 +16892,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19088,13 +16919,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -19140,7 +16967,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19151,7 +16977,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19165,11 +16991,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19180,7 +17005,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -19192,12 +17017,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19225,7 +17046,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19233,16 +17054,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19262,10 +17082,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -19289,7 +17105,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19300,7 +17115,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19314,11 +17129,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19329,7 +17143,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -19340,12 +17154,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19373,7 +17184,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19381,16 +17192,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19402,13 +17212,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -19423,7 +17229,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19442,21 +17247,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19466,11 +17270,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -19494,7 +17294,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19513,22 +17312,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19540,13 +17338,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -19567,7 +17361,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19591,17 +17384,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19612,14 +17404,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19629,11 +17421,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19645,10 +17437,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -19662,7 +17450,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19686,16 +17473,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19706,14 +17492,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19723,11 +17509,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19739,10 +17525,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -19769,7 +17551,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -19795,7 +17577,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19804,16 +17586,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19833,10 +17614,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -19863,14 +17640,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19884,7 +17653,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19900,18 +17669,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19925,7 +17686,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -19939,19 +17700,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19985,7 +17735,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19995,24 +17745,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -20038,7 +17786,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -20047,16 +17795,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20070,13 +17817,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20102,14 +17845,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20123,7 +17858,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20137,18 +17872,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20162,7 +17889,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -20174,19 +17901,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20220,7 +17936,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20230,17 +17946,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20263,22 +17977,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20288,11 +18001,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20312,14 +18021,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20347,7 +18048,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20356,17 +18057,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20391,7 +18088,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -20410,23 +18106,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20438,13 +18133,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -20457,7 +18148,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_03_01.models.CapacityReservation] @@ -20480,17 +18170,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20501,14 +18190,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -20518,11 +18207,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20534,10 +18223,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -20560,7 +18245,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20585,23 +18270,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20615,13 +18299,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20643,14 +18323,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20660,7 +18332,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20669,18 +18341,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20690,7 +18354,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20698,19 +18362,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20742,7 +18396,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20755,20 +18409,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20793,23 +18445,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20823,13 +18474,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20850,14 +18497,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20867,7 +18506,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20875,18 +18514,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20896,7 +18527,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20904,19 +18535,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20948,7 +18568,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20961,17 +18581,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -21000,7 +18618,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -21024,16 +18641,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21044,14 +18660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -21061,11 +18677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21077,10 +18693,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -21089,7 +18701,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -21108,21 +18719,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21134,20 +18744,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21173,7 +18779,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21182,16 +18788,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21211,10 +18816,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21241,14 +18842,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21262,7 +18855,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21278,18 +18871,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21303,7 +18888,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21316,19 +18901,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21362,7 +18936,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21372,24 +18946,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21415,7 +18987,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21424,16 +18996,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21445,13 +19016,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21479,14 +19046,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21500,7 +19059,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21516,18 +19075,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21541,7 +19092,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21554,19 +19105,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21600,7 +19141,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21610,17 +19151,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21639,22 +19178,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21664,11 +19202,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21683,14 +19217,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21718,7 +19244,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21727,17 +19253,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21759,7 +19281,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21778,23 +19299,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21806,13 +19326,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21826,7 +19342,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21850,18 +19365,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21872,14 +19386,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21889,11 +19403,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21905,12 +19419,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21936,7 +19446,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21962,7 +19472,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21972,16 +19482,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22001,10 +19510,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -22033,14 +19538,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22055,7 +19552,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22072,18 +19569,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22098,7 +19587,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -22112,19 +19601,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22159,7 +19637,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22169,17 +19647,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -22187,7 +19663,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22213,7 +19689,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22223,16 +19699,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22244,13 +19719,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22280,14 +19751,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22302,7 +19765,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22319,18 +19782,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22345,7 +19800,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22359,19 +19814,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22406,7 +19851,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22416,17 +19861,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22445,23 +19888,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22471,11 +19913,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22491,14 +19929,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22527,7 +19957,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22536,17 +19966,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22571,7 +19997,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22590,7 +20015,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22598,16 +20023,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22619,13 +20043,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22647,7 +20067,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22671,19 +20090,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22694,14 +20112,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22711,11 +20129,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22726,7 +20144,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_compute_management_client_enums.py index 73db15e1da00..57fd9cf91250 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_compute_management_client_enums.py @@ -429,7 +429,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" @@ -475,10 +474,10 @@ class RestorePointEncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk Restore Point is encrypted at rest with Platform managed key.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk Restore Point is encrypted at rest with Customer managed key that can be changed and - #: revoked by a customer.""" + revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk Restore Point is encrypted at rest with 2 layers of encryption. One of the keys is - #: Customer managed and the other key is Platform managed.""" + Customer managed and the other key is Platform managed.""" class RestorePointExpandOptions(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_models_py3.py index 27c935c2450b..acfa482cb9e1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -302,7 +298,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -333,7 +329,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -353,7 +349,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -436,7 +432,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -590,7 +586,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet] @@ -898,7 +894,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1024,7 +1020,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1126,7 +1122,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup] @@ -1286,7 +1282,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.CapacityReservation] @@ -1536,7 +1532,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1737,7 +1733,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1917,7 +1913,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2045,7 +2041,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup] @@ -2078,7 +2074,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group. Minimum api-version: 2022-03-01. :ivar ultra_ssd_enabled: The flag that enables or disables a capability to have UltraSSD @@ -2293,7 +2289,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -2571,17 +2567,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2966,7 +2951,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3144,7 +3129,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_03_01.models.SubResource @@ -3295,7 +3280,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.Image] @@ -3329,7 +3314,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_03_01.models.SubResource @@ -3467,7 +3452,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3526,7 +3511,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3748,7 +3733,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3781,7 +3766,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4086,7 +4071,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -4131,7 +4116,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.Usage] @@ -4165,7 +4150,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4519,7 +4504,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4595,7 +4580,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. Possible values are: **Windows,** @@ -4745,7 +4730,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5282,7 +5267,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5391,7 +5376,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup] @@ -5451,17 +5436,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5502,7 +5476,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2023_03_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2023_03_01.models.PublicIPAddressSkuTier """ @@ -5521,8 +5495,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2023_03_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2023_03_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5533,7 +5506,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5604,7 +5577,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5840,7 +5813,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6618,7 +6591,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6681,7 +6654,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6744,7 +6717,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6818,7 +6791,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6865,7 +6838,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6898,7 +6871,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.RunCommandDocumentBase] @@ -6971,7 +6944,7 @@ def __init__(self, *, client_id: Optional[str] = None, object_id: Optional[str] class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7146,7 +7119,7 @@ class SecurityPostureReference(_serialization.Model): api-version: 2023-03-01. :ivar id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :vartype id: str :ivar exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7168,7 +7141,7 @@ def __init__( ) -> None: """ :keyword id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :paramtype id: str :keyword exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7238,7 +7211,7 @@ class ServiceArtifactReference(_serialization.Model): machines in the scale set when using 'latest' image version. Minimum api-version: 2022-11-01. :ivar id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :vartype id: str """ @@ -7249,7 +7222,7 @@ class ServiceArtifactReference(_serialization.Model): def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :paramtype id: str """ super().__init__(**kwargs) @@ -7389,7 +7362,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7399,7 +7372,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7428,7 +7401,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7443,7 +7416,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7499,7 +7472,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource] @@ -7739,7 +7712,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7760,67 +7733,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7889,7 +7801,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8032,7 +7944,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8250,7 +8162,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8729,7 +8641,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8977,7 +8889,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -9021,7 +8933,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9336,7 +9248,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_03_01.models.UserAssignedIdentitiesValue] """ @@ -9368,7 +9280,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2023_03_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_03_01.models.UserAssignedIdentitiesValue] """ @@ -9382,7 +9294,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9445,7 +9357,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9608,7 +9520,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9930,7 +9842,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -9963,10 +9875,12 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineNetworkInterfaceConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -10082,7 +9996,7 @@ def __init__( self.dscp_configuration = dscp_configuration -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -10102,10 +10016,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -10271,10 +10185,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10374,10 +10288,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -10455,7 +10369,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10794,7 +10708,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand] @@ -11014,7 +10928,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11281,7 +11195,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11557,10 +11471,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension] @@ -11818,7 +11732,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_03_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_03_01.models.UserAssignedIdentitiesValue] """ @@ -11852,7 +11766,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_03_01.models.UserAssignedIdentitiesValue] """ @@ -11906,7 +11820,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11933,7 +11847,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(_serialization.Model): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -12082,10 +11996,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -12127,7 +12041,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet] @@ -12163,7 +12077,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetSku] @@ -12199,7 +12113,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet] @@ -12232,7 +12146,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -12283,10 +12197,10 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): +class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network configuration name. Required. :vartype name: str @@ -12392,7 +12306,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12417,7 +12331,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12448,7 +12362,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12467,7 +12381,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12761,10 +12675,10 @@ def __init__( self.require_guest_provision_signal = require_guest_provision_signal -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12853,10 +12767,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12883,7 +12799,7 @@ def __init__(self, *, domain_name_label: str, **kwargs: Any) -> None: self.domain_name_label = domain_name_label -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12897,33 +12813,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype os_profile: ~azure.mgmt.compute.v2023_03_01.models.OSProfileProvisioningData """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - "exact_version": {"key": "exactVersion", "type": "str"}, - "os_profile": {"key": "osProfile", "type": "OSProfileProvisioningData"}, - } - - def __init__( - self, - *, - temp_disk: Optional[bool] = None, - exact_version: Optional[str] = None, - os_profile: Optional["_models.OSProfileProvisioningData"] = None, - **kwargs: Any - ) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - :keyword exact_version: Specifies in decimal number, the version the OS disk should be reimaged - to. If exact version is not provided, the OS disk is reimaged to the existing version of OS - Disk. - :paramtype exact_version: str - :keyword os_profile: Specifies information required for reimaging the non-ephemeral OS disk. - :paramtype os_profile: ~azure.mgmt.compute.v2023_03_01.models.OSProfileProvisioningData - """ - super().__init__(temp_disk=temp_disk, exact_version=exact_version, os_profile=os_profile, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -13277,7 +13166,7 @@ def __init__( self.spot_restore_policy = spot_restore_policy -class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -13389,7 +13278,7 @@ def __init__( self.load_balancer_inbound_nat_pools = load_balancer_inbound_nat_pools -class VirtualMachineScaleSetUpdateNetworkConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar name: The network configuration name. @@ -13493,12 +13382,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_03_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -13531,7 +13420,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_03_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13689,7 +13578,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -13749,7 +13638,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -13915,7 +13804,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14328,7 +14217,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -14351,7 +14240,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -14534,10 +14423,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -14601,7 +14490,7 @@ class VirtualMachineScaleSetVMInstanceView(_serialization.Model): # pylint: dis :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar os_name: The Operating System running on the hybrid machine. :vartype os_name: str @@ -14686,7 +14575,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword os_name: The Operating System running on the hybrid machine. :paramtype os_name: str @@ -14718,7 +14607,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM] @@ -14751,7 +14640,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -15609,14 +15498,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -15661,7 +15550,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -15717,7 +15606,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -15941,7 +15830,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/operations/_operations.py index 1f060ceb0d24..d32be3e6587b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_03_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -176,7 +176,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -210,7 +210,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -248,7 +248,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -286,7 +286,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -327,7 +327,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -368,7 +368,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -399,7 +399,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -429,7 +429,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -455,7 +457,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -486,7 +488,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -517,7 +519,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -558,7 +560,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +594,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -626,7 +628,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reapply_request( +def build_virtual_machine_scale_sets_reapply_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -657,7 +659,7 @@ def build_virtual_machine_scale_sets_reapply_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -691,7 +693,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -725,7 +727,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -759,7 +761,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -793,7 +795,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -827,7 +829,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -870,7 +872,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -904,7 +906,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -938,7 +940,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -973,7 +975,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1008,7 +1010,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1040,7 +1042,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1080,7 +1082,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1111,7 +1113,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1142,7 +1144,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1173,7 +1175,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1204,7 +1206,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1235,7 +1237,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1276,7 +1278,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1317,7 +1319,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1355,7 +1357,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1397,7 +1399,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1437,7 +1439,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1472,7 +1474,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1504,7 +1506,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1536,7 +1538,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1571,7 +1573,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1611,7 +1613,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1651,7 +1653,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1683,7 +1685,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1729,7 +1731,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1769,7 +1771,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1801,7 +1803,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1833,7 +1835,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1865,7 +1867,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1907,7 +1909,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1939,7 +1941,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1971,7 +1973,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2006,7 +2008,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2041,7 +2043,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2076,7 +2078,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2108,7 +2110,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2148,7 +2150,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2181,7 +2183,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2243,7 +2247,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2387,7 +2391,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2418,7 +2422,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2449,7 +2453,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2482,7 +2486,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2589,7 +2593,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2811,7 +2815,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2851,7 +2855,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2882,7 +2886,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2913,7 +2917,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2944,7 +2948,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2978,7 +2982,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3046,7 +3050,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3094,7 +3098,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3125,7 +3129,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3154,7 +3158,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3186,7 +3190,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3217,7 +3221,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3259,7 +3263,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3309,7 +3313,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3341,7 +3345,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3372,7 +3376,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3405,7 +3409,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3438,7 +3442,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3469,7 +3473,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3515,7 +3519,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3645,7 +3649,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3701,7 +3705,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3732,7 +3736,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3768,7 +3772,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3804,7 +3808,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3837,7 +3841,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -3879,7 +3883,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3905,7 +3911,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3935,7 +3941,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3969,7 +3975,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4003,7 +4009,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4072,7 +4078,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4102,7 +4108,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4126,7 +4134,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4268,7 +4276,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4331,7 +4339,7 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_available_sizes_request( +def build_dedicated_hosts_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4363,7 +4371,9 @@ def build_dedicated_hosts_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4387,7 +4397,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4547,7 +4557,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4710,7 +4720,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4764,7 +4774,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4800,7 +4810,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4836,7 +4846,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4869,7 +4879,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4909,7 +4919,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4939,7 +4949,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5086,7 +5098,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5122,7 +5134,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5158,7 +5170,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5191,7 +5203,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5231,7 +5243,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5267,7 +5279,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5300,7 +5312,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5341,7 +5353,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5382,7 +5394,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5462,7 +5474,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5495,7 +5507,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5528,7 +5540,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5561,7 +5573,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5588,7 +5602,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5619,7 +5633,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5654,7 +5668,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5689,7 +5703,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5721,7 +5735,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5761,7 +5775,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5794,7 +5808,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5835,7 +5849,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5876,7 +5890,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5914,7 +5928,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5956,7 +5970,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6020,7 +6034,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -6044,14 +6057,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6062,14 +6074,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -6079,11 +6091,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6095,8 +6107,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6125,7 +6135,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6147,16 +6156,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6167,14 +6175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6184,11 +6192,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6200,8 +6208,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6230,7 +6236,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6252,16 +6257,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6272,14 +6276,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6289,11 +6293,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6305,8 +6309,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6335,7 +6337,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6359,16 +6360,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6379,14 +6379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6396,11 +6396,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6412,15 +6412,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6446,7 +6442,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6454,16 +6450,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6483,10 +6478,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6508,14 +6499,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6528,7 +6511,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6540,18 +6523,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6564,7 +6539,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6573,20 +6548,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6619,7 +6583,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6629,23 +6593,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6671,7 +6633,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6679,16 +6641,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6700,13 +6661,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6729,14 +6686,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6749,7 +6698,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6761,18 +6710,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6785,7 +6726,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6794,20 +6735,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6840,7 +6771,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6850,17 +6781,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6879,22 +6808,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6904,11 +6832,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6923,14 +6847,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6958,7 +6874,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6967,17 +6883,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6998,7 +6910,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7017,22 +6928,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7044,19 +6954,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7087,7 +6993,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7096,16 +7002,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7115,11 +7020,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7151,14 +7052,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7169,7 +7062,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -7185,7 +7078,7 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -7193,14 +7086,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7211,7 +7096,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7225,24 +7110,13 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7273,7 +7147,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7282,23 +7156,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7326,7 +7196,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7335,16 +7205,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7354,11 +7223,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7387,14 +7252,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7405,7 +7262,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7419,21 +7276,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7444,7 +7293,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7456,23 +7305,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7503,7 +7341,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7512,17 +7350,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7534,7 +7368,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7553,21 +7386,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7579,13 +7411,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7593,7 +7421,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7617,16 +7444,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7637,14 +7463,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7654,11 +7480,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7670,17 +7496,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7704,15 +7525,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7723,14 +7543,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7740,11 +7560,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7756,8 +7576,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7769,7 +7587,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7793,17 +7610,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7814,14 +7630,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7831,11 +7647,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7847,10 +7663,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7861,7 +7673,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7885,17 +7696,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7906,14 +7716,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7923,11 +7733,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7939,15 +7749,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -7978,7 +7784,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7987,16 +7793,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8006,11 +7811,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -8042,14 +7843,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8060,7 +7853,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -8076,7 +7869,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -8084,14 +7877,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8102,7 +7887,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8116,24 +7901,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8164,7 +7938,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8173,23 +7947,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8218,7 +7988,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8226,16 +7996,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8245,11 +8014,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8274,14 +8039,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8292,7 +8049,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8305,18 +8062,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8327,7 +8076,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8337,20 +8086,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8380,7 +8118,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8389,23 +8127,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8434,7 +8168,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8442,16 +8176,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8461,11 +8194,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8490,14 +8219,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8508,7 +8229,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8521,18 +8242,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8543,7 +8256,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8553,20 +8266,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8596,7 +8298,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8605,17 +8307,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8634,21 +8332,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8662,11 +8359,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8677,14 +8370,6 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8711,7 +8396,7 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -8722,23 +8407,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8767,7 +8448,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8775,16 +8456,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8794,11 +8474,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -8824,14 +8500,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8842,7 +8510,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8856,18 +8524,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8878,7 +8538,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8889,20 +8549,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8932,7 +8581,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8941,23 +8590,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8986,7 +8631,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8994,16 +8639,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9013,11 +8657,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -9045,14 +8685,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9063,7 +8695,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9079,18 +8711,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9101,7 +8725,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -9114,20 +8738,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9157,7 +8770,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9166,23 +8779,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9208,7 +8817,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9216,16 +8825,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9235,11 +8843,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9264,14 +8868,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9282,7 +8878,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9295,18 +8891,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9317,7 +8905,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9327,20 +8915,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9370,7 +8947,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9379,23 +8956,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9424,7 +8997,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9432,16 +9005,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9451,11 +9023,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9481,14 +9049,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9499,7 +9059,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9513,18 +9073,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9535,7 +9087,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9547,20 +9099,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9590,7 +9131,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9599,23 +9140,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9644,7 +9181,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9652,16 +9189,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9671,11 +9207,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9701,14 +9233,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9719,7 +9243,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9733,18 +9257,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9755,7 +9271,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9766,20 +9282,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9809,7 +9314,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9818,20 +9323,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -9857,7 +9358,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9876,7 +9376,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9884,16 +9384,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9905,13 +9404,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -9936,7 +9431,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9947,7 +9441,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9960,11 +9454,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9975,7 +9468,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -9986,13 +9479,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_03_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10020,7 +9510,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10028,16 +9518,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10047,17 +9536,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10083,7 +9568,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10091,16 +9576,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10110,11 +9594,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -10138,14 +9618,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10156,7 +9628,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10169,18 +9641,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10191,7 +9655,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -10202,19 +9666,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10244,7 +9698,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10253,20 +9707,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10291,7 +9741,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10317,7 +9767,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10326,16 +9776,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10355,10 +9804,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10386,14 +9831,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10407,7 +9844,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10423,18 +9860,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10448,7 +9877,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10461,20 +9890,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10508,7 +9926,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10518,24 +9936,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10561,7 +9977,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10570,16 +9986,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10599,10 +10014,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -10630,14 +10041,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10651,7 +10054,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10667,18 +10070,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10692,7 +10087,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -10705,20 +10100,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10752,7 +10137,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10762,17 +10147,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10791,22 +10174,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10816,11 +10198,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10835,14 +10213,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10870,7 +10240,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10879,17 +10249,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10911,7 +10277,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10930,23 +10295,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10958,13 +10322,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10976,7 +10336,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11000,17 +10359,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11021,14 +10379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -11038,11 +10396,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11054,12 +10412,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11096,21 +10450,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11120,11 +10473,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11134,14 +10483,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11168,7 +10509,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11177,17 +10518,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11206,21 +10543,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11230,11 +10566,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11248,14 +10580,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11282,7 +10606,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11291,17 +10615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11320,21 +10640,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11344,11 +10663,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11362,14 +10677,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11396,7 +10703,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11405,17 +10712,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11427,7 +10730,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11446,21 +10748,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11472,16 +10773,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11507,7 +10804,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11533,7 +10830,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11543,16 +10840,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11572,10 +10868,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11605,14 +10897,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11627,7 +10911,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11644,18 +10928,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11670,7 +10946,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -11684,20 +10960,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11732,7 +10997,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11742,17 +11007,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -11760,7 +11023,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11786,7 +11049,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11796,16 +11059,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11817,13 +11079,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11854,14 +11112,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11876,7 +11126,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11893,18 +11143,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11919,7 +11161,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -11933,20 +11175,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11981,7 +11213,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11991,17 +11223,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -12020,23 +11250,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12046,11 +11275,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12066,14 +11291,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12102,7 +11319,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12111,17 +11328,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12146,7 +11359,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -12165,7 +11377,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12173,16 +11385,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12194,13 +11405,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12222,7 +11429,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12241,23 +11447,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12269,13 +11474,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12303,7 +11504,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12332,7 +11535,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12341,16 +11544,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12360,11 +11562,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12392,14 +11590,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12411,7 +11601,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12426,18 +11616,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12449,7 +11631,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12461,20 +11645,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12505,7 +11679,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12514,17 +11688,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12543,22 +11713,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12568,11 +11737,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -12587,14 +11752,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12622,7 +11779,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12631,17 +11788,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12660,22 +11813,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12685,11 +11837,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12705,14 +11853,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12740,7 +11880,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12749,24 +11889,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12792,7 +11928,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12801,16 +11937,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12830,10 +11965,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -12860,14 +11991,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12881,7 +12004,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12897,18 +12020,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12922,7 +12037,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12935,19 +12050,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12981,7 +12085,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12991,17 +12095,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -13026,23 +12128,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13052,11 +12153,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13079,14 +12176,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13115,7 +12204,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13124,17 +12213,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13158,7 +12243,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -13177,23 +12261,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13205,13 +12288,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13225,7 +12304,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13244,22 +12322,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13271,13 +12348,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13306,7 +12379,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13330,7 +12402,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13338,12 +12410,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13354,14 +12425,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13371,11 +12442,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13387,10 +12458,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13414,23 +12481,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13440,11 +12506,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13470,14 +12532,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13506,7 +12560,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13515,17 +12569,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13544,22 +12594,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13569,11 +12618,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -13587,14 +12632,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13622,7 +12659,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13631,17 +12668,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13660,22 +12693,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13685,11 +12717,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -13703,14 +12731,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13738,7 +12758,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13747,17 +12767,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13776,22 +12792,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13801,11 +12816,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -13820,14 +12831,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13855,7 +12858,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13864,17 +12867,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13899,7 +12898,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13918,23 +12916,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13946,13 +12943,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13971,22 +12964,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13996,11 +12988,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -14014,14 +13002,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14049,7 +13029,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14058,17 +13038,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -14082,7 +13058,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14101,22 +13076,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14126,18 +13100,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -14163,7 +13133,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14172,16 +13142,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14195,13 +13164,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14227,14 +13192,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -14247,7 +13204,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14261,18 +13218,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -14285,7 +13234,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14297,19 +13246,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -14342,7 +13280,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14354,17 +13292,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -14392,7 +13328,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14418,7 +13354,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14427,16 +13363,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14456,10 +13391,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -14486,14 +13417,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14507,7 +13430,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14523,18 +13446,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14548,7 +13463,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -14561,20 +13476,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14608,7 +13512,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14618,24 +13522,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14661,7 +13563,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14670,16 +13572,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14691,13 +13592,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14726,14 +13623,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14747,7 +13636,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14763,18 +13652,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14788,7 +13669,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -14801,20 +13682,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14848,7 +13718,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14858,17 +13728,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14887,22 +13755,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14912,11 +13779,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14931,14 +13794,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14966,7 +13821,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14975,17 +13830,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15007,7 +13858,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -15026,23 +13876,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15054,13 +13903,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15074,7 +13919,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -15093,22 +13937,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15120,13 +13963,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -15156,7 +13995,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -15178,16 +14016,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15198,14 +14035,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -15215,11 +14052,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15231,15 +14068,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15265,7 +14098,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15273,16 +14106,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15296,13 +14128,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15326,14 +14154,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15346,7 +14166,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15359,18 +14179,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15383,7 +14195,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15394,19 +14206,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15439,7 +14241,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15451,20 +14253,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15489,7 +14293,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15497,16 +14301,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15526,10 +14329,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15552,14 +14351,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15571,7 +14362,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15584,18 +14375,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15604,7 +14387,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15614,19 +14401,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15658,7 +14434,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15668,20 +14444,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15706,7 +14484,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15714,16 +14492,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15735,13 +14512,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15764,14 +14537,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15783,7 +14548,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15795,18 +14560,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15815,7 +14572,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15824,19 +14585,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] @@ -15868,7 +14618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15878,17 +14628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15907,22 +14655,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15932,11 +14679,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15951,14 +14694,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15986,7 +14721,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15995,17 +14730,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -16028,7 +14759,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -16047,22 +14777,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16074,13 +14803,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -16092,7 +14817,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -16111,21 +14835,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16137,13 +14860,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16162,21 +14881,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16186,11 +14904,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16201,14 +14915,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16235,7 +14941,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16244,17 +14950,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -16273,22 +14975,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16298,11 +14999,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -16318,14 +15015,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16353,7 +15042,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16362,17 +15051,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16389,7 +15074,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16408,21 +15092,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16432,11 +15115,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -16461,7 +15140,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16483,18 +15161,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16505,14 +15182,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16522,11 +15199,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16538,10 +15215,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -16566,7 +15239,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16588,18 +15260,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16610,14 +15281,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16627,11 +15298,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16643,8 +15314,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16656,7 +15325,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16678,17 +15346,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16699,14 +15366,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16716,11 +15383,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16732,10 +15399,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16753,22 +15416,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16778,11 +15440,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16799,14 +15457,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16834,7 +15484,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16843,17 +15493,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16872,21 +15518,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16896,11 +15541,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16910,14 +15551,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16944,7 +15577,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16953,17 +15586,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16982,21 +15611,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17006,11 +15634,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17020,14 +15644,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17054,7 +15670,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17063,17 +15679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17092,21 +15704,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17116,11 +15727,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17130,14 +15737,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17164,7 +15763,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17173,17 +15772,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17202,21 +15797,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17226,11 +15820,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17240,14 +15830,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17274,7 +15856,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17283,23 +15865,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17328,7 +15906,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17336,16 +15914,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17355,11 +15932,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17388,14 +15961,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17406,7 +15971,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17424,18 +15989,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17446,7 +16003,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -17461,19 +16018,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17503,7 +16050,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17512,17 +16059,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17543,7 +16086,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17562,22 +16104,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17589,13 +16130,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17614,21 +16151,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17638,11 +16174,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17652,14 +16184,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17686,7 +16210,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17695,17 +16219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17717,7 +16237,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17736,21 +16255,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17760,11 +16278,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17783,21 +16297,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17811,13 +16324,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17829,14 +16338,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17866,7 +16367,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17878,23 +16379,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17920,7 +16419,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17928,16 +16427,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17951,13 +16449,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17982,14 +16476,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18002,7 +16488,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18015,18 +16501,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18039,7 +16517,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -18049,20 +16527,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18095,7 +16562,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18107,20 +16574,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -18145,7 +16614,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18153,16 +16622,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18176,13 +16644,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -18205,14 +16669,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -18224,7 +16680,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18236,18 +16692,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -18256,7 +16704,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18265,19 +16717,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RunCommandResult] @@ -18309,7 +16750,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18321,17 +16762,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -18370,7 +16809,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18389,7 +16827,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18397,16 +16835,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18418,13 +16855,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18456,7 +16889,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18475,7 +16907,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18485,16 +16917,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18506,13 +16937,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18524,7 +16951,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18543,21 +16969,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18569,13 +16994,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -18583,7 +17004,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18602,20 +17022,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18627,13 +17046,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18647,7 +17062,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18666,22 +17080,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18693,13 +17106,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -18709,7 +17118,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18728,21 +17136,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18754,13 +17161,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -18801,7 +17204,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18820,7 +17222,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18829,16 +17231,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18850,13 +17251,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18893,7 +17290,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18912,7 +17308,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18923,16 +17319,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18944,13 +17339,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18965,7 +17356,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18984,22 +17374,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19011,13 +17400,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -19029,7 +17414,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19048,21 +17432,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19074,13 +17457,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -19097,7 +17476,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19116,23 +17494,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19144,13 +17521,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -19187,7 +17560,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -19206,23 +17578,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19234,13 +17605,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -19252,7 +17619,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19271,21 +17637,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19297,13 +17662,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -19331,7 +17692,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19350,7 +17710,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -19359,16 +17719,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19380,13 +17739,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -19430,7 +17785,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19441,7 +17795,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19453,11 +17807,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19468,7 +17821,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -19478,12 +17831,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19511,7 +17860,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19519,16 +17868,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19540,13 +17888,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -19569,7 +17913,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19580,7 +17923,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19592,11 +17935,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19607,7 +17949,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -19617,12 +17959,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19650,7 +17988,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19658,16 +17996,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19679,13 +18016,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -19697,7 +18030,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19716,21 +18048,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19740,11 +18071,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -19754,7 +18081,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19773,21 +18099,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19799,13 +18124,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -19816,7 +18137,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19838,16 +18158,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19858,14 +18177,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19875,11 +18194,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19891,17 +18210,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19923,16 +18237,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19943,14 +18256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19960,11 +18273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19976,10 +18289,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -19991,7 +18300,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -20013,17 +18321,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20034,14 +18341,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -20051,11 +18358,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20067,10 +18374,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -20114,7 +18417,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20125,7 +18427,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20138,11 +18440,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20153,7 +18454,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -20163,12 +18464,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20196,7 +18493,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20204,16 +18501,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20233,10 +18529,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -20259,7 +18551,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20270,7 +18561,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20283,11 +18574,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20298,7 +18588,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -20308,12 +18598,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20341,7 +18628,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20349,16 +18636,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20370,13 +18656,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20388,7 +18670,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20407,21 +18688,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20431,11 +18711,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20455,7 +18731,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20474,22 +18749,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20501,19 +18775,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20537,15 +18806,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20556,14 +18824,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20573,11 +18841,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20589,10 +18857,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -20601,7 +18865,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20625,16 +18888,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20645,14 +18907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20662,11 +18924,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20678,10 +18940,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -20726,7 +18984,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20737,7 +18994,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20751,11 +19008,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20766,7 +19022,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -20778,12 +19034,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20811,7 +19063,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20819,16 +19071,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20848,10 +19099,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -20873,7 +19120,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20884,7 +19130,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20896,11 +19142,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20911,7 +19156,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -20921,12 +19166,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20954,7 +19195,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20962,16 +19203,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20983,13 +19223,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -21001,7 +19237,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21020,21 +19255,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21044,11 +19278,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -21070,7 +19300,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21089,22 +19318,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21116,13 +19344,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -21131,7 +19355,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -21153,16 +19376,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21173,14 +19395,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21190,11 +19412,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21206,16 +19428,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -21237,15 +19454,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21256,14 +19472,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21273,11 +19489,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21289,8 +19505,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -21317,7 +19531,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21343,7 +19557,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21352,16 +19566,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21381,10 +19594,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -21409,14 +19618,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21429,7 +19630,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21443,18 +19644,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21467,7 +19660,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -21479,19 +19672,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21524,7 +19706,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21534,24 +19716,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21577,7 +19757,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21586,16 +19766,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21607,13 +19786,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21639,14 +19814,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21659,7 +19826,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21673,18 +19840,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21697,7 +19856,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -21709,19 +19868,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] @@ -21754,7 +19902,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21764,17 +19912,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -21793,22 +19939,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21818,11 +19963,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21836,14 +19977,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21871,7 +20004,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21880,17 +20013,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21914,7 +20043,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -21933,23 +20061,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21961,13 +20088,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -21980,7 +20103,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -22002,17 +20124,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22023,14 +20144,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -22040,11 +20161,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22056,10 +20177,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -22077,22 +20194,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22102,11 +20218,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -22123,14 +20235,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22158,7 +20262,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22167,17 +20271,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -22193,7 +20293,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -22215,18 +20314,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22237,14 +20335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -22254,11 +20352,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22270,10 +20368,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -22300,7 +20394,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22324,15 +20417,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22343,14 +20435,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22360,11 +20452,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22376,8 +20468,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -22387,7 +20477,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22411,16 +20500,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22431,14 +20519,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22448,11 +20536,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22464,10 +20552,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -22489,7 +20573,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22500,7 +20583,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22512,11 +20595,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22527,7 +20609,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -22537,12 +20619,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22570,7 +20648,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22578,16 +20656,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22607,10 +20684,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -22632,7 +20705,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22643,7 +20715,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22655,11 +20727,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22670,7 +20741,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -22680,12 +20751,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22713,7 +20781,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22721,16 +20789,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22742,13 +20809,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22760,7 +20823,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22779,21 +20841,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22803,11 +20864,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -22817,7 +20874,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22836,21 +20892,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22862,13 +20917,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -22882,7 +20933,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22901,21 +20951,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22927,13 +20976,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -22957,7 +21002,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22982,7 +21027,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22990,16 +21035,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23019,10 +21063,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -23044,14 +21084,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23062,7 +21094,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23074,18 +21106,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23093,7 +21117,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -23102,19 +21126,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23145,7 +21158,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23155,20 +21168,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -23193,7 +21208,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -23201,16 +21216,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23230,10 +21244,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -23255,14 +21265,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23273,7 +21275,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23285,18 +21287,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23304,7 +21298,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -23313,19 +21311,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23356,7 +21343,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23366,17 +21353,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -23395,21 +21380,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23419,11 +21403,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23433,14 +21413,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23467,7 +21439,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23476,17 +21448,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23500,7 +21468,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -23519,22 +21486,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23546,13 +21512,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -23560,7 +21522,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23582,16 +21543,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23602,14 +21562,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23619,11 +21579,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23635,16 +21595,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23666,15 +21621,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23685,14 +21639,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23702,11 +21656,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23718,8 +21672,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23765,7 +21717,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23776,7 +21727,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23791,11 +21742,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23806,7 +21756,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23818,12 +21768,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23851,7 +21797,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23859,16 +21805,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23888,10 +21833,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23914,7 +21855,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23925,7 +21865,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23938,11 +21878,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23953,7 +21892,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23963,12 +21902,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23996,7 +21932,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -24004,16 +21940,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24025,13 +21960,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -24050,21 +21981,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24074,11 +22004,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24091,14 +22017,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24125,7 +22043,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24134,17 +22052,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24166,7 +22080,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -24185,22 +22098,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24212,13 +22124,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -24226,7 +22134,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -24250,16 +22157,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24270,14 +22176,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24287,11 +22193,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24303,17 +22209,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -24337,15 +22238,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24356,14 +22256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24373,11 +22273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24389,8 +22289,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -24417,7 +22315,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -24443,7 +22341,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -24452,16 +22350,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24473,13 +22370,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -24506,14 +22399,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -24526,7 +22411,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24541,18 +22426,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -24565,7 +22442,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -24578,19 +22455,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_03_01.models.RestorePoint] @@ -24623,7 +22489,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24633,17 +22499,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -24662,22 +22526,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24687,11 +22550,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24705,14 +22564,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24740,7 +22591,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24749,17 +22600,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24782,7 +22629,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24801,23 +22647,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24829,13 +22674,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -24881,7 +22722,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24892,7 +22732,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24906,11 +22746,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24921,7 +22760,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -24933,12 +22772,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24966,7 +22801,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24974,16 +22809,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25003,10 +22837,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -25030,7 +22860,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25041,7 +22870,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25055,11 +22884,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25070,7 +22898,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -25081,12 +22909,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25114,7 +22939,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -25122,16 +22947,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25143,13 +22967,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -25164,7 +22984,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -25183,21 +23002,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25207,11 +23025,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -25235,7 +23049,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25254,22 +23067,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25281,13 +23093,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -25308,7 +23116,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25332,17 +23139,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25353,14 +23159,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25370,11 +23176,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25386,10 +23192,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -25403,7 +23205,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25427,16 +23228,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25447,14 +23247,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25464,11 +23264,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25480,10 +23280,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -25510,7 +23306,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -25536,7 +23332,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25545,16 +23341,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25574,10 +23369,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -25604,14 +23395,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25625,7 +23408,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25641,18 +23424,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25666,7 +23441,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -25680,19 +23455,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25726,7 +23490,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25736,24 +23500,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -25779,7 +23541,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25788,16 +23550,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25811,13 +23572,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25843,14 +23600,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25864,7 +23613,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25878,18 +23627,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25903,7 +23644,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -25915,19 +23656,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25961,7 +23691,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25971,17 +23701,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -26004,22 +23732,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26029,11 +23756,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26053,14 +23776,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26088,7 +23803,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26097,17 +23812,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -26132,7 +23843,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_03_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -26151,23 +23861,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26179,13 +23888,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -26198,7 +23903,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_03_01.models.CapacityReservation] @@ -26221,17 +23925,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26242,14 +23945,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -26259,11 +23962,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26275,10 +23978,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -26301,7 +24000,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26326,23 +24025,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26356,13 +24054,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26384,14 +24078,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26401,7 +24087,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26410,18 +24096,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26431,7 +24109,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26439,19 +24117,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26483,7 +24151,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26495,20 +24163,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26533,23 +24199,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26563,13 +24228,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26590,14 +24251,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26607,7 +24260,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26615,18 +24268,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26636,7 +24281,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26644,19 +24289,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_03_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26688,7 +24322,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26700,17 +24334,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26739,7 +24371,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26763,16 +24394,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26783,14 +24413,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26800,11 +24430,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26816,10 +24446,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26828,7 +24454,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26847,21 +24472,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26873,20 +24497,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26912,7 +24532,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26921,16 +24541,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26950,10 +24569,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26980,14 +24595,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27001,7 +24608,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27017,18 +24624,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27042,7 +24641,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -27055,19 +24654,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27101,7 +24689,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27111,24 +24699,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27154,7 +24740,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27163,16 +24749,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27184,13 +24769,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27218,14 +24799,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27239,7 +24812,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27255,18 +24828,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27280,7 +24845,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -27293,19 +24858,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27339,7 +24894,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27349,17 +24904,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27378,22 +24931,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27403,11 +24955,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27422,14 +24970,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27457,7 +24997,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27466,17 +25006,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27498,7 +25034,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27517,23 +25052,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27545,13 +25079,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27565,7 +25095,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27589,18 +25118,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27611,14 +25139,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27628,11 +25156,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27644,12 +25172,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27675,7 +25199,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27701,7 +25225,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27711,16 +25235,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27740,10 +25263,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27772,14 +25291,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27794,7 +25305,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27811,18 +25322,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27837,7 +25340,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27851,19 +25354,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27898,7 +25390,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27908,17 +25400,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27926,7 +25416,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27952,7 +25442,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27962,16 +25452,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27983,13 +25472,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -28019,14 +25504,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28041,7 +25518,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28058,18 +25535,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28084,7 +25553,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -28098,19 +25567,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28145,7 +25604,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28155,17 +25614,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -28184,23 +25641,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28210,11 +25666,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28230,14 +25682,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28266,7 +25710,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28275,17 +25719,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -28310,7 +25750,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_03_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -28329,7 +25768,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-03-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28337,16 +25776,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28358,13 +25796,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28386,7 +25820,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28410,19 +25843,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28433,14 +25865,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28450,11 +25882,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28465,7 +25897,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_compute_management_client.py index 11dc3aae151d..480bed24ed42 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-04-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_configuration.py index 9587ff210270..11e789644ff0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-04-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_metadata.json index bf4614b7de45..167d7124f447 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_compute_management_client.py index b3842ac15ff6..9d080222f45c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-04-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_configuration.py index f7c19754313d..98854e571a44 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-04-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/operations/_operations.py index e6239630eff9..0ae26d870891 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_04_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_04_02.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_04_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_04_02.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -1238,7 +1086,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1115,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1123,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1151,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -1328,14 +1175,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1347,7 +1186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1362,18 +1201,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1382,7 +1213,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -1393,19 +1228,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1437,7 +1261,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1447,23 +1271,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -1489,7 +1311,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1497,16 +1319,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1526,10 +1347,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -1554,14 +1371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1573,7 +1382,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1588,18 +1397,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1611,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -1623,19 +1424,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1667,7 +1457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1677,17 +1467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -1699,7 +1487,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,21 +1505,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1744,13 +1530,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -1769,21 +1551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,11 +1574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1811,14 +1588,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1845,7 +1614,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1854,17 +1623,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -1872,7 +1637,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1895,16 +1659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1915,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -1932,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1948,15 +1711,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -1979,15 +1737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1998,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -2015,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2031,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2045,7 +1800,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2064,21 +1818,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2090,20 +1843,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -2129,7 +1878,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -2138,16 +1887,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2167,12 +1915,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -2200,14 +1944,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2216,12 +1952,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2239,18 +1975,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2259,12 +1987,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -2280,20 +2008,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2327,7 +2044,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2337,17 +2054,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -2363,7 +2078,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -2382,22 +2096,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2409,15 +2122,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -2434,22 +2143,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2459,14 +2167,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -2479,14 +2183,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2514,7 +2210,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2523,17 +2219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -2547,7 +2239,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2571,17 +2262,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2592,14 +2282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -2609,11 +2299,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2625,10 +2315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -2654,7 +2340,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2680,7 +2366,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2374,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2717,10 +2402,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2745,14 +2426,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2765,7 +2438,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2780,18 +2453,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2804,7 +2469,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2816,19 +2481,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2861,7 +2516,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2871,23 +2526,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2913,7 +2566,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2921,16 +2574,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2950,10 +2602,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2978,14 +2626,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2998,7 +2638,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3013,18 +2653,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3037,7 +2669,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3049,19 +2681,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3094,7 +2717,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3104,17 +2727,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3128,7 +2749,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3147,21 +2767,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2792,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3198,21 +2813,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,11 +2836,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3240,14 +2850,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3274,7 +2876,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3283,17 +2885,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3303,7 +2901,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -3326,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3346,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3363,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3379,15 +2975,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -3410,15 +3001,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3429,14 +3019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3446,11 +3036,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3462,8 +3052,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3476,7 +3064,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3498,17 +3085,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3519,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3536,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3552,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -3598,7 +3180,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -3617,23 +3198,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3645,13 +3225,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -3667,7 +3243,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskRestorePoint] @@ -3690,18 +3265,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3712,14 +3286,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -3729,11 +3303,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3745,17 +3319,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3781,7 +3351,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -3791,16 +3361,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3383,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -3852,14 +3417,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -3873,7 +3430,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3892,18 +3449,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -3917,7 +3466,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -3933,19 +3482,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -3979,7 +3517,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3991,17 +3529,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -4025,23 +3561,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4051,11 +3586,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -4078,14 +3609,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4114,7 +3637,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -4125,17 +3648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -4159,7 +3678,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4184,7 +3703,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4192,16 +3711,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4221,10 +3739,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -4248,14 +3762,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4267,7 +3773,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4281,18 +3787,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4301,7 +3799,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -4312,19 +3810,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4356,7 +3843,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4366,20 +3853,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4404,7 +3893,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4412,16 +3901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4441,10 +3929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -4469,14 +3953,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4488,7 +3964,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4503,18 +3979,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4523,7 +3991,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -4534,19 +4006,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4578,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4588,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -4610,7 +4069,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -4629,21 +4087,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4655,13 +4112,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4680,21 +4133,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4704,11 +4156,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4720,14 +4168,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4754,7 +4194,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4763,17 +4203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -4781,7 +4217,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4804,16 +4239,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4824,14 +4258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4841,11 +4275,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4857,15 +4291,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_04_02.models.Snapshot] @@ -4888,15 +4317,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4907,14 +4335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4924,11 +4352,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4368,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4972,7 +4398,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4980,16 +4406,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4428,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5035,14 +4456,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -5054,7 +4467,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5069,18 +4482,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -5092,7 +4497,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -5104,19 +4509,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] @@ -5148,7 +4542,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5160,17 +4554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5189,21 +4581,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5213,11 +4604,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5231,14 +4618,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_compute_management_client_enums.py index 92c7dd7b7a47..10d4a44fa9f0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_compute_management_client_enums.py @@ -32,7 +32,7 @@ class CopyCompletionErrorReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): COPY_SOURCE_NOT_FOUND = "CopySourceNotFound" """Indicates that the source snapshot was deleted while the background copy of the resource - #: created via CopyStart operation was in progress.""" + created via CopyStart operation was in progress.""" class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -40,8 +40,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -55,33 +55,30 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" + disk and upload using write token in both disk and VM guest state""" COPY_FROM_SAN_SNAPSHOT = "CopyFromSanSnapshot" """Create a new disk by exporting from elastic san volume snapshot""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -102,10 +99,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -115,17 +112,17 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -158,19 +155,19 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" PREMIUM_V2_LRS = "PremiumV2_LRS" """Premium SSD v2 locally redundant storage. Best for production and performance-sensitive - #: workloads that consistently require low latency and high IOPS and throughput.""" + workloads that consistently require low latency and high IOPS and throughput.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -178,13 +175,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -249,12 +246,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_models_py3.py index b8e83b24f3ca..8aa04712cad0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/models/_models_py3.py @@ -135,7 +135,7 @@ class CopyCompletionError(_serialization.Model): """Indicates the error details if the background copy of a resource created via the CopyStart operation fails. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar error_code: Indicates the error code if the background copy of a resource created via the CopyStart operation fails. Required. "CopySourceNotFound" @@ -176,11 +176,11 @@ class CreationData(_serialization.Model): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", "CopyFromSanSnapshot", and "Import". + "ImportSecure", "UploadPreparedSecure", and "CopyFromSanSnapshot". :vartype create_option: str or ~azure.mgmt.compute.v2023_04_02.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -258,7 +258,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", "CopyFromSanSnapshot", and "Import". + "CopyStart", "ImportSecure", "UploadPreparedSecure", and "CopyFromSanSnapshot". :paramtype create_option: str or ~azure.mgmt.compute.v2023_04_02.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -313,7 +313,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -362,7 +362,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -731,7 +731,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -808,7 +808,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] @@ -864,7 +864,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -985,7 +985,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -1096,7 +1096,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2023_04_02.models.Disk] @@ -1321,7 +1321,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2023_04_02.models.DiskRestorePoint] @@ -1724,7 +1724,7 @@ class EncryptionSetIdentity(_serialization.Model): :vartype tenant_id: str :ivar user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_04_02.models.UserAssignedIdentitiesValue] """ @@ -1758,7 +1758,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_04_02.models.UserAssignedIdentitiesValue] """ @@ -1772,7 +1772,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1896,7 +1896,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2023_04_02.models.AccessLevel @@ -2035,7 +2035,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -2074,7 +2074,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_04_02.models.SourceVault @@ -2107,7 +2107,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_04_02.models.SourceVault @@ -2399,7 +2399,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2449,7 +2449,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2556,7 +2556,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2823,7 +2823,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2023_04_02.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/operations/_operations.py index e8e06a2e8645..492cfbf5ca13 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_04_02/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -276,7 +276,7 @@ def build_disks_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +406,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +460,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -491,7 +491,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -532,7 +532,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -570,7 +570,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -608,7 +608,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -639,7 +639,7 @@ def build_disk_accesses_list_private_endpoint_connections_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -673,7 +673,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -738,7 +738,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -769,7 +769,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +823,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -894,7 +894,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -932,7 +932,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -975,7 +975,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1139,7 +1139,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_04_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_04_02.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_04_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_04_02.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -2409,7 +2257,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2286,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2294,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2322,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -2499,14 +2346,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2517,7 +2356,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2532,18 +2371,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2551,7 +2382,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -2562,19 +2397,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2605,7 +2429,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2615,23 +2439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -2657,7 +2479,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2665,16 +2487,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2694,10 +2515,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -2722,14 +2539,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2549,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2564,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2578,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -2789,19 +2590,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_04_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2622,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -2864,7 +2652,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -2883,21 +2670,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2909,13 +2695,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2934,21 +2716,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,11 +2739,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2974,14 +2751,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2777,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3017,17 +2786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -3035,7 +2800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3057,16 +2821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3077,14 +2840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3094,11 +2857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3110,15 +2873,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3140,15 +2898,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3176,11 +2933,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2949,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3206,7 +2961,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,21 +2979,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3251,20 +3004,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -3290,7 +3039,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -3299,16 +3048,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3328,12 +3076,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -3361,14 +3105,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3377,12 +3113,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3136,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3420,12 +3148,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -3441,20 +3169,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3488,7 +3205,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3498,17 +3215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -3524,7 +3239,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -3543,22 +3257,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3570,15 +3283,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -3595,22 +3304,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,14 +3328,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -3640,14 +3344,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3675,7 +3371,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3684,17 +3380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -3708,7 +3400,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3732,17 +3423,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3753,14 +3443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -3770,11 +3460,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3786,10 +3476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -3815,7 +3501,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3841,7 +3527,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3849,16 +3535,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3878,10 +3563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3906,14 +3587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -3925,7 +3598,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3940,18 +3613,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -3963,7 +3628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3975,19 +3640,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -4019,7 +3674,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4029,23 +3684,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4071,7 +3724,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4079,16 +3732,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4108,10 +3760,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4136,14 +3784,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -4155,7 +3795,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4170,18 +3810,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -4193,7 +3825,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -4205,19 +3837,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] @@ -4249,7 +3872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4259,17 +3882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4281,7 +3902,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4300,21 +3920,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4326,13 +3945,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4351,21 +3966,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4375,11 +3989,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4391,14 +4001,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4425,7 +4027,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4434,17 +4036,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4452,7 +4050,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4474,16 +4071,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4494,14 +4090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4511,11 +4107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4527,15 +4123,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4557,15 +4148,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4576,14 +4166,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4593,11 +4183,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4609,8 +4199,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4623,7 +4211,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4645,17 +4232,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4683,11 +4269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,10 +4285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -4745,7 +4327,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4764,23 +4345,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4792,13 +4372,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4814,7 +4390,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,18 +4411,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4858,14 +4432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4875,11 +4449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4891,17 +4465,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4927,7 +4497,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4937,16 +4507,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4529,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4998,14 +4563,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5018,7 +4575,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5037,18 +4594,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5061,7 +4610,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5077,19 +4626,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5122,7 +4660,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5134,17 +4672,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5168,23 +4704,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5194,11 +4729,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -5221,14 +4752,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5257,7 +4780,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -5268,17 +4791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -5302,7 +4821,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5327,7 +4846,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5335,16 +4854,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5364,10 +4882,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -5391,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5409,7 +4915,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5423,18 +4929,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,7 +4940,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -5453,19 +4951,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5496,7 +4983,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5506,20 +4993,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5544,7 +5033,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5552,16 +5041,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5581,10 +5069,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -5609,14 +5093,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5627,7 +5103,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5642,18 +5118,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5661,7 +5129,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -5672,19 +5144,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_04_02.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5715,7 +5176,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5725,17 +5186,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -5747,7 +5206,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_04_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -5766,21 +5224,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5792,13 +5249,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5817,21 +5270,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5841,11 +5293,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5857,14 +5305,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5891,7 +5331,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5900,17 +5340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -5918,7 +5354,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5940,16 +5375,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5960,14 +5394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -5977,11 +5411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5993,15 +5427,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_04_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -6023,15 +5452,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6042,14 +5470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -6059,11 +5487,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6075,13 +5503,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6107,7 +5533,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -6115,16 +5541,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6138,13 +5563,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6170,14 +5591,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,7 +5601,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6203,18 +5616,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6225,7 +5630,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -6237,19 +5642,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_04_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_04_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6280,7 +5674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6292,17 +5686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -6321,21 +5713,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-04-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6345,11 +5736,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6361,14 +5748,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_compute_management_client.py index 80d7dcca441b..bb7543f33e1d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_configuration.py index 7d1d3102a97b..89dca8241ecc 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_metadata.json index 3b69bb5ede5e..169083d8a673 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_compute_management_client.py index 1df2136ddb3f..96565dd331c5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-07-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_configuration.py index 796dca269df5..45ed7936baa4 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-07-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/operations/_operations.py index 916ebea9aada..187f75352f77 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -236,7 +236,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -260,14 +259,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -278,14 +276,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -295,11 +293,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -311,8 +309,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -341,7 +337,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -363,16 +358,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -383,14 +377,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -400,11 +394,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -416,8 +410,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -446,7 +438,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] @@ -469,16 +460,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -489,14 +479,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -506,11 +496,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -522,8 +512,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -552,7 +540,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -576,16 +563,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -596,14 +582,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -613,11 +599,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -629,15 +615,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -663,7 +645,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -671,16 +653,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -700,10 +681,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -725,14 +702,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -745,7 +714,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -757,18 +726,10 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -781,7 +742,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -790,20 +751,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -836,7 +786,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -846,23 +796,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -888,7 +836,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -896,16 +844,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -917,13 +864,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -946,14 +889,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -966,7 +901,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -978,18 +913,10 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1002,7 +929,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -1011,20 +938,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1057,7 +974,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1067,17 +984,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1096,22 +1011,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1121,11 +1035,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1140,14 +1050,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1175,7 +1077,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1184,17 +1086,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1215,7 +1113,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1234,22 +1131,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1261,19 +1157,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1304,7 +1196,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1313,16 +1205,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1332,11 +1223,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1368,14 +1255,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1386,7 +1265,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -1402,7 +1281,7 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -1410,14 +1289,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1428,7 +1299,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1442,24 +1313,13 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1490,7 +1350,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1499,23 +1359,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1543,7 +1399,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1552,16 +1408,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1571,11 +1426,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1604,14 +1455,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1622,7 +1465,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1636,21 +1479,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1661,7 +1496,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1673,23 +1508,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1720,7 +1544,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1729,17 +1553,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1751,7 +1571,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1770,21 +1589,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1796,13 +1614,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1810,7 +1624,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1834,16 +1647,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1854,14 +1666,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1871,11 +1683,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1887,17 +1699,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1921,15 +1728,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1940,14 +1746,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -1957,11 +1763,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1973,8 +1779,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -1986,7 +1790,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -2010,17 +1813,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2031,14 +1833,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2048,11 +1850,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2064,10 +1866,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2078,7 +1876,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2102,17 +1899,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2123,14 +1919,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2140,11 +1936,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2156,15 +1952,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2195,7 +1987,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2204,16 +1996,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2223,11 +2014,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2259,14 +2046,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2277,7 +2056,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2293,7 +2072,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2301,14 +2080,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2319,7 +2090,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2333,24 +2104,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2381,7 +2141,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2390,23 +2150,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2435,7 +2191,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2443,16 +2199,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2462,11 +2217,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2491,14 +2242,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2509,7 +2252,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2522,18 +2265,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2544,7 +2279,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2554,20 +2289,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2597,7 +2321,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2606,23 +2330,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2651,7 +2371,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2659,16 +2379,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2678,11 +2397,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2707,14 +2422,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2725,7 +2432,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2738,18 +2445,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2760,7 +2459,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2770,20 +2469,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2813,7 +2501,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2822,17 +2510,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2851,21 +2535,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2879,11 +2562,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async async def begin_reapply( @@ -2896,14 +2575,6 @@ async def begin_reapply( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2930,7 +2601,7 @@ async def begin_reapply( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -2941,23 +2612,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2986,7 +2653,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2994,16 +2661,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3013,11 +2679,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -3043,14 +2705,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3061,7 +2715,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3075,18 +2729,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3097,7 +2743,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -3108,20 +2754,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3151,7 +2786,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3160,23 +2795,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3205,7 +2836,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3213,16 +2844,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3232,11 +2862,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3264,14 +2890,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3282,7 +2900,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3298,18 +2916,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3320,7 +2930,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3333,20 +2943,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3376,7 +2975,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3385,23 +2984,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3427,7 +3022,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3435,16 +3030,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3454,11 +3048,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3483,14 +3073,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3501,7 +3083,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3514,18 +3096,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3536,7 +3110,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3546,20 +3120,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3589,7 +3152,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3598,23 +3161,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3643,7 +3202,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3651,16 +3210,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3670,11 +3228,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3700,14 +3254,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3718,7 +3264,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3732,18 +3278,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3754,7 +3292,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3766,20 +3304,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3809,7 +3336,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3818,23 +3345,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3863,7 +3386,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3871,16 +3394,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3890,11 +3412,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -3920,14 +3438,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3938,7 +3448,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3952,18 +3462,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3974,7 +3476,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -3985,20 +3487,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4028,7 +3519,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4037,20 +3528,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -4076,7 +3563,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -4095,7 +3581,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4103,16 +3589,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4124,13 +3609,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4155,7 +3636,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4166,7 +3646,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4179,11 +3659,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4194,7 +3673,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4205,13 +3684,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4239,7 +3715,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4247,16 +3723,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4266,17 +3741,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4302,7 +3773,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4310,16 +3781,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4329,11 +3799,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4357,14 +3823,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4375,7 +3833,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4388,18 +3846,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4410,7 +3860,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4421,19 +3871,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4463,7 +3903,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4472,20 +3912,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4510,7 +3946,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4536,7 +3972,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4545,16 +3981,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4574,10 +4009,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4605,14 +4036,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4626,7 +4049,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4642,18 +4065,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4667,7 +4082,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4680,20 +4095,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4727,7 +4131,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4737,24 +4141,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4780,7 +4182,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4789,16 +4191,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4818,10 +4219,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -4849,14 +4246,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4870,7 +4259,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4886,18 +4275,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4911,7 +4292,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -4924,20 +4305,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4971,7 +4342,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4981,17 +4352,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -5010,22 +4379,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5035,11 +4403,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -5054,14 +4418,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5089,7 +4445,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5098,17 +4454,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -5130,7 +4482,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5149,23 +4500,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4527,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4541,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5219,17 +4564,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4584,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5257,11 +4601,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,12 +4617,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5315,21 +4655,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5339,11 +4678,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5355,14 +4690,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5389,7 +4716,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5398,17 +4725,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5427,21 +4750,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5451,11 +4773,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5469,14 +4787,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5503,7 +4813,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5512,17 +4822,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5541,21 +4847,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5565,11 +4870,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5583,14 +4884,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5617,7 +4910,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5626,17 +4919,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5648,7 +4937,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5667,21 +4955,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5693,16 +4980,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5728,7 +5011,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -5754,7 +5037,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -5764,16 +5047,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5793,10 +5075,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -5826,14 +5104,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5848,7 +5118,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5865,18 +5135,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5891,7 +5153,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -5905,20 +5167,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -5953,7 +5204,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5963,17 +5214,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -5981,7 +5230,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -6007,7 +5256,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6017,16 +5266,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6038,13 +5286,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6075,14 +5319,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6097,7 +5333,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6114,18 +5350,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6140,7 +5368,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6154,20 +5382,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6202,7 +5420,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6212,17 +5430,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6241,23 +5457,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6267,11 +5482,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6287,14 +5498,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6323,7 +5526,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6332,17 +5535,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6367,7 +5566,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6386,7 +5584,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6394,16 +5592,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6415,13 +5612,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6443,7 +5636,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6462,23 +5654,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6490,13 +5681,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6524,7 +5711,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6553,7 +5742,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6562,16 +5751,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6581,11 +5769,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6613,14 +5797,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6632,7 +5808,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6647,18 +5823,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6670,7 +5838,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6682,20 +5852,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -6726,7 +5886,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6735,17 +5895,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6764,22 +5920,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6789,11 +5944,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -6808,14 +5959,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6843,7 +5986,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6852,17 +5995,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -6881,22 +6020,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6906,11 +6044,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -6926,14 +6060,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6961,7 +6087,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6970,24 +6096,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -7013,7 +6135,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -7022,16 +6144,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7051,10 +6172,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -7081,14 +6198,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7102,7 +6211,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -7118,18 +6227,10 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7143,7 +6244,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -7156,19 +6257,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7202,7 +6292,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7212,17 +6302,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7247,23 +6335,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7273,11 +6360,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7300,14 +6383,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7336,7 +6411,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7345,17 +6420,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7379,7 +6450,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7398,23 +6468,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7426,13 +6495,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7446,7 +6511,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7465,22 +6529,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7492,13 +6555,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7527,7 +6586,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7551,7 +6609,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -7559,12 +6617,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7575,14 +6632,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -7592,11 +6649,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7608,10 +6665,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -7635,23 +6688,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7661,11 +6713,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -7691,14 +6739,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7727,7 +6767,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7736,17 +6776,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7765,22 +6801,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7790,11 +6825,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -7808,14 +6839,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7843,7 +6866,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7852,17 +6875,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7881,22 +6900,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7906,11 +6924,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -7924,14 +6938,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7959,7 +6965,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7968,17 +6974,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7997,22 +6999,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8022,11 +7023,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -8041,14 +7038,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8076,7 +7065,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8085,17 +7074,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -8120,7 +7105,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8139,23 +7123,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8167,13 +7150,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8192,22 +7171,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8217,11 +7195,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8235,14 +7209,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8270,7 +7236,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8279,17 +7245,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8303,7 +7265,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8322,22 +7283,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8347,18 +7307,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -8384,7 +7340,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8393,16 +7349,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8416,13 +7371,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -8448,14 +7399,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8469,7 +7412,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8483,18 +7426,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8508,7 +7443,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -8520,19 +7455,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -8566,7 +7490,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8578,17 +7502,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -8616,7 +7538,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8642,7 +7564,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8651,16 +7573,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8680,10 +7601,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -8710,14 +7627,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8731,7 +7640,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8747,18 +7656,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8772,7 +7673,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -8785,20 +7686,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8832,7 +7722,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -8842,24 +7732,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -8885,7 +7773,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -8894,16 +7782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8915,13 +7802,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -8950,14 +7833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -8971,7 +7846,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8987,18 +7862,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9012,7 +7879,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -9025,20 +7892,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9072,7 +7928,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9082,17 +7938,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -9111,22 +7965,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9136,11 +7989,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9155,14 +8004,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9190,7 +8031,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9199,17 +8040,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9231,7 +8068,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9250,23 +8086,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9278,13 +8113,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -9298,7 +8129,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -9317,22 +8147,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9344,13 +8173,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -9380,7 +8205,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -9403,16 +8227,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -9423,14 +8246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -9440,11 +8263,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9456,15 +8279,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -9490,7 +8309,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9498,16 +8317,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9521,13 +8339,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -9551,14 +8365,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9571,7 +8377,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9584,18 +8390,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9608,7 +8406,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -9619,19 +8417,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -9664,7 +8452,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9676,20 +8464,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9714,7 +8504,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9722,16 +8512,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9751,10 +8540,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -9777,14 +8562,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9797,7 +8574,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9810,18 +8587,10 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9831,7 +8600,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -9841,19 +8614,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -9886,7 +8648,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9896,20 +8658,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -9934,7 +8698,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -9942,16 +8706,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9963,13 +8726,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9992,14 +8751,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10012,7 +8763,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10024,18 +8775,10 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10045,7 +8788,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -10054,19 +8801,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10099,7 +8835,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10109,17 +8845,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -10138,22 +8872,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10163,11 +8896,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10182,14 +8911,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10217,7 +8938,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10226,17 +8947,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -10259,7 +8976,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -10278,22 +8994,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10305,13 +9020,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -10323,7 +9034,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -10342,21 +9052,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10368,13 +9077,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10393,21 +9098,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10417,11 +9121,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -10434,14 +9134,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10468,7 +9160,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10477,17 +9169,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -10506,22 +9194,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10531,11 +9218,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -10551,14 +9234,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10586,7 +9261,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -10595,17 +9270,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -10622,7 +9293,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10641,21 +9311,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10665,11 +9334,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -10694,7 +9359,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -10717,18 +9381,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10739,14 +9402,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10756,11 +9419,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10772,10 +9435,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -10800,7 +9459,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -10823,18 +9481,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10845,14 +9502,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10862,11 +9519,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10878,8 +9535,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -10891,7 +9546,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] @@ -10914,17 +9568,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10935,14 +9588,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -10952,11 +9605,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10968,10 +9621,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -10989,22 +9638,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11014,11 +9662,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11035,14 +9679,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11070,7 +9706,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11079,17 +9715,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11108,21 +9740,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11132,11 +9763,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11146,14 +9773,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11180,7 +9799,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11189,17 +9808,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11218,21 +9833,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11242,11 +9856,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11256,14 +9866,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11290,7 +9892,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11299,17 +9901,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11328,21 +9926,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11352,11 +9949,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11366,14 +9959,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11400,7 +9985,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11409,17 +9994,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11438,21 +10019,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11462,11 +10042,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11476,14 +10052,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11510,7 +10078,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11519,23 +10087,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -11564,7 +10128,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -11572,16 +10136,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11591,11 +10154,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -11624,14 +10183,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11642,7 +10193,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -11660,18 +10211,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11682,7 +10225,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -11697,19 +10240,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -11739,7 +10272,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11748,17 +10281,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -11779,7 +10308,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -11798,22 +10326,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11825,13 +10352,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11850,21 +10373,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11874,11 +10396,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -11890,14 +10408,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11924,7 +10434,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11933,17 +10443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -11955,7 +10461,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11974,21 +10479,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11998,11 +10502,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12021,21 +10521,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12049,13 +10548,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -12067,14 +10562,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -12104,7 +10591,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12116,23 +10603,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -12158,7 +10643,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12166,16 +10651,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12189,13 +10673,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -12220,14 +10700,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12240,7 +10712,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12253,18 +10725,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12277,7 +10741,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -12287,20 +10751,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -12333,7 +10786,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12345,20 +10798,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -12383,7 +10838,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12391,16 +10846,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12414,13 +10868,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -12443,14 +10893,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12463,7 +10905,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12475,18 +10917,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12496,7 +10930,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -12505,19 +10943,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -12550,7 +10977,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12562,17 +10989,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -12611,7 +11036,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -12630,7 +11054,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12638,16 +11062,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12659,13 +11082,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -12697,7 +11116,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12716,7 +11134,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -12726,16 +11144,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12747,13 +11164,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -12765,7 +11178,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12784,21 +11196,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12810,13 +11221,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -12824,7 +11231,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12843,20 +11249,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12868,13 +11273,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -12888,7 +11289,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -12907,22 +11307,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12934,13 +11333,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -12952,7 +11347,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12971,21 +11365,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12997,13 +11390,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -13044,7 +11433,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13063,7 +11451,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -13072,16 +11460,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13093,13 +11480,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -13136,7 +11519,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13155,7 +11537,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -13166,16 +11548,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13187,13 +11568,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13208,7 +11585,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13227,22 +11603,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13254,13 +11629,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -13272,7 +11643,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13291,21 +11661,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13317,13 +11686,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13340,7 +11705,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13359,23 +11723,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13387,13 +11750,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -13430,7 +11789,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13449,23 +11807,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13477,13 +11834,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -13495,7 +11848,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13514,21 +11866,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13540,13 +11891,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -13574,7 +11921,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -13593,7 +11939,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -13602,16 +11948,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13623,13 +11968,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -13673,7 +12014,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13684,7 +12024,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13696,11 +12036,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13711,7 +12050,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -13721,12 +12060,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13754,7 +12089,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13762,16 +12097,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13783,13 +12117,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -13812,7 +12142,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13823,7 +12152,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13835,11 +12164,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13850,7 +12178,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -13860,12 +12188,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -13893,7 +12217,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -13901,16 +12225,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13922,13 +12245,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -13940,7 +12259,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -13959,21 +12277,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13983,11 +12300,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -13997,7 +12310,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14016,21 +12328,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14042,13 +12353,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -14059,7 +12366,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet] @@ -14082,16 +12388,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14102,14 +12407,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14119,11 +12424,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14135,17 +12440,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet] @@ -14168,16 +12468,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14188,14 +12487,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -14205,11 +12504,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14221,10 +12520,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -14236,7 +12531,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] @@ -14259,17 +12553,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14280,14 +12573,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -14297,11 +12590,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14313,10 +12606,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -14360,7 +12649,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14371,7 +12659,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14384,11 +12672,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14399,7 +12686,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -14409,12 +12696,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14442,7 +12725,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14450,16 +12733,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14479,10 +12761,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -14505,7 +12783,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14516,7 +12793,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14529,11 +12806,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14544,7 +12820,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -14554,12 +12830,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14587,7 +12860,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -14595,16 +12868,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14616,13 +12888,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14634,7 +12902,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14653,21 +12920,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14677,11 +12943,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -14701,7 +12963,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14720,22 +12981,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14747,19 +13007,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14783,15 +13038,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14802,14 +13056,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14819,11 +13073,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14835,10 +13089,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -14847,7 +13097,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -14871,16 +13120,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14891,14 +13139,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -14908,11 +13156,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14924,10 +13172,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -14972,7 +13216,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -14983,7 +13226,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14997,11 +13240,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15012,7 +13254,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -15024,12 +13266,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15057,7 +13295,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15065,16 +13303,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15094,10 +13331,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -15119,7 +13352,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15130,7 +13362,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15142,11 +13374,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15157,7 +13388,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -15167,12 +13398,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15200,7 +13427,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -15208,16 +13435,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15229,13 +13455,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15247,7 +13469,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15266,21 +13487,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15290,11 +13510,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15316,7 +13532,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15335,22 +13550,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15362,13 +13576,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -15379,7 +13589,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup] @@ -15402,16 +13611,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15422,14 +13630,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15439,11 +13647,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15455,16 +13663,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup] @@ -15487,15 +13690,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15506,14 +13708,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -15523,11 +13725,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15539,8 +13741,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -15567,7 +13767,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15593,7 +13793,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15602,16 +13802,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15631,10 +13830,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -15659,14 +13854,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15680,7 +13867,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15694,18 +13881,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15719,7 +13898,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -15731,19 +13910,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15777,7 +13945,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15787,24 +13955,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -15830,7 +13996,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -15839,16 +14005,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15860,13 +14025,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -15892,14 +14053,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15913,7 +14066,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15927,18 +14080,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -15952,7 +14097,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -15964,19 +14109,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16010,7 +14144,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16020,17 +14154,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -16049,22 +14181,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16074,11 +14205,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -16092,14 +14219,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16127,7 +14246,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16136,17 +14255,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -16170,7 +14285,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -16189,23 +14303,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16217,13 +14330,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -16236,7 +14345,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -16259,17 +14367,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16280,14 +14387,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -16297,11 +14404,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16313,10 +14420,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -16334,22 +14437,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16359,11 +14461,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -16380,14 +14478,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16415,7 +14505,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -16424,17 +14514,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -16450,7 +14536,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -16472,18 +14557,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16494,14 +14578,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -16511,11 +14595,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16527,10 +14611,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -16557,7 +14637,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16581,15 +14660,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16600,14 +14678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16617,11 +14695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16633,8 +14711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -16644,7 +14720,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -16668,16 +14743,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16688,14 +14762,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -16705,11 +14779,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16721,10 +14795,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -16746,7 +14816,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16757,7 +14826,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16769,11 +14838,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16784,7 +14852,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -16794,12 +14862,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16827,7 +14891,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16835,16 +14899,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16864,10 +14927,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -16889,7 +14948,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16900,7 +14958,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16912,11 +14970,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16927,7 +14984,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -16937,12 +14994,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -16970,7 +15024,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -16978,16 +15032,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16999,13 +15052,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -17017,7 +15066,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17036,21 +15084,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17060,11 +15107,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -17076,7 +15119,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -17095,21 +15137,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17121,13 +15162,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def generate_key_pair( @@ -17141,7 +15178,6 @@ async def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17160,21 +15196,20 @@ async def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17186,13 +15221,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -17216,7 +15247,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17241,7 +15272,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17249,16 +15280,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17278,10 +15308,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -17303,14 +15329,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17321,7 +15339,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17333,18 +15351,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17352,7 +15362,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -17361,19 +15371,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17404,7 +15403,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17414,20 +15413,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -17452,7 +15453,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -17460,16 +15461,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17489,10 +15489,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -17514,14 +15510,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17532,7 +15520,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17544,18 +15532,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17563,7 +15543,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -17572,19 +15556,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17615,7 +15588,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17625,17 +15598,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -17654,21 +15625,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17678,11 +15648,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -17692,14 +15658,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17726,7 +15684,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17735,17 +15693,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17759,7 +15713,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -17778,22 +15731,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17805,13 +15757,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -17820,7 +15768,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17842,16 +15789,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17862,14 +15808,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17879,11 +15825,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17895,16 +15841,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -17926,15 +15867,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17945,14 +15885,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -17962,11 +15902,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17978,8 +15918,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -18025,7 +15963,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18036,7 +15973,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18051,11 +15988,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18066,7 +16002,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -18078,12 +16014,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18111,7 +16043,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -18119,16 +16051,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18148,10 +16079,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -18174,7 +16101,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18185,7 +16111,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18198,11 +16124,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18213,7 +16138,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -18223,12 +16148,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18256,7 +16178,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -18264,16 +16186,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18285,13 +16206,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -18310,21 +16227,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18334,11 +16250,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18351,14 +16263,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18385,7 +16289,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18394,17 +16298,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18426,7 +16326,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -18445,22 +16344,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18472,13 +16370,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -18486,7 +16380,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18510,16 +16403,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18530,14 +16422,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18547,11 +16439,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18563,17 +16455,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -18597,15 +16484,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -18616,14 +16502,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -18633,11 +16519,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18649,8 +16535,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -18677,7 +16561,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -18703,7 +16587,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -18712,16 +16596,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18733,13 +16616,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -18766,14 +16645,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -18786,7 +16657,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18801,18 +16672,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -18825,7 +16688,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -18838,19 +16701,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -18883,7 +16735,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18893,17 +16745,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -18922,22 +16772,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18947,11 +16796,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -18965,14 +16810,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19000,7 +16837,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19009,17 +16846,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -19042,7 +16875,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -19061,23 +16893,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19089,13 +16920,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -19141,7 +16968,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19152,7 +16978,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19166,11 +16992,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19181,7 +17006,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -19193,12 +17018,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19226,7 +17047,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19234,16 +17055,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19263,10 +17083,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -19290,7 +17106,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19301,7 +17116,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19315,11 +17130,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19330,7 +17144,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -19341,12 +17155,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19374,7 +17185,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -19382,16 +17193,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19403,13 +17213,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -19424,7 +17230,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19443,21 +17248,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19467,11 +17271,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -19495,7 +17295,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -19514,22 +17313,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19541,13 +17339,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -19568,7 +17362,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19592,17 +17385,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19613,14 +17405,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19630,11 +17422,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19646,10 +17438,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -19663,7 +17451,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -19687,16 +17474,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19707,14 +17493,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -19724,11 +17510,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19740,10 +17526,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -19770,7 +17552,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -19796,7 +17578,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -19805,16 +17587,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19834,10 +17615,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -19864,14 +17641,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19885,7 +17654,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19901,18 +17670,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19926,7 +17687,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -19940,19 +17701,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -19986,7 +17736,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19996,24 +17746,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -20039,7 +17787,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -20048,16 +17796,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20071,13 +17818,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -20103,14 +17846,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20124,7 +17859,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20138,18 +17873,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20163,7 +17890,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -20175,19 +17902,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -20221,7 +17937,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20231,17 +17947,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -20264,22 +17978,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20289,11 +18002,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20313,14 +18022,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20348,7 +18049,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20357,17 +18058,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20392,7 +18089,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -20411,23 +18107,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20439,13 +18134,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -20458,7 +18149,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_01.models.CapacityReservation] @@ -20481,17 +18171,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20502,14 +18191,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -20519,11 +18208,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20535,10 +18224,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -20561,7 +18246,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20586,23 +18271,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20616,13 +18300,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -20644,14 +18324,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20661,7 +18333,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20670,18 +18342,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20691,7 +18355,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -20699,19 +18363,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20743,7 +18397,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20756,20 +18410,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -20794,23 +18446,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20824,13 +18475,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -20851,14 +18498,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20868,7 +18507,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20876,18 +18515,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20897,7 +18528,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -20905,19 +18536,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -20949,7 +18569,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20962,17 +18582,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -21001,7 +18619,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -21025,16 +18642,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21045,14 +18661,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -21062,11 +18678,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21078,10 +18694,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -21090,7 +18702,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -21109,21 +18720,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21135,20 +18745,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21174,7 +18780,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21183,16 +18789,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21212,10 +18817,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -21242,14 +18843,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21263,7 +18856,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21279,18 +18872,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21304,7 +18889,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -21317,19 +18902,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21363,7 +18937,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21373,24 +18947,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21416,7 +18988,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -21425,16 +18997,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21446,13 +19017,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21480,14 +19047,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21501,7 +19060,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21517,18 +19076,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21542,7 +19093,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -21555,19 +19106,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21601,7 +19142,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21611,17 +19152,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -21640,22 +19179,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21665,11 +19203,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21684,14 +19218,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21719,7 +19245,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21728,17 +19254,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -21760,7 +19282,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -21779,23 +19300,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21807,13 +19327,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -21827,7 +19343,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -21851,18 +19366,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21873,14 +19387,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -21890,11 +19404,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21906,12 +19420,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -21937,7 +19447,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -21963,7 +19473,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -21973,16 +19483,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22002,10 +19511,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -22034,14 +19539,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22056,7 +19553,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22073,18 +19570,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22099,7 +19588,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -22113,19 +19602,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22160,7 +19638,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22170,17 +19648,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -22188,7 +19664,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22214,7 +19690,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22224,16 +19700,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22245,13 +19720,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22281,14 +19752,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22303,7 +19766,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22320,18 +19783,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22346,7 +19801,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -22360,19 +19815,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22407,7 +19852,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22417,17 +19862,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -22446,23 +19889,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22472,11 +19914,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22492,14 +19930,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22528,7 +19958,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22537,17 +19967,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -22572,7 +19998,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -22591,7 +20016,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -22599,16 +20024,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22620,13 +20044,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -22648,7 +20068,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22672,19 +20091,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22695,14 +20113,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -22712,11 +20130,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22727,7 +20145,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_compute_management_client_enums.py index 70fa0ffa1ab3..ef5fd9ac2fde 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_compute_management_client_enums.py @@ -459,7 +459,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" @@ -505,10 +504,10 @@ class RestorePointEncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk Restore Point is encrypted at rest with Platform managed key.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk Restore Point is encrypted at rest with Customer managed key that can be changed and - #: revoked by a customer.""" + revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk Restore Point is encrypted at rest with 2 layers of encryption. One of the keys is - #: Customer managed and the other key is Platform managed.""" + Customer managed and the other key is Platform managed.""" class RestorePointExpandOptions(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_models_py3.py index 94dd2d3ea2e9..300894f449b8 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -302,7 +298,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -333,7 +329,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -353,7 +349,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -436,7 +432,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -493,7 +489,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -590,7 +586,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet] @@ -898,7 +894,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1024,7 +1020,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1126,7 +1122,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup] @@ -1286,7 +1282,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.CapacityReservation] @@ -1536,7 +1532,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1737,7 +1733,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1917,7 +1913,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2045,7 +2041,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup] @@ -2078,7 +2074,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group. Minimum api-version: 2022-03-01. :ivar ultra_ssd_enabled: The flag that enables or disables a capability to have UltraSSD @@ -2293,7 +2289,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -2571,17 +2567,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -2966,7 +2951,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3144,7 +3129,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_07_01.models.SubResource @@ -3295,7 +3280,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.Image] @@ -3329,7 +3314,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_07_01.models.SubResource @@ -3467,7 +3452,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3526,7 +3511,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3748,7 +3733,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3781,7 +3766,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4086,7 +4071,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -4131,7 +4116,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.Usage] @@ -4165,7 +4150,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4519,7 +4504,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4595,7 +4580,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. Possible values are: **Windows,** @@ -4745,7 +4730,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5282,7 +5267,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5391,7 +5376,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup] @@ -5451,17 +5436,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyResource(_serialization.Model): """The resource model definition for an Azure Resource Manager proxy resource. It will not have @@ -5502,7 +5476,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2023_07_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2023_07_01.models.PublicIPAddressSkuTier """ @@ -5521,8 +5495,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2023_07_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2023_07_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5533,7 +5506,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5604,7 +5577,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -5840,7 +5813,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6618,7 +6591,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6681,7 +6654,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6744,7 +6717,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -6818,7 +6791,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -6865,7 +6838,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -6898,7 +6871,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.RunCommandDocumentBase] @@ -6971,7 +6944,7 @@ def __init__(self, *, client_id: Optional[str] = None, object_id: Optional[str] class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7146,7 +7119,7 @@ class SecurityPostureReference(_serialization.Model): api-version: 2023-03-01. :ivar id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :vartype id: str :ivar exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7168,7 +7141,7 @@ def __init__( ) -> None: """ :keyword id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :paramtype id: str :keyword exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7238,7 +7211,7 @@ class ServiceArtifactReference(_serialization.Model): machines in the scale set when using 'latest' image version. Minimum api-version: 2022-11-01. :ivar id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :vartype id: str """ @@ -7249,7 +7222,7 @@ class ServiceArtifactReference(_serialization.Model): def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :paramtype id: str """ super().__init__(**kwargs) @@ -7389,7 +7362,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7399,7 +7372,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7428,7 +7401,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7443,7 +7416,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7499,7 +7472,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource] @@ -7739,7 +7712,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -7760,67 +7733,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -7889,7 +7801,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8032,7 +7944,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8250,7 +8162,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -8729,7 +8641,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -8977,7 +8889,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -9021,7 +8933,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9336,7 +9248,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_07_01.models.UserAssignedIdentitiesValue] """ @@ -9368,7 +9280,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2023_07_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_07_01.models.UserAssignedIdentitiesValue] """ @@ -9382,7 +9294,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9445,7 +9357,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9608,7 +9520,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -9930,7 +9842,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -9963,10 +9875,12 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineNetworkInterfaceConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -10104,7 +10018,7 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -10124,10 +10038,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -10293,10 +10207,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10396,10 +10310,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -10497,7 +10411,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10836,7 +10750,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand] @@ -11056,7 +10970,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11323,7 +11237,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -11599,10 +11513,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension] @@ -11860,7 +11774,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_07_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_07_01.models.UserAssignedIdentitiesValue] """ @@ -11894,7 +11808,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_07_01.models.UserAssignedIdentitiesValue] """ @@ -11948,7 +11862,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -11975,7 +11889,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(_serialization.Model): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -12124,10 +12038,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -12169,7 +12083,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet] @@ -12205,7 +12119,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetSku] @@ -12241,7 +12155,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet] @@ -12274,7 +12188,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -12325,10 +12239,12 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineScaleSetNetworkConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network configuration name. Required. :vartype name: str @@ -12456,7 +12372,7 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12481,7 +12397,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12512,7 +12428,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12531,7 +12447,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12825,10 +12741,10 @@ def __init__( self.require_guest_provision_signal = require_guest_provision_signal -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -12917,10 +12833,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -12967,7 +12885,7 @@ def __init__( self.domain_name_label_scope = domain_name_label_scope -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -12981,33 +12899,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype os_profile: ~azure.mgmt.compute.v2023_07_01.models.OSProfileProvisioningData """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - "exact_version": {"key": "exactVersion", "type": "str"}, - "os_profile": {"key": "osProfile", "type": "OSProfileProvisioningData"}, - } - - def __init__( - self, - *, - temp_disk: Optional[bool] = None, - exact_version: Optional[str] = None, - os_profile: Optional["_models.OSProfileProvisioningData"] = None, - **kwargs: Any - ) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - :keyword exact_version: Specifies in decimal number, the version the OS disk should be reimaged - to. If exact version is not provided, the OS disk is reimaged to the existing version of OS - Disk. - :paramtype exact_version: str - :keyword os_profile: Specifies information required for reimaging the non-ephemeral OS disk. - :paramtype os_profile: ~azure.mgmt.compute.v2023_07_01.models.OSProfileProvisioningData - """ - super().__init__(temp_disk=temp_disk, exact_version=exact_version, os_profile=os_profile, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -13361,7 +13252,7 @@ def __init__( self.spot_restore_policy = spot_restore_policy -class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -13475,7 +13366,7 @@ def __init__( class VirtualMachineScaleSetUpdateNetworkConfiguration( _serialization.Model -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar name: The network configuration name. @@ -13601,12 +13492,12 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_07_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -13639,7 +13530,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_07_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -13797,7 +13688,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -13857,7 +13748,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -14023,7 +13914,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14442,7 +14333,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -14465,7 +14356,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -14648,10 +14539,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -14715,7 +14606,7 @@ class VirtualMachineScaleSetVMInstanceView(_serialization.Model): # pylint: dis :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar os_name: The Operating System running on the hybrid machine. :vartype os_name: str @@ -14800,7 +14691,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword os_name: The Operating System running on the hybrid machine. :paramtype os_name: str @@ -14832,7 +14723,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM] @@ -14865,7 +14756,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -15723,14 +15614,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -15775,7 +15666,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -15831,7 +15722,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -16055,7 +15946,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/operations/_operations.py index 0329d102a523..fdec4ad707bb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -176,7 +176,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -210,7 +210,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -248,7 +248,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -286,7 +286,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -327,7 +327,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -368,7 +368,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -399,7 +399,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -429,7 +429,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -455,7 +457,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -486,7 +488,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -517,7 +519,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -558,7 +560,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +594,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -626,7 +628,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reapply_request( +def build_virtual_machine_scale_sets_reapply_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -657,7 +659,7 @@ def build_virtual_machine_scale_sets_reapply_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -691,7 +693,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -725,7 +727,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -759,7 +761,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -793,7 +795,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -827,7 +829,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -870,7 +872,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -904,7 +906,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -938,7 +940,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -973,7 +975,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1008,7 +1010,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1040,7 +1042,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1080,7 +1082,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1111,7 +1113,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1142,7 +1144,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1173,7 +1175,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1204,7 +1206,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1235,7 +1237,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1276,7 +1278,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1317,7 +1319,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1355,7 +1357,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1397,7 +1399,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1437,7 +1439,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1472,7 +1474,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1504,7 +1506,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1536,7 +1538,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1571,7 +1573,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1611,7 +1613,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1651,7 +1653,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1683,7 +1685,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1729,7 +1731,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1769,7 +1771,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1801,7 +1803,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1833,7 +1835,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1865,7 +1867,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1907,7 +1909,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1939,7 +1941,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1971,7 +1973,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2006,7 +2008,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2041,7 +2043,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2076,7 +2078,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2108,7 +2110,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2148,7 +2150,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2181,7 +2183,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2243,7 +2247,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2387,7 +2391,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2418,7 +2422,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2449,7 +2453,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2482,7 +2486,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2589,7 +2593,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2811,7 +2815,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2851,7 +2855,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2882,7 +2886,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2913,7 +2917,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2944,7 +2948,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2978,7 +2982,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3046,7 +3050,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3094,7 +3098,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3125,7 +3129,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3154,7 +3158,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3186,7 +3190,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3217,7 +3221,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3259,7 +3263,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3309,7 +3313,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3341,7 +3345,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3372,7 +3376,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3405,7 +3409,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3438,7 +3442,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3469,7 +3473,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3515,7 +3519,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3645,7 +3649,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3701,7 +3705,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3732,7 +3736,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3768,7 +3772,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3804,7 +3808,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3837,7 +3841,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -3879,7 +3883,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -3905,7 +3911,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3935,7 +3941,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3969,7 +3975,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4003,7 +4009,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4072,7 +4078,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4102,7 +4108,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4126,7 +4134,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4268,7 +4276,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4331,7 +4339,7 @@ def build_dedicated_hosts_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_available_sizes_request( +def build_dedicated_hosts_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4363,7 +4371,9 @@ def build_dedicated_hosts_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4387,7 +4397,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4547,7 +4557,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4710,7 +4720,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4764,7 +4774,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4800,7 +4810,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4836,7 +4846,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4869,7 +4879,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -4909,7 +4919,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4939,7 +4949,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5086,7 +5098,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5122,7 +5134,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5158,7 +5170,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5191,7 +5203,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5231,7 +5243,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5267,7 +5279,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5300,7 +5312,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5341,7 +5353,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5382,7 +5394,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5462,7 +5474,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5495,7 +5507,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5528,7 +5540,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5561,7 +5573,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5588,7 +5602,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5619,7 +5633,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5654,7 +5668,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5689,7 +5703,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5721,7 +5735,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5761,7 +5775,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5794,7 +5808,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5835,7 +5849,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5876,7 +5890,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5914,7 +5928,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -5956,7 +5970,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6020,7 +6034,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -6044,14 +6057,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6062,14 +6074,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -6079,11 +6091,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6095,8 +6107,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6125,7 +6135,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6147,16 +6156,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6167,14 +6175,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6184,11 +6192,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6200,8 +6208,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6230,7 +6236,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6252,16 +6257,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6272,14 +6276,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6289,11 +6293,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6305,8 +6309,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6335,7 +6337,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6359,16 +6360,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6379,14 +6379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6396,11 +6396,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6412,15 +6412,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6446,7 +6442,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6454,16 +6450,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6483,10 +6478,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6508,14 +6499,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6528,7 +6511,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6540,18 +6523,10 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6564,7 +6539,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Create or update a VM scale set. @@ -6573,20 +6548,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6619,7 +6583,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6629,23 +6593,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSet: error_map = { @@ -6671,7 +6633,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6679,16 +6641,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6700,13 +6661,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -6729,14 +6686,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6749,7 +6698,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6761,18 +6710,10 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6785,7 +6726,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSet]: """Update a VM scale set. @@ -6794,20 +6735,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6840,7 +6771,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6850,17 +6781,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -6879,22 +6808,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6904,11 +6832,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -6923,14 +6847,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6958,7 +6874,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -6967,17 +6883,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -6998,7 +6910,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7017,22 +6928,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7044,19 +6954,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7087,7 +6993,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7096,16 +7002,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7115,11 +7020,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7151,14 +7052,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7169,7 +7062,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -7185,7 +7078,7 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -7193,14 +7086,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7211,7 +7096,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7225,24 +7110,13 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7273,7 +7147,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7282,23 +7156,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7326,7 +7196,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7335,16 +7205,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7354,11 +7223,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7387,14 +7252,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7405,7 +7262,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7419,21 +7276,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7444,7 +7293,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7456,23 +7305,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7503,7 +7341,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7512,17 +7350,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7534,7 +7368,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7553,21 +7386,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7579,13 +7411,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7593,7 +7421,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7617,16 +7444,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7637,14 +7463,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7654,11 +7480,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7670,17 +7496,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7704,15 +7525,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7723,14 +7543,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -7740,11 +7560,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7756,8 +7576,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7769,7 +7587,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -7793,17 +7610,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7814,14 +7630,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -7831,11 +7647,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7847,10 +7663,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -7861,7 +7673,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -7885,17 +7696,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7906,14 +7716,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -7923,11 +7733,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7939,15 +7749,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -7978,7 +7784,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7987,16 +7793,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8006,11 +7811,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -8042,14 +7843,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8060,7 +7853,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -8076,7 +7869,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -8084,14 +7877,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8102,7 +7887,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8116,24 +7901,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8164,7 +7938,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8173,23 +7947,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8218,7 +7988,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8226,16 +7996,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8245,11 +8014,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8274,14 +8039,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8292,7 +8049,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8305,18 +8062,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8327,7 +8076,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8337,20 +8086,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8380,7 +8118,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8389,23 +8127,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8434,7 +8168,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8442,16 +8176,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8461,11 +8194,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8490,14 +8219,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8508,7 +8229,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8521,18 +8242,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8543,7 +8256,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8553,20 +8266,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8596,7 +8298,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8605,17 +8307,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8634,21 +8332,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8662,11 +8359,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8677,14 +8370,6 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8711,7 +8396,7 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -8722,23 +8407,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8767,7 +8448,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8775,16 +8456,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8794,11 +8474,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -8824,14 +8500,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8842,7 +8510,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8856,18 +8524,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8878,7 +8538,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -8889,20 +8549,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8932,7 +8581,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8941,23 +8590,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8986,7 +8631,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8994,16 +8639,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9013,11 +8657,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -9045,14 +8685,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9063,7 +8695,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9079,18 +8711,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9101,7 +8725,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -9114,20 +8738,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9157,7 +8770,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9166,23 +8779,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9208,7 +8817,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9216,16 +8825,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9235,11 +8843,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9264,14 +8868,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9282,7 +8878,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9295,18 +8891,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9317,7 +8905,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9327,20 +8915,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9370,7 +8947,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9379,23 +8956,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9424,7 +8997,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9432,16 +9005,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9451,11 +9023,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9481,14 +9049,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9499,7 +9059,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9513,18 +9073,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9535,7 +9087,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9547,20 +9099,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9590,7 +9131,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9599,23 +9140,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9644,7 +9181,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9652,16 +9189,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9671,11 +9207,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9701,14 +9233,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9719,7 +9243,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9733,18 +9257,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9755,7 +9271,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -9766,20 +9282,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9809,7 +9314,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9818,20 +9323,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -9857,7 +9358,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -9876,7 +9376,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9884,16 +9384,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9905,13 +9404,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -9936,7 +9431,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9947,7 +9441,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9960,11 +9454,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -9975,7 +9468,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -9986,13 +9479,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_07_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10020,7 +9510,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10028,16 +9518,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10047,17 +9536,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10083,7 +9568,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10091,16 +9576,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10110,11 +9594,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -10138,14 +9618,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10156,7 +9628,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10169,18 +9641,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10191,7 +9655,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -10202,19 +9666,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10244,7 +9698,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10253,20 +9707,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10291,7 +9741,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10317,7 +9767,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10326,16 +9776,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10355,10 +9804,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10386,14 +9831,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10407,7 +9844,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10423,18 +9860,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10448,7 +9877,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10461,20 +9890,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10508,7 +9926,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10518,24 +9936,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10561,7 +9977,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10570,16 +9986,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10599,10 +10014,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -10630,14 +10041,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10651,7 +10054,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10667,18 +10070,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10692,7 +10087,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -10705,20 +10100,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10752,7 +10137,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10762,17 +10147,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -10791,22 +10174,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10816,11 +10198,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -10835,14 +10213,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10870,7 +10240,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10879,17 +10249,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -10911,7 +10277,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -10930,23 +10295,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10958,13 +10322,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -10976,7 +10336,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11000,17 +10359,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11021,14 +10379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -11038,11 +10396,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11054,12 +10412,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11096,21 +10450,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11120,11 +10473,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11134,14 +10483,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11168,7 +10509,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11177,17 +10518,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11206,21 +10543,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11230,11 +10566,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11248,14 +10580,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11282,7 +10606,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11291,17 +10615,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11320,21 +10640,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11344,11 +10663,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11362,14 +10677,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11396,7 +10703,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11405,17 +10712,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11427,7 +10730,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11446,21 +10748,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11472,16 +10773,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11507,7 +10804,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11533,7 +10830,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11543,16 +10840,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11572,10 +10868,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -11605,14 +10897,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11627,7 +10911,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11644,18 +10928,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11670,7 +10946,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -11684,20 +10960,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11732,7 +10997,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11742,17 +11007,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -11760,7 +11023,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -11786,7 +11049,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -11796,16 +11059,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11817,13 +11079,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -11854,14 +11112,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11876,7 +11126,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11893,18 +11143,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11919,7 +11161,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -11933,20 +11175,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -11981,7 +11213,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11991,17 +11223,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -12020,23 +11250,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12046,11 +11275,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12066,14 +11291,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12102,7 +11319,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12111,17 +11328,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12146,7 +11359,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -12165,7 +11377,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12173,16 +11385,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12194,13 +11405,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12222,7 +11429,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12241,23 +11447,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12269,13 +11474,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12303,7 +11504,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12332,7 +11535,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12341,16 +11544,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12360,11 +11562,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12392,14 +11590,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12411,7 +11601,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12426,18 +11616,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12449,7 +11631,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12461,20 +11645,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12505,7 +11679,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12514,17 +11688,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12543,22 +11713,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12568,11 +11737,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -12587,14 +11752,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12622,7 +11779,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12631,17 +11788,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -12660,22 +11813,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12685,11 +11837,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -12705,14 +11853,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12740,7 +11880,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12749,24 +11889,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVM: error_map = { @@ -12792,7 +11928,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12801,16 +11937,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12830,10 +11965,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -12860,14 +11991,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12881,7 +12004,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12897,18 +12020,10 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12922,7 +12037,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVM]: """Updates a virtual machine of a VM scale set. @@ -12935,19 +12050,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -12981,7 +12085,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12991,17 +12095,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -13026,23 +12128,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13052,11 +12153,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13079,14 +12176,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13115,7 +12204,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13124,17 +12213,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13158,7 +12243,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -13177,23 +12261,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13205,13 +12288,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13225,7 +12304,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13244,22 +12322,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13271,13 +12348,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13306,7 +12379,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13330,7 +12402,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -13338,12 +12410,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -13354,14 +12425,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -13371,11 +12442,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13387,10 +12458,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -13414,23 +12481,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13440,11 +12506,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -13470,14 +12532,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13506,7 +12560,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13515,17 +12569,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13544,22 +12594,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13569,11 +12618,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -13587,14 +12632,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13622,7 +12659,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13631,17 +12668,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13660,22 +12693,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13685,11 +12717,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -13703,14 +12731,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13738,7 +12758,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13747,17 +12767,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13776,22 +12792,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13801,11 +12816,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -13820,14 +12831,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13855,7 +12858,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13864,17 +12867,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -13899,7 +12898,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13918,23 +12916,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13946,13 +12943,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13971,22 +12964,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13996,11 +12988,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -14014,14 +13002,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14049,7 +13029,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14058,17 +13038,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -14082,7 +13058,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14101,22 +13076,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14126,18 +13100,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -14163,7 +13133,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14172,16 +13142,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14195,13 +13164,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -14227,14 +13192,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -14247,7 +13204,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14261,18 +13218,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -14285,7 +13234,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -14297,19 +13246,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -14342,7 +13280,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14354,17 +13292,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -14392,7 +13328,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14418,7 +13354,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14427,16 +13363,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14456,10 +13391,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -14486,14 +13417,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14507,7 +13430,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14523,18 +13446,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14548,7 +13463,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -14561,20 +13476,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14608,7 +13512,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14618,24 +13522,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -14661,7 +13563,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -14670,16 +13572,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14691,13 +13592,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -14726,14 +13623,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14747,7 +13636,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14763,18 +13652,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14788,7 +13669,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -14801,20 +13682,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -14848,7 +13718,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -14858,17 +13728,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -14887,22 +13755,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14912,11 +13779,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -14931,14 +13794,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14966,7 +13821,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14975,17 +13830,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15007,7 +13858,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -15026,23 +13876,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15054,13 +13903,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15074,7 +13919,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -15093,22 +13937,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15120,13 +13963,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -15156,7 +13995,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -15178,16 +14016,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15198,14 +14035,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -15215,11 +14052,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15231,15 +14068,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -15265,7 +14098,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15273,16 +14106,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15296,13 +14128,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -15326,14 +14154,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15346,7 +14166,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15359,18 +14179,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15383,7 +14195,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -15394,19 +14206,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -15439,7 +14241,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15451,20 +14253,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15489,7 +14293,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15497,16 +14301,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15526,10 +14329,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -15552,14 +14351,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15571,7 +14362,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15584,18 +14375,10 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15604,7 +14387,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachine, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachine, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to create or update a virtual machine. Please note some properties can be set only during virtual machine creation. @@ -15614,19 +14401,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15658,7 +14434,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15668,20 +14444,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> _models.VirtualMachine: error_map = { 401: ClientAuthenticationError, @@ -15706,7 +14484,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -15714,16 +14492,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15735,13 +14512,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15764,14 +14537,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15783,7 +14548,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15795,18 +14560,10 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15815,7 +14572,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.VirtualMachineUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.VirtualMachine]: """The operation to update a virtual machine. @@ -15824,19 +14585,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] @@ -15868,7 +14618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15878,17 +14628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -15907,22 +14655,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15932,11 +14679,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15951,14 +14694,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15986,7 +14721,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15995,17 +14730,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -16028,7 +14759,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -16047,22 +14777,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16074,13 +14803,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -16092,7 +14817,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -16111,21 +14835,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16137,13 +14860,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16162,21 +14881,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16186,11 +14904,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16201,14 +14915,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16235,7 +14941,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16244,17 +14950,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -16273,22 +14975,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16298,11 +14999,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -16318,14 +15015,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16353,7 +15042,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16362,17 +15051,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -16389,7 +15074,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16408,21 +15092,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16432,11 +15115,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -16461,7 +15140,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16483,18 +15161,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16505,14 +15182,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16522,11 +15199,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16538,10 +15215,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -16566,7 +15239,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16588,18 +15260,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16610,14 +15281,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16627,11 +15298,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16643,8 +15314,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16656,7 +15325,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -16678,17 +15346,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16699,14 +15366,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -16716,11 +15383,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16732,10 +15399,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -16753,22 +15416,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16778,11 +15440,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -16799,14 +15457,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16834,7 +15484,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16843,17 +15493,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16872,21 +15518,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16896,11 +15541,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -16910,14 +15551,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16944,7 +15577,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -16953,17 +15586,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -16982,21 +15611,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17006,11 +15634,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17020,14 +15644,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17054,7 +15670,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17063,17 +15679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17092,21 +15704,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17116,11 +15727,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17130,14 +15737,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17164,7 +15763,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17173,17 +15772,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17202,21 +15797,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17226,11 +15820,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17240,14 +15830,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17274,7 +15856,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17283,23 +15865,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -17328,7 +15906,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17336,16 +15914,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17355,11 +15932,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -17388,14 +15961,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17406,7 +15971,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -17424,18 +15989,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17446,7 +16003,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -17461,19 +16018,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -17503,7 +16050,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17512,17 +16059,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -17543,7 +16086,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -17562,22 +16104,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17589,13 +16130,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17614,21 +16151,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17638,11 +16174,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17652,14 +16184,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17686,7 +16210,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17695,17 +16219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -17717,7 +16237,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17736,21 +16255,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17760,11 +16278,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17783,21 +16297,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17811,13 +16324,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -17829,14 +16338,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -17866,7 +16367,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17878,23 +16379,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -17920,7 +16419,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -17928,16 +16427,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17951,13 +16449,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -17982,14 +16476,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18002,7 +16488,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18015,18 +16501,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18039,7 +16517,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -18049,20 +16527,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -18095,7 +16562,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18107,20 +16574,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -18145,7 +16614,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18153,16 +16622,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18176,13 +16644,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -18205,14 +16669,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -18224,7 +16680,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18236,18 +16692,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -18256,7 +16704,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -18265,19 +16717,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RunCommandResult] @@ -18309,7 +16750,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18321,17 +16762,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -18370,7 +16809,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18389,7 +16827,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18397,16 +16835,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18418,13 +16855,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18456,7 +16889,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18475,7 +16907,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -18485,16 +16917,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18506,13 +16937,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18524,7 +16951,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18543,21 +16969,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18569,13 +16994,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -18583,7 +17004,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18602,20 +17022,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18627,13 +17046,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -18647,7 +17062,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18666,22 +17080,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18693,13 +17106,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -18709,7 +17118,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18728,21 +17136,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18754,13 +17161,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -18801,7 +17204,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -18820,7 +17222,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18829,16 +17231,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18850,13 +17251,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -18893,7 +17290,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18912,7 +17308,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -18923,16 +17319,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18944,13 +17339,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -18965,7 +17356,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -18984,22 +17374,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19011,13 +17400,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -19029,7 +17414,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19048,21 +17432,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19074,13 +17457,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -19097,7 +17476,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19116,23 +17494,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19144,13 +17521,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -19187,7 +17560,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -19206,23 +17578,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19234,13 +17605,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -19252,7 +17619,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19271,21 +17637,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19297,13 +17662,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -19331,7 +17692,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -19350,7 +17710,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -19359,16 +17719,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19380,13 +17739,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -19430,7 +17785,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19441,7 +17795,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19453,11 +17807,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19468,7 +17821,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -19478,12 +17831,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19511,7 +17860,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19519,16 +17868,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19540,13 +17888,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -19569,7 +17913,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19580,7 +17923,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19592,11 +17935,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19607,7 +17949,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -19617,12 +17959,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19650,7 +17988,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -19658,16 +17996,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19679,13 +18016,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -19697,7 +18030,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -19716,21 +18048,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19740,11 +18071,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -19754,7 +18081,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -19773,21 +18099,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19799,13 +18124,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -19816,7 +18137,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19838,16 +18158,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19858,14 +18177,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19875,11 +18194,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19891,17 +18210,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -19923,16 +18237,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19943,14 +18256,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -19960,11 +18273,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19976,10 +18289,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -19991,7 +18300,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -20013,17 +18321,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20034,14 +18341,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -20051,11 +18358,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20067,10 +18374,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -20114,7 +18417,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20125,7 +18427,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20138,11 +18440,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20153,7 +18454,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -20163,12 +18464,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20196,7 +18493,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20204,16 +18501,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20233,10 +18529,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -20259,7 +18551,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20270,7 +18561,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20283,11 +18574,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20298,7 +18588,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -20308,12 +18598,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20341,7 +18628,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -20349,16 +18636,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20370,13 +18656,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20388,7 +18670,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20407,21 +18688,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20431,11 +18711,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -20455,7 +18731,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20474,22 +18749,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20501,19 +18775,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20537,15 +18806,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20556,14 +18824,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20573,11 +18841,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20589,10 +18857,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -20601,7 +18865,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -20625,16 +18888,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20645,14 +18907,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -20662,11 +18924,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20678,10 +18940,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -20726,7 +18984,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20737,7 +18994,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20751,11 +19008,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20766,7 +19022,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -20778,12 +19034,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20811,7 +19063,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20819,16 +19071,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20848,10 +19099,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -20873,7 +19120,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20884,7 +19130,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20896,11 +19142,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20911,7 +19156,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -20921,12 +19166,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20954,7 +19195,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -20962,16 +19203,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20983,13 +19223,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -21001,7 +19237,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21020,21 +19255,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21044,11 +19278,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -21070,7 +19300,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21089,22 +19318,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21116,13 +19344,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -21131,7 +19355,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -21153,16 +19376,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21173,14 +19395,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21190,11 +19412,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21206,16 +19428,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -21237,15 +19454,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21256,14 +19472,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -21273,11 +19489,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21289,8 +19505,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -21317,7 +19531,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21343,7 +19557,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21352,16 +19566,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21381,10 +19594,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -21409,14 +19618,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21429,7 +19630,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21443,18 +19644,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21467,7 +19660,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -21479,19 +19672,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21524,7 +19706,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21534,24 +19716,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -21577,7 +19757,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -21586,16 +19766,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21607,13 +19786,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -21639,14 +19814,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21659,7 +19826,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21673,18 +19840,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21697,7 +19856,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -21709,19 +19868,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] @@ -21754,7 +19902,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21764,17 +19912,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -21793,22 +19939,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21818,11 +19963,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -21836,14 +19977,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21871,7 +20004,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -21880,17 +20013,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -21914,7 +20043,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -21933,23 +20061,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21961,13 +20088,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -21980,7 +20103,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -22002,17 +20124,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22023,14 +20144,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -22040,11 +20161,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22056,10 +20177,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -22077,22 +20194,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22102,11 +20218,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -22123,14 +20235,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22158,7 +20262,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -22167,17 +20271,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -22193,7 +20293,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -22215,18 +20314,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22237,14 +20335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -22254,11 +20352,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22270,10 +20368,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -22300,7 +20394,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22324,15 +20417,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22343,14 +20435,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22360,11 +20452,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22376,8 +20468,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -22387,7 +20477,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -22411,16 +20500,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22431,14 +20519,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -22448,11 +20536,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22464,10 +20552,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -22489,7 +20573,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22500,7 +20583,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22512,11 +20595,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22527,7 +20609,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -22537,12 +20619,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22570,7 +20648,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22578,16 +20656,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22607,10 +20684,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -22632,7 +20705,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22643,7 +20715,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22655,11 +20727,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22670,7 +20741,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -22680,12 +20751,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22713,7 +20781,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -22721,16 +20789,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22742,13 +20809,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22760,7 +20823,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22779,21 +20841,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22803,11 +20864,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -22817,7 +20874,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -22836,21 +20892,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22862,13 +20917,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def generate_key_pair( @@ -22882,7 +20933,6 @@ def generate_key_pair( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -22901,21 +20951,20 @@ def generate_key_pair( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.SshPublicKeyGenerateKeyPairResult] = kwargs.pop("cls", None) - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22927,13 +20976,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -22957,7 +21002,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -22982,7 +21027,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -22990,16 +21035,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23019,10 +21063,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -23044,14 +21084,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23062,7 +21094,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23074,18 +21106,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23093,7 +21117,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -23102,19 +21126,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23145,7 +21158,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23155,20 +21168,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -23193,7 +21208,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -23201,16 +21216,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23230,10 +21244,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -23255,14 +21265,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23273,7 +21275,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23285,18 +21287,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23304,7 +21298,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -23313,19 +21311,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23356,7 +21343,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23366,17 +21353,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -23395,21 +21380,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23419,11 +21403,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -23433,14 +21413,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23467,7 +21439,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23476,17 +21448,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23500,7 +21468,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -23519,22 +21486,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23546,13 +21512,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -23561,7 +21523,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23583,16 +21544,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23603,14 +21563,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23620,11 +21580,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23636,16 +21596,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -23667,15 +21622,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23686,14 +21640,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -23703,11 +21657,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23719,8 +21673,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -23766,7 +21718,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23777,7 +21728,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23792,11 +21743,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23807,7 +21757,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -23819,12 +21769,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23852,7 +21798,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -23860,16 +21806,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23889,10 +21834,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -23915,7 +21856,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23926,7 +21866,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23939,11 +21879,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23954,7 +21893,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -23964,12 +21903,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -23997,7 +21933,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -24005,16 +21941,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24026,13 +21961,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -24051,21 +21982,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24075,11 +22005,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24092,14 +22018,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24126,7 +22044,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24135,17 +22053,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24167,7 +22081,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -24186,22 +22099,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24213,13 +22125,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -24227,7 +22135,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -24251,16 +22158,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24271,14 +22177,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24288,11 +22194,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24304,17 +22210,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -24338,15 +22239,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -24357,14 +22257,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -24374,11 +22274,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24390,8 +22290,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -24418,7 +22316,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -24444,7 +22342,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -24453,16 +22351,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24474,13 +22371,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -24507,14 +22400,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -24527,7 +22412,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24542,18 +22427,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -24566,7 +22443,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -24579,19 +22456,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_01.models.RestorePoint] @@ -24624,7 +22490,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24634,17 +22500,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -24663,22 +22527,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24688,11 +22551,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -24706,14 +22565,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24741,7 +22592,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24750,17 +22601,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24783,7 +22630,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -24802,23 +22648,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24830,13 +22675,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -24882,7 +22723,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24893,7 +22733,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24907,11 +22747,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24922,7 +22761,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -24934,12 +22773,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -24967,7 +22802,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -24975,16 +22810,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25004,10 +22838,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -25031,7 +22861,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25042,7 +22871,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25056,11 +22885,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25071,7 +22899,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -25082,12 +22910,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25115,7 +22940,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -25123,16 +22948,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25144,13 +22968,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -25165,7 +22985,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -25184,21 +23003,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25208,11 +23026,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -25236,7 +23050,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -25255,22 +23068,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25282,13 +23094,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -25309,7 +23117,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25333,17 +23140,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25354,14 +23160,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25371,11 +23177,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25387,10 +23193,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -25404,7 +23206,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -25428,16 +23229,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25448,14 +23248,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -25465,11 +23265,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25481,10 +23281,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -25511,7 +23307,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -25537,7 +23333,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25546,16 +23342,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25575,10 +23370,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -25605,14 +23396,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25626,7 +23409,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25642,18 +23425,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25667,7 +23442,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -25681,19 +23456,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25727,7 +23491,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25737,24 +23501,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -25780,7 +23542,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -25789,16 +23551,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25812,13 +23573,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -25844,14 +23601,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25865,7 +23614,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25879,18 +23628,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25904,7 +23645,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -25916,19 +23657,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -25962,7 +23692,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -25972,17 +23702,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -26005,22 +23733,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26030,11 +23757,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26054,14 +23777,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26089,7 +23804,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26098,17 +23813,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -26133,7 +23844,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -26152,23 +23862,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26180,13 +23889,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -26199,7 +23904,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_01.models.CapacityReservation] @@ -26222,17 +23926,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26243,14 +23946,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -26260,11 +23963,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26276,10 +23979,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -26302,7 +24001,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26327,23 +24026,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26357,13 +24055,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -26385,14 +24079,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26402,7 +24088,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26411,18 +24097,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26432,7 +24110,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -26440,19 +24118,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26484,7 +24152,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26496,20 +24164,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -26534,23 +24200,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26564,13 +24229,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -26591,14 +24252,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26608,7 +24261,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26616,18 +24269,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26637,7 +24282,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -26645,19 +24290,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_07_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -26689,7 +24323,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26701,17 +24335,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -26740,7 +24372,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -26764,16 +24395,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26784,14 +24414,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -26801,11 +24431,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26817,10 +24447,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -26829,7 +24455,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -26848,21 +24473,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26874,20 +24498,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -26913,7 +24533,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -26922,16 +24542,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26951,10 +24570,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -26981,14 +24596,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27002,7 +24609,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27018,18 +24625,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27043,7 +24642,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -27056,19 +24655,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27102,7 +24690,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27112,24 +24700,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27155,7 +24741,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -27164,16 +24750,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27185,13 +24770,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27219,14 +24800,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27240,7 +24813,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27256,18 +24829,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27281,7 +24846,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -27294,19 +24859,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27340,7 +24895,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27350,17 +24905,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -27379,22 +24932,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27404,11 +24956,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27423,14 +24971,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27458,7 +24998,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27467,17 +25007,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -27499,7 +25035,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -27518,23 +25053,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27546,13 +25080,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -27566,7 +25096,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27590,18 +25119,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27612,14 +25140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -27629,11 +25157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27645,12 +25173,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -27676,7 +25200,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27702,7 +25226,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27712,16 +25236,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27741,10 +25264,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -27773,14 +25292,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27795,7 +25306,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27812,18 +25323,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27838,7 +25341,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -27852,19 +25355,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -27899,7 +25391,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27909,17 +25401,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -27927,7 +25417,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -27953,7 +25443,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -27963,16 +25453,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27984,13 +25473,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -28020,14 +25505,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28042,7 +25519,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28059,18 +25536,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28085,7 +25554,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -28099,19 +25568,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28146,7 +25605,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28156,17 +25615,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -28185,23 +25642,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28211,11 +25667,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28231,14 +25683,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28267,7 +25711,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28276,17 +25720,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -28311,7 +25751,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -28330,7 +25769,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -28338,16 +25777,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28359,13 +25797,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -28387,7 +25821,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28411,19 +25844,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28434,14 +25866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -28451,11 +25883,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28466,7 +25898,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_compute_management_client.py index fef24235f4fe..b66baa7d704e 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -95,7 +97,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -138,7 +158,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-07-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -158,7 +178,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_configuration.py index 89a1413791ba..b7651da109e1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-07-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_metadata.json index ee7040f75301..bfc629d981f3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "galleries": "GalleriesOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_compute_management_client.py index 3aa9a6e4ce22..5a27058afc15 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -96,7 +98,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -139,7 +159,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-07-03" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -159,7 +181,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_configuration.py index 89c883ae7f01..51366c97105b 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-07-03") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/operations/_operations.py index 9b5f0c7e2b5d..1fd6035e28a9 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -162,10 +161,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload async def begin_create_or_update( self, @@ -190,14 +185,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -209,7 +196,7 @@ async def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -224,18 +211,10 @@ async def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -244,7 +223,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -255,19 +234,8 @@ async def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2023_07_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2023_07_03.models.Gallery or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -299,7 +267,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -309,20 +277,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -347,7 +317,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -355,16 +325,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,13 +345,9 @@ async def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -407,14 +372,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -426,7 +383,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -440,18 +397,10 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -460,7 +409,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -471,19 +424,8 @@ async def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2023_07_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2023_07_03.models.GalleryUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -515,7 +457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -525,17 +467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -559,7 +499,6 @@ async def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -578,23 +517,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -606,13 +544,9 @@ async def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -631,21 +565,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -655,11 +588,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -669,14 +598,6 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -703,7 +624,7 @@ async def begin_delete(self, resource_group_name: str, gallery_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -712,17 +633,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: @@ -730,7 +647,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -752,16 +668,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -772,14 +687,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -789,11 +704,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -805,15 +720,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -835,15 +745,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -854,14 +763,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -871,11 +780,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -887,8 +796,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -915,7 +822,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -941,7 +848,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -950,16 +857,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -982,10 +888,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload async def begin_create_or_update( self, @@ -1014,14 +916,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1034,7 +928,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1052,18 +946,10 @@ async def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1076,7 +962,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -1091,19 +977,8 @@ async def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1136,7 +1011,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1146,24 +1021,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -1189,7 +1062,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1198,16 +1071,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1219,13 +1091,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1254,14 +1122,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1274,7 +1134,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1291,18 +1151,10 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1315,7 +1167,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -1330,19 +1182,8 @@ async def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1375,7 +1216,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1385,17 +1226,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -1410,7 +1249,6 @@ async def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -1429,22 +1267,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1456,13 +1293,9 @@ async def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -1481,22 +1314,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1506,11 +1338,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1525,14 +1353,6 @@ async def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1560,7 +1380,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1569,17 +1389,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -1592,7 +1408,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -1615,17 +1430,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1636,14 +1450,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -1653,11 +1467,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,10 +1483,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -1700,7 +1510,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1726,7 +1536,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -1736,16 +1546,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1768,10 +1577,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload async def begin_create_or_update( self, @@ -1805,14 +1610,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1827,7 +1624,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1849,18 +1646,10 @@ async def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1875,7 +1664,7 @@ async def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -1894,19 +1683,9 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -1941,7 +1720,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1951,17 +1730,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -1969,7 +1746,7 @@ async def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -1995,7 +1772,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2005,16 +1782,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2026,13 +1802,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2067,14 +1839,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2089,7 +1853,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2111,18 +1875,10 @@ async def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2137,7 +1893,7 @@ async def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -2156,20 +1912,9 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -2204,7 +1949,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2214,17 +1959,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -2253,7 +1996,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -2272,7 +2014,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2280,16 +2022,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2301,13 +2042,9 @@ async def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -2331,23 +2068,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2357,11 +2093,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -2385,14 +2117,6 @@ async def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2421,7 +2145,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2430,17 +2154,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -2456,7 +2176,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion] @@ -2479,18 +2198,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2501,14 +2219,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -2518,11 +2236,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2534,10 +2252,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -2564,7 +2278,7 @@ async def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2590,7 +2304,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2599,16 +2313,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2631,10 +2344,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload async def begin_create_or_update( self, @@ -2663,14 +2372,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2684,7 +2385,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2702,18 +2403,10 @@ async def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2727,7 +2420,7 @@ async def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -2742,19 +2435,9 @@ async def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication or + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2788,7 +2471,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2798,24 +2481,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -2841,7 +2522,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -2850,16 +2531,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2871,13 +2551,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -2907,14 +2583,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2928,7 +2596,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2946,18 +2614,10 @@ async def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -2971,7 +2631,7 @@ async def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -2986,20 +2646,9 @@ async def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3033,7 +2682,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3043,17 +2692,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3069,7 +2716,6 @@ async def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -3088,22 +2734,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3115,13 +2760,9 @@ async def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -3140,22 +2781,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3165,11 +2805,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3185,14 +2821,6 @@ async def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3220,7 +2848,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3229,17 +2857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -3252,7 +2876,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryApplication] @@ -3275,17 +2898,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3296,14 +2918,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -3313,11 +2935,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3329,10 +2951,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -3360,7 +2978,7 @@ async def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3386,7 +3004,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3396,16 +3014,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3428,10 +3045,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload async def begin_create_or_update( self, @@ -3466,14 +3079,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3488,7 +3093,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3510,18 +3115,10 @@ async def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3536,7 +3133,7 @@ async def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -3555,21 +3152,10 @@ async def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3604,7 +3190,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3614,17 +3200,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -3632,7 +3216,7 @@ async def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -3658,7 +3242,7 @@ async def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3668,16 +3252,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3689,13 +3272,9 @@ async def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -3731,14 +3310,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3753,7 +3324,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3775,18 +3346,10 @@ async def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3801,7 +3364,7 @@ async def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -3820,20 +3383,10 @@ async def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -3868,7 +3421,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3878,17 +3431,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3917,7 +3468,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3936,7 +3486,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3944,16 +3494,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3965,13 +3514,9 @@ async def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3995,23 +3540,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4021,11 +3565,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -4049,14 +3589,6 @@ async def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4085,7 +3617,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4094,17 +3626,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -4120,7 +3648,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4144,18 +3671,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4166,14 +3692,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -4183,11 +3709,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4199,10 +3725,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -4228,7 +3750,7 @@ async def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -4254,7 +3776,7 @@ async def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -4262,16 +3784,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4291,10 +3812,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload async def begin_update( self, @@ -4316,14 +3833,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4336,7 +3845,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4348,18 +3857,10 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4372,7 +3873,7 @@ async def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -4382,19 +3883,8 @@ async def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2023_07_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2023_07_03.models.SharingUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either SharingUpdate or the result of cls(response) :rtype: @@ -4427,7 +3917,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4437,17 +3927,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return AsyncLROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -4481,7 +3969,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.SharedGallery] @@ -4504,17 +3991,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4525,14 +4011,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -4542,11 +4028,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4558,10 +4044,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace_async async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -4570,7 +4052,6 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -4589,21 +4070,20 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4615,13 +4095,9 @@ async def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -4662,7 +4138,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImage] @@ -4685,18 +4160,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4707,14 +4181,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -4724,11 +4198,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4740,10 +4214,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace_async async def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -4757,7 +4227,6 @@ async def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -4776,22 +4245,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4803,13 +4271,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -4854,7 +4318,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -4878,19 +4341,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4901,14 +4363,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -4918,11 +4380,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4934,10 +4396,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace_async async def get( self, @@ -4961,7 +4419,6 @@ async def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -4980,23 +4437,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5008,13 +4464,9 @@ async def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -5045,7 +4497,6 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5064,21 +4515,20 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5090,13 +4540,9 @@ async def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _ deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -5131,7 +4577,6 @@ async def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -5150,22 +4595,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5177,13 +4621,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5195,7 +4635,6 @@ def list( :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -5219,17 +4658,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5240,14 +4678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -5257,11 +4695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5273,10 +4711,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -5320,7 +4754,6 @@ async def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5339,23 +4772,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5367,13 +4799,9 @@ async def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5387,7 +4815,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -5411,18 +4838,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5433,14 +4859,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -5450,11 +4876,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5465,7 +4891,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/models/_models_py3.py index 803bc12d0d52..aac7e05b77c3 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/models/_models_py3.py @@ -400,7 +400,7 @@ def __init__( class CommunityGalleryImageList(_serialization.Model): """The List Community Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImage] @@ -527,7 +527,7 @@ def __init__( class CommunityGalleryImageVersionList(_serialization.Model): """The List Community Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of community gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImageVersion] @@ -629,7 +629,7 @@ def __init__( class CommunityGalleryMetadata(_serialization.Model): """The metadata of community gallery. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher_uri: The publisher URI of this community gallery. :vartype publisher_uri: str @@ -713,7 +713,7 @@ def __init__(self, *, disk_encryption_set_id: Optional[str] = None, **kwargs: An class DataDiskImageEncryption(DiskImageEncryption): """Contains encryption settings for a data disk image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_encryption_set_id: A relative URI containing the resource ID of the disk encryption set. @@ -838,7 +838,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -887,7 +887,7 @@ class Gallery(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -981,7 +981,7 @@ class GalleryApplication(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1091,7 +1091,7 @@ def __init__( class GalleryApplicationCustomAction(_serialization.Model): """A custom action that can be performed with a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1149,7 +1149,7 @@ class GalleryApplicationCustomActionParameter(_serialization.Model): """The definition of a parameter that can be passed to a custom action of a Gallery Application Version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the custom action. Must be unique within the Gallery Application Version. Required. @@ -1217,7 +1217,7 @@ def __init__( class GalleryApplicationList(_serialization.Model): """The List Gallery Applications operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Gallery Applications. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.GalleryApplication] @@ -1399,7 +1399,7 @@ class GalleryApplicationVersion(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1480,7 +1480,7 @@ def __init__( class GalleryApplicationVersionList(_serialization.Model): """The List Gallery Application version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery Application Versions. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion] @@ -1613,12 +1613,12 @@ def __init__( class GalleryApplicationVersionPublishingProfile( GalleryArtifactPublishingProfileBase -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """The publishing profile of a gallery image version. Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar target_regions: The target regions where the Image Version is going to be replicated to. This property is updatable. @@ -1796,18 +1796,6 @@ class GalleryApplicationVersionSafetyProfile(GalleryArtifactSafetyProfileBase): :vartype allow_deletion_of_replicated_locations: bool """ - _attribute_map = { - "allow_deletion_of_replicated_locations": {"key": "allowDeletionOfReplicatedLocations", "type": "bool"}, - } - - def __init__(self, *, allow_deletion_of_replicated_locations: Optional[bool] = None, **kwargs: Any) -> None: - """ - :keyword allow_deletion_of_replicated_locations: Indicates whether or not removing this Gallery - Image Version from replicated regions is allowed. - :paramtype allow_deletion_of_replicated_locations: bool - """ - super().__init__(allow_deletion_of_replicated_locations=allow_deletion_of_replicated_locations, **kwargs) - class GalleryApplicationVersionUpdate(UpdateResourceDefinition): """Specifies information about the gallery Application Version that you want to update. @@ -1886,7 +1874,7 @@ def __init__( class GalleryArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar managed_image: The managed artifact. Required. :vartype managed_image: ~azure.mgmt.compute.v2023_07_03.models.ManagedArtifact @@ -2020,7 +2008,7 @@ class GalleryDataDiskImage(GalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar size_in_gb: This property indicates the size of the VHD to be created. :vartype size_in_gb: int @@ -2173,7 +2161,7 @@ class GalleryImage(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2372,7 +2360,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class GalleryImageIdentifier(_serialization.Model): """This is the gallery image definition identifier. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The name of the gallery image definition publisher. Required. :vartype publisher: str @@ -2412,7 +2400,7 @@ def __init__(self, *, publisher: str, offer: str, sku: str, **kwargs: Any) -> No class GalleryImageList(_serialization.Model): """The List Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of Shared Image Gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2614,7 +2602,7 @@ class GalleryImageVersion(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2708,7 +2696,7 @@ def __init__( class GalleryImageVersionList(_serialization.Model): """The List Gallery Image version operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of gallery image versions. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion] @@ -2774,71 +2762,6 @@ class GalleryImageVersionPublishingProfile(GalleryArtifactPublishingProfileBase) list[~azure.mgmt.compute.v2023_07_03.models.GalleryTargetExtendedLocation] """ - _validation = { - "published_date": {"readonly": True}, - } - - _attribute_map = { - "target_regions": {"key": "targetRegions", "type": "[TargetRegion]"}, - "replica_count": {"key": "replicaCount", "type": "int"}, - "exclude_from_latest": {"key": "excludeFromLatest", "type": "bool"}, - "published_date": {"key": "publishedDate", "type": "iso-8601"}, - "end_of_life_date": {"key": "endOfLifeDate", "type": "iso-8601"}, - "storage_account_type": {"key": "storageAccountType", "type": "str"}, - "replication_mode": {"key": "replicationMode", "type": "str"}, - "target_extended_locations": {"key": "targetExtendedLocations", "type": "[GalleryTargetExtendedLocation]"}, - } - - def __init__( - self, - *, - target_regions: Optional[List["_models.TargetRegion"]] = None, - replica_count: Optional[int] = None, - exclude_from_latest: Optional[bool] = None, - end_of_life_date: Optional[datetime.datetime] = None, - storage_account_type: Optional[Union[str, "_models.StorageAccountType"]] = None, - replication_mode: Optional[Union[str, "_models.ReplicationMode"]] = None, - target_extended_locations: Optional[List["_models.GalleryTargetExtendedLocation"]] = None, - **kwargs: Any - ) -> None: - """ - :keyword target_regions: The target regions where the Image Version is going to be replicated - to. This property is updatable. - :paramtype target_regions: list[~azure.mgmt.compute.v2023_07_03.models.TargetRegion] - :keyword replica_count: The number of replicas of the Image Version to be created per region. - This property would take effect for a region when regionalReplicaCount is not specified. This - property is updatable. - :paramtype replica_count: int - :keyword exclude_from_latest: If set to true, Virtual Machines deployed from the latest version - of the Image Definition won't use this Image Version. - :paramtype exclude_from_latest: bool - :keyword end_of_life_date: The end of life date of the gallery image version. This property can - be used for decommissioning purposes. This property is updatable. - :paramtype end_of_life_date: ~datetime.datetime - :keyword storage_account_type: Specifies the storage account type to be used to store the - image. This property is not updatable. Known values are: "Standard_LRS", "Standard_ZRS", and - "Premium_LRS". - :paramtype storage_account_type: str or - ~azure.mgmt.compute.v2023_07_03.models.StorageAccountType - :keyword replication_mode: Optional parameter which specifies the mode to be used for - replication. This property is not updatable. Known values are: "Full" and "Shallow". - :paramtype replication_mode: str or ~azure.mgmt.compute.v2023_07_03.models.ReplicationMode - :keyword target_extended_locations: The target extended locations where the Image Version is - going to be replicated to. This property is updatable. - :paramtype target_extended_locations: - list[~azure.mgmt.compute.v2023_07_03.models.GalleryTargetExtendedLocation] - """ - super().__init__( - target_regions=target_regions, - replica_count=replica_count, - exclude_from_latest=exclude_from_latest, - end_of_life_date=end_of_life_date, - storage_account_type=storage_account_type, - replication_mode=replication_mode, - target_extended_locations=target_extended_locations, - **kwargs - ) - class GalleryImageVersionSafetyProfile(GalleryArtifactSafetyProfileBase): """This is the safety profile of the Gallery Image Version. @@ -3045,7 +2968,7 @@ def __init__( class GalleryList(_serialization.Model): """The List Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of galleries. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.Gallery] @@ -3090,32 +3013,6 @@ class GalleryOSDiskImage(GalleryDiskImage): :vartype source: ~azure.mgmt.compute.v2023_07_03.models.GalleryDiskImageSource """ - _validation = { - "size_in_gb": {"readonly": True}, - } - - _attribute_map = { - "size_in_gb": {"key": "sizeInGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - "source": {"key": "source", "type": "GalleryDiskImageSource"}, - } - - def __init__( - self, - *, - host_caching: Optional[Union[str, "_models.HostCaching"]] = None, - source: Optional["_models.GalleryDiskImageSource"] = None, - **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2023_07_03.models.HostCaching - :keyword source: The source for the disk image. - :paramtype source: ~azure.mgmt.compute.v2023_07_03.models.GalleryDiskImageSource - """ - super().__init__(host_caching=host_caching, source=source, **kwargs) - class GalleryTargetExtendedLocation(_serialization.Model): """GalleryTargetExtendedLocation. @@ -3385,7 +3282,7 @@ def __init__( class ManagedArtifact(_serialization.Model): """The managed artifact. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: The managed artifact id. Required. :vartype id: str @@ -3867,7 +3764,7 @@ class SharedGalleryDataDiskImage(SharedGalleryDiskImage): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_size_gb: This property indicates the size of the VHD to be created. :vartype disk_size_gb: int @@ -4056,7 +3953,7 @@ def __init__( class SharedGalleryImageList(_serialization.Model): """The List Shared Gallery Images operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImage] @@ -4172,7 +4069,7 @@ def __init__( class SharedGalleryImageVersionList(_serialization.Model): """The List Shared Gallery Image versions operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared gallery images versions. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImageVersion] @@ -4242,7 +4139,7 @@ def __init__( class SharedGalleryList(_serialization.Model): """The List Shared Galleries operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of shared galleries. Required. :vartype value: list[~azure.mgmt.compute.v2023_07_03.models.SharedGallery] @@ -4285,25 +4182,6 @@ class SharedGalleryOSDiskImage(SharedGalleryDiskImage): :vartype host_caching: str or ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryHostCaching """ - _validation = { - "disk_size_gb": {"readonly": True}, - } - - _attribute_map = { - "disk_size_gb": {"key": "diskSizeGB", "type": "int"}, - "host_caching": {"key": "hostCaching", "type": "str"}, - } - - def __init__( - self, *, host_caching: Optional[Union[str, "_models.SharedGalleryHostCaching"]] = None, **kwargs: Any - ) -> None: - """ - :keyword host_caching: The host caching of the disk. Valid values are 'None', 'ReadOnly', and - 'ReadWrite'. Known values are: "None", "ReadOnly", and "ReadWrite". - :paramtype host_caching: str or ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryHostCaching - """ - super().__init__(host_caching=host_caching, **kwargs) - class SharingProfile(_serialization.Model): """Profile for gallery sharing to subscription or tenant. @@ -4423,7 +4301,7 @@ def __init__(self, *, summary: Optional[List["_models.RegionalSharingStatus"]] = class SharingUpdate(_serialization.Model): """Specifies information about the gallery sharing profile update. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operation_type: This property allows you to specify the operation type of gallery sharing update. Possible values are: **Add,** **Remove,** **Reset.**. Required. Known values are: @@ -4562,7 +4440,7 @@ def __init__(self, **kwargs: Any) -> None: class TargetRegion(_serialization.Model): """Describes the target region information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The name of the region. Required. :vartype name: str @@ -4711,7 +4589,7 @@ def __init__( class UserArtifactManage(_serialization.Model): """UserArtifactManage. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar install: Required. The path and arguments to install the gallery application. This is limited to 4096 characters. Required. @@ -4795,7 +4673,7 @@ def __init__( class UserArtifactSource(_serialization.Model): """The source image from which the Image Version is going to be created. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar media_link: Required. The mediaLink of the artifact, must be a readable storage page blob. Required. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/operations/_operations.py index 732e44f30065..45f43b070b78 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_07_03/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -179,7 +179,7 @@ def build_galleries_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_galleries_list_by_resource_group_request( +def build_galleries_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -233,7 +233,7 @@ def build_galleries_list_request(subscription_id: str, **kwargs: Any) -> HttpReq return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_create_or_update_request( +def build_gallery_images_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -367,7 +367,7 @@ def build_gallery_images_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_images_list_by_gallery_request( +def build_gallery_images_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -398,7 +398,7 @@ def build_gallery_images_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_create_or_update_request( +def build_gallery_image_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -439,7 +439,7 @@ def build_gallery_image_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_update_request( +def build_gallery_image_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -522,7 +522,7 @@ def build_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_delete_request( +def build_gallery_image_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, @@ -560,7 +560,7 @@ def build_gallery_image_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_image_versions_list_by_gallery_image_request( +def build_gallery_image_versions_list_by_gallery_image_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -592,7 +592,7 @@ def build_gallery_image_versions_list_by_gallery_image_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_create_or_update_request( +def build_gallery_applications_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -627,7 +627,7 @@ def build_gallery_applications_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_update_request( +def build_gallery_applications_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -694,7 +694,7 @@ def build_gallery_applications_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_delete_request( +def build_gallery_applications_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -726,7 +726,7 @@ def build_gallery_applications_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_applications_list_by_gallery_request( +def build_gallery_applications_list_by_gallery_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -757,7 +757,7 @@ def build_gallery_applications_list_by_gallery_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_create_or_update_request( +def build_gallery_application_versions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -800,7 +800,7 @@ def build_gallery_application_versions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_update_request( +def build_gallery_application_versions_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -843,7 +843,7 @@ def build_gallery_application_versions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_get_request( +def build_gallery_application_versions_get_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -887,7 +887,7 @@ def build_gallery_application_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_delete_request( +def build_gallery_application_versions_delete_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, @@ -927,7 +927,7 @@ def build_gallery_application_versions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_application_versions_list_by_gallery_application_request( +def build_gallery_application_versions_list_by_gallery_application_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, gallery_application_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -959,7 +959,7 @@ def build_gallery_application_versions_list_by_gallery_application_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_gallery_sharing_profile_update_request( +def build_gallery_sharing_profile_update_request( # pylint: disable=name-too-long resource_group_name: str, gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1130,7 +1130,7 @@ def build_shared_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_list_request( +def build_shared_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1170,7 +1170,7 @@ def build_shared_gallery_image_versions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_shared_gallery_image_versions_get_request( +def build_shared_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, gallery_unique_name: str, gallery_image_name: str, @@ -1239,7 +1239,7 @@ def build_community_galleries_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_get_request( +def build_community_gallery_images_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1271,7 +1271,7 @@ def build_community_gallery_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_images_list_request( +def build_community_gallery_images_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1302,7 +1302,7 @@ def build_community_gallery_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_get_request( +def build_community_gallery_image_versions_get_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, @@ -1340,7 +1340,7 @@ def build_community_gallery_image_versions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_community_gallery_image_versions_list_request( +def build_community_gallery_image_versions_list_request( # pylint: disable=name-too-long location: str, public_gallery_name: str, gallery_image_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1393,7 +1393,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1418,7 +1418,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery, "Gallery") - request = build_galleries_create_or_update_request( + _request = build_galleries_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1426,16 +1426,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1458,10 +1457,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } - @overload def begin_create_or_update( self, @@ -1486,14 +1481,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1504,7 +1491,7 @@ def begin_create_or_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1519,18 +1506,10 @@ def begin_create_or_update( :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1538,7 +1517,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO], **kwargs: Any + self, resource_group_name: str, gallery_name: str, gallery: Union[_models.Gallery, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Gallery]: """Create or update a Shared Image Gallery. @@ -1549,19 +1528,8 @@ def begin_create_or_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the create or update Shared Image Gallery operation. Is - either a Gallery type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2023_07_03.models.Gallery or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a Gallery type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2023_07_03.models.Gallery or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1592,7 +1560,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1602,20 +1570,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Gallery: error_map = { 401: ClientAuthenticationError, @@ -1640,7 +1610,7 @@ def _update_initial( else: _json = self._serialize.body(gallery, "GalleryUpdate") - request = build_galleries_update_request( + _request = build_galleries_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -1648,16 +1618,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1669,13 +1638,9 @@ def _update_initial( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -1700,14 +1665,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,7 +1675,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - gallery: IO, + gallery: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1732,18 +1689,10 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Required. - :type gallery: IO + :type gallery: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1751,7 +1700,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, gallery_name: str, gallery: Union[_models.GalleryUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + gallery_name: str, + gallery: Union[_models.GalleryUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Gallery]: """Update a Shared Image Gallery. @@ -1762,19 +1715,8 @@ def begin_update( Required. :type gallery_name: str :param gallery: Parameters supplied to the update Shared Image Gallery operation. Is either a - GalleryUpdate type or a IO type. Required. - :type gallery: ~azure.mgmt.compute.v2023_07_03.models.GalleryUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryUpdate type or a IO[bytes] type. Required. + :type gallery: ~azure.mgmt.compute.v2023_07_03.models.GalleryUpdate or IO[bytes] :return: An instance of LROPoller that returns either Gallery or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -1805,7 +1747,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1815,17 +1757,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Gallery].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[_models.Gallery]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -1849,7 +1789,6 @@ def get( :keyword expand: The expand query option to apply on the operation. "SharingProfile/Groups" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.GalleryExpandParams - :keyword callable cls: A custom type or function that will be passed the direct response :return: Gallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.Gallery :raises ~azure.core.exceptions.HttpResponseError: @@ -1868,23 +1807,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.Gallery] = kwargs.pop("cls", None) - request = build_galleries_get_request( + _request = build_galleries_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, select=select, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1896,13 +1834,9 @@ def get( deserialized = self._deserialize("Gallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, **kwargs: Any @@ -1921,21 +1855,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_galleries_delete_request( + _request = build_galleries_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1945,11 +1878,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1959,14 +1888,6 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An :type resource_group_name: str :param gallery_name: The name of the Shared Image Gallery to be deleted. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1993,7 +1914,7 @@ def begin_delete(self, resource_group_name: str, gallery_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2002,17 +1923,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Gallery"]: @@ -2020,7 +1937,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2042,16 +1958,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_by_resource_group_request( + _request = build_galleries_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2062,14 +1977,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2079,11 +1994,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2095,15 +2010,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: """List galleries under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Gallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.Gallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -2125,15 +2035,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Gallery"]: def prepare_request(next_link=None): if not next_link: - request = build_galleries_list_request( + _request = build_galleries_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2144,14 +2053,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryList", pipeline_response) @@ -2161,11 +2070,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2177,8 +2086,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/galleries"} - class GalleryImagesOperations: """ @@ -2205,7 +2112,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2231,7 +2138,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImage") - request = build_gallery_images_create_or_update_request( + _request = build_gallery_images_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2240,16 +2147,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2272,10 +2178,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } - @overload def begin_create_or_update( self, @@ -2304,14 +2206,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2324,7 +2218,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2342,18 +2236,10 @@ def begin_create_or_update( :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2366,7 +2252,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImage, IO], + gallery_image: Union[_models.GalleryImage, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Create or update a gallery image definition. @@ -2381,19 +2267,8 @@ def begin_create_or_update( middle. The maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the create or update gallery image operation. Is - either a GalleryImage type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a GalleryImage type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2426,7 +2301,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2436,24 +2311,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImage: error_map = { @@ -2479,7 +2352,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image, "GalleryImageUpdate") - request = build_gallery_images_update_request( + _request = build_gallery_images_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -2488,16 +2361,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2509,13 +2381,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -2544,14 +2412,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2564,7 +2424,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: IO, + gallery_image: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2581,18 +2441,10 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Required. - :type gallery_image: IO + :type gallery_image: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2605,7 +2457,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_image_name: str, - gallery_image: Union[_models.GalleryImageUpdate, IO], + gallery_image: Union[_models.GalleryImageUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImage]: """Update a gallery image definition. @@ -2620,19 +2472,8 @@ def begin_update( maximum length is 80 characters. Required. :type gallery_image_name: str :param gallery_image: Parameters supplied to the update gallery image operation. Is either a - GalleryImageUpdate type or a IO type. Required. - :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + GalleryImageUpdate type or a IO[bytes] type. Required. + :type gallery_image: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryImage or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] @@ -2665,7 +2506,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2675,17 +2516,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImage].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[_models.GalleryImage]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -2700,7 +2539,6 @@ def get( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be retrieved. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -2719,22 +2557,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryImage] = kwargs.pop("cls", None) - request = build_gallery_images_get_request( + _request = build_gallery_images_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2746,13 +2583,9 @@ def get( deserialized = self._deserialize("GalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_image_name: str, **kwargs: Any @@ -2771,22 +2604,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_images_delete_request( + _request = build_gallery_images_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2796,11 +2628,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -2815,14 +2643,6 @@ def begin_delete( :type gallery_name: str :param gallery_image_name: The name of the gallery image definition to be deleted. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2850,7 +2670,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -2859,17 +2679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -2882,7 +2698,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Image Gallery from which Image Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -2904,17 +2719,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_images_list_by_gallery_request( + _request = build_gallery_images_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2925,14 +2739,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageList", pipeline_response) @@ -2942,11 +2756,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,10 +2772,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images" - } - class GalleryImageVersionsOperations: """ @@ -2989,7 +2799,7 @@ def _create_or_update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3015,7 +2825,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersion") - request = build_gallery_image_versions_create_or_update_request( + _request = build_gallery_image_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3025,16 +2835,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3057,10 +2866,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } - @overload def begin_create_or_update( self, @@ -3094,14 +2899,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3116,7 +2913,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3138,18 +2935,10 @@ def begin_create_or_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3164,7 +2953,7 @@ def begin_create_or_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersion, IO], + gallery_image_version: Union[_models.GalleryImageVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Create or update a gallery image version. @@ -3183,19 +2972,9 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the create or update gallery image version - operation. Is either a GalleryImageVersion type or a IO type. Required. - :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryImageVersion type or a IO[bytes] type. Required. + :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion or + IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3230,7 +3009,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3240,17 +3019,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -3258,7 +3035,7 @@ def _update_initial( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryImageVersion: error_map = { @@ -3284,7 +3061,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_image_version, "GalleryImageVersionUpdate") - request = build_gallery_image_versions_update_request( + _request = build_gallery_image_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3294,16 +3071,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3315,13 +3091,9 @@ def _update_initial( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -3356,14 +3128,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3378,7 +3142,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: IO, + gallery_image_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3164,10 @@ def begin_update( :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version operation. Required. - :type gallery_image_version: IO + :type gallery_image_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3426,7 +3182,7 @@ def begin_update( gallery_name: str, gallery_image_name: str, gallery_image_version_name: str, - gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO], + gallery_image_version: Union[_models.GalleryImageVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryImageVersion]: """Update a gallery image version. @@ -3445,20 +3201,9 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str :param gallery_image_version: Parameters supplied to the update gallery image version - operation. Is either a GalleryImageVersionUpdate type or a IO type. Required. + operation. Is either a GalleryImageVersionUpdate type or a IO[bytes] type. Required. :type gallery_image_version: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersionUpdate - or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or IO[bytes] :return: An instance of LROPoller that returns either GalleryImageVersion or the result of cls(response) :rtype: @@ -3493,7 +3238,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3503,17 +3248,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryImageVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[_models.GalleryImageVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -3542,7 +3285,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -3561,7 +3303,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryImageVersion] = kwargs.pop("cls", None) - request = build_gallery_image_versions_get_request( + _request = build_gallery_image_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, @@ -3569,16 +3311,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3590,13 +3331,9 @@ def get( deserialized = self._deserialize("GalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -3620,23 +3357,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_image_versions_delete_request( + _request = build_gallery_image_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3646,11 +3382,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -3674,14 +3406,6 @@ def begin_delete( :param gallery_image_version_name: The name of the gallery image version to be deleted. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3710,7 +3434,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3719,17 +3443,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_image( @@ -3745,7 +3465,6 @@ def list_by_gallery_image( :param gallery_image_name: The name of the Shared Image Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryImageVersion or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryImageVersion] @@ -3768,18 +3487,17 @@ def list_by_gallery_image( def prepare_request(next_link=None): if not next_link: - request = build_gallery_image_versions_list_by_gallery_image_request( + _request = build_gallery_image_versions_list_by_gallery_image_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_image.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3790,14 +3508,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryImageVersionList", pipeline_response) @@ -3807,11 +3525,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3823,10 +3541,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_image.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{galleryImageName}/versions" - } - class GalleryApplicationsOperations: """ @@ -3853,7 +3567,7 @@ def _create_or_update_initial( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -3879,7 +3593,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplication") - request = build_gallery_applications_create_or_update_request( + _request = build_gallery_applications_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -3888,16 +3602,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3920,10 +3633,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } - @overload def begin_create_or_update( self, @@ -3952,14 +3661,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -3973,7 +3674,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3991,18 +3692,10 @@ def begin_create_or_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4016,7 +3709,7 @@ def begin_create_or_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplication, IO], + gallery_application: Union[_models.GalleryApplication, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Create or update a gallery Application Definition. @@ -4031,19 +3724,9 @@ def begin_create_or_update( allowed in the middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the create or update gallery Application - operation. Is either a GalleryApplication type or a IO type. Required. - :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GalleryApplication type or a IO[bytes] type. Required. + :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication or + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4077,7 +3760,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4087,24 +3770,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplication: error_map = { @@ -4130,7 +3811,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application, "GalleryApplicationUpdate") - request = build_gallery_applications_update_request( + _request = build_gallery_applications_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4139,16 +3820,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4160,13 +3840,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -4196,14 +3872,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4217,7 +3885,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: IO, + gallery_application: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4235,18 +3903,10 @@ def begin_update( :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Required. - :type gallery_application: IO + :type gallery_application: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4260,7 +3920,7 @@ def begin_update( resource_group_name: str, gallery_name: str, gallery_application_name: str, - gallery_application: Union[_models.GalleryApplicationUpdate, IO], + gallery_application: Union[_models.GalleryApplicationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplication]: """Update a gallery Application Definition. @@ -4275,20 +3935,9 @@ def begin_update( middle. The maximum length is 80 characters. Required. :type gallery_application_name: str :param gallery_application: Parameters supplied to the update gallery Application operation. Is - either a GalleryApplicationUpdate type or a IO type. Required. + either a GalleryApplicationUpdate type or a IO[bytes] type. Required. :type gallery_application: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationUpdate or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either GalleryApplication or the result of cls(response) :rtype: @@ -4322,7 +3971,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4332,17 +3981,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplication].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[_models.GalleryApplication]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -4358,7 +4005,6 @@ def get( :param gallery_application_name: The name of the gallery Application Definition to be retrieved. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplication or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplication :raises ~azure.core.exceptions.HttpResponseError: @@ -4377,22 +4023,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryApplication] = kwargs.pop("cls", None) - request = build_gallery_applications_get_request( + _request = build_gallery_applications_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4404,13 +4049,9 @@ def get( deserialized = self._deserialize("GalleryApplication", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, gallery_name: str, gallery_application_name: str, **kwargs: Any @@ -4429,22 +4070,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_applications_delete_request( + _request = build_gallery_applications_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4454,11 +4094,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -4474,14 +4110,6 @@ def begin_delete( :param gallery_application_name: The name of the gallery Application Definition to be deleted. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4509,7 +4137,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4518,17 +4146,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery( @@ -4541,7 +4165,6 @@ def list_by_gallery( :param gallery_name: The name of the Shared Application Gallery from which Application Definitions are to be listed. Required. :type gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplication or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.GalleryApplication] :raises ~azure.core.exceptions.HttpResponseError: @@ -4563,17 +4186,16 @@ def list_by_gallery( def prepare_request(next_link=None): if not next_link: - request = build_gallery_applications_list_by_gallery_request( + _request = build_gallery_applications_list_by_gallery_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4584,14 +4206,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationList", pipeline_response) @@ -4601,11 +4223,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4617,10 +4239,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications" - } - class GalleryApplicationVersionsOperations: """ @@ -4648,7 +4266,7 @@ def _create_or_update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4674,7 +4292,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersion") - request = build_gallery_application_versions_create_or_update_request( + _request = build_gallery_application_versions_create_or_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4684,16 +4302,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4716,10 +4333,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } - @overload def begin_create_or_update( self, @@ -4754,14 +4367,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4776,7 +4381,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4798,18 +4403,10 @@ def begin_create_or_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4824,7 +4421,7 @@ def begin_create_or_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersion, IO], + gallery_application_version: Union[_models.GalleryApplicationVersion, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Create or update a gallery Application Version. @@ -4843,21 +4440,10 @@ def begin_create_or_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the create or update gallery - Application Version operation. Is either a GalleryApplicationVersion type or a IO type. + Application Version operation. Is either a GalleryApplicationVersion type or a IO[bytes] type. Required. :type gallery_application_version: - ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -4892,7 +4478,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4902,17 +4488,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -4920,7 +4504,7 @@ def _update_initial( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> _models.GalleryApplicationVersion: error_map = { @@ -4946,7 +4530,7 @@ def _update_initial( else: _json = self._serialize.body(gallery_application_version, "GalleryApplicationVersionUpdate") - request = build_gallery_application_versions_update_request( + _request = build_gallery_application_versions_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -4956,16 +4540,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4977,13 +4560,9 @@ def _update_initial( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -5019,14 +4598,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5041,7 +4612,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: IO, + gallery_application_version: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5063,18 +4634,10 @@ def begin_update( :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application Version operation. Required. - :type gallery_application_version: IO + :type gallery_application_version: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5089,7 +4652,7 @@ def begin_update( gallery_name: str, gallery_application_name: str, gallery_application_version_name: str, - gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO], + gallery_application_version: Union[_models.GalleryApplicationVersionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.GalleryApplicationVersion]: """Update a gallery Application Version. @@ -5108,20 +4671,10 @@ def begin_update( :code:``.:code:``.:code:``. Required. :type gallery_application_version_name: str :param gallery_application_version: Parameters supplied to the update gallery Application - Version operation. Is either a GalleryApplicationVersionUpdate type or a IO type. Required. + Version operation. Is either a GalleryApplicationVersionUpdate type or a IO[bytes] type. + Required. :type gallery_application_version: - ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersionUpdate or IO[bytes] :return: An instance of LROPoller that returns either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5156,7 +4709,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5166,17 +4719,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.GalleryApplicationVersion].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[_models.GalleryApplicationVersion]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get( @@ -5205,7 +4756,6 @@ def get( :keyword expand: The expand expression to apply on the operation. Known values are: "ReplicationStatus" and "UefiSettings". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_07_03.models.ReplicationStatusTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: GalleryApplicationVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.GalleryApplicationVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -5224,7 +4774,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.GalleryApplicationVersion] = kwargs.pop("cls", None) - request = build_gallery_application_versions_get_request( + _request = build_gallery_application_versions_get_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, @@ -5232,16 +4782,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5253,13 +4802,9 @@ def get( deserialized = self._deserialize("GalleryApplicationVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -5283,23 +4828,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_gallery_application_versions_delete_request( + _request = build_gallery_application_versions_delete_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, gallery_application_version_name=gallery_application_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5309,11 +4853,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -5337,14 +4877,6 @@ def begin_delete( :param gallery_application_version_name: The name of the gallery Application Version to be deleted. Required. :type gallery_application_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5373,7 +4905,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5382,17 +4914,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions/{galleryApplicationVersionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_gallery_application( @@ -5408,7 +4936,6 @@ def list_by_gallery_application( :param gallery_application_name: The name of the Shared Application Gallery Application Definition from which the Application Versions are to be listed. Required. :type gallery_application_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either GalleryApplicationVersion or the result of cls(response) :rtype: @@ -5432,18 +4959,17 @@ def list_by_gallery_application( def prepare_request(next_link=None): if not next_link: - request = build_gallery_application_versions_list_by_gallery_application_request( + _request = build_gallery_application_versions_list_by_gallery_application_request( resource_group_name=resource_group_name, gallery_name=gallery_name, gallery_application_name=gallery_application_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_gallery_application.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5454,14 +4980,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("GalleryApplicationVersionList", pipeline_response) @@ -5471,11 +4997,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5487,10 +5013,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_gallery_application.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{galleryApplicationName}/versions" - } - class GallerySharingProfileOperations: """ @@ -5516,7 +5038,7 @@ def _update_initial( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> _models.SharingUpdate: error_map = { @@ -5542,7 +5064,7 @@ def _update_initial( else: _json = self._serialize.body(sharing_update, "SharingUpdate") - request = build_gallery_sharing_profile_update_request( + _request = build_gallery_sharing_profile_update_request( resource_group_name=resource_group_name, gallery_name=gallery_name, subscription_id=self._config.subscription_id, @@ -5550,16 +5072,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5579,10 +5100,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } - @overload def begin_update( self, @@ -5604,14 +5121,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.SharingUpdate] @@ -5623,7 +5132,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: IO, + sharing_update: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5635,18 +5144,10 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Required. - :type sharing_update: IO + :type sharing_update: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.SharingUpdate] @@ -5658,7 +5159,7 @@ def begin_update( self, resource_group_name: str, gallery_name: str, - sharing_update: Union[_models.SharingUpdate, IO], + sharing_update: Union[_models.SharingUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.SharingUpdate]: """Update sharing profile of a gallery. @@ -5668,19 +5169,8 @@ def begin_update( :param gallery_name: The name of the Shared Image Gallery. Required. :type gallery_name: str :param sharing_update: Parameters supplied to the update gallery sharing profile. Is either a - SharingUpdate type or a IO type. Required. - :type sharing_update: ~azure.mgmt.compute.v2023_07_03.models.SharingUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + SharingUpdate type or a IO[bytes] type. Required. + :type sharing_update: ~azure.mgmt.compute.v2023_07_03.models.SharingUpdate or IO[bytes] :return: An instance of LROPoller that returns either SharingUpdate or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_07_03.models.SharingUpdate] @@ -5712,7 +5202,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("SharingUpdate", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5722,17 +5212,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.SharingUpdate].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/share" - } + return LROPoller[_models.SharingUpdate]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class SharedGalleriesOperations: @@ -5766,7 +5254,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGallery or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.SharedGallery] :raises ~azure.core.exceptions.HttpResponseError: @@ -5788,17 +5275,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_galleries_list_request( + _request = build_shared_galleries_list_request( location=location, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5809,14 +5295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryList", pipeline_response) @@ -5826,11 +5312,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5842,10 +5328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries" - } - @distributed_trace def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models.SharedGallery: """Get a shared gallery by subscription id or tenant id. @@ -5854,7 +5336,6 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models :type location: str :param gallery_unique_name: The unique name of the Shared Gallery. Required. :type gallery_unique_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -5873,21 +5354,20 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGallery] = kwargs.pop("cls", None) - request = build_shared_galleries_get_request( + _request = build_shared_galleries_get_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5899,13 +5379,9 @@ def get(self, location: str, gallery_unique_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("SharedGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}" - } + return deserialized # type: ignore class SharedGalleryImagesOperations: @@ -5946,7 +5422,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -5968,18 +5443,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_images_list_request( + _request = build_shared_gallery_images_list_request( location=location, gallery_unique_name=gallery_unique_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5990,14 +5464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageList", pipeline_response) @@ -6007,11 +5481,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6023,10 +5497,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images" - } - @distributed_trace def get( self, location: str, gallery_unique_name: str, gallery_image_name: str, **kwargs: Any @@ -6040,7 +5510,6 @@ def get( :param gallery_image_name: The name of the Shared Gallery Image Definition from which the Image Versions are to be listed. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6059,22 +5528,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGalleryImage] = kwargs.pop("cls", None) - request = build_shared_gallery_images_get_request( + _request = build_shared_gallery_images_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6086,13 +5554,9 @@ def get( deserialized = self._deserialize("SharedGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}" - } + return deserialized # type: ignore class SharedGalleryImageVersionsOperations: @@ -6137,7 +5601,6 @@ def list( :keyword shared_to: The query parameter to decide what shared galleries to fetch when doing listing operations. "tenant" Default value is None. :paramtype shared_to: str or ~azure.mgmt.compute.v2023_07_03.models.SharedToValues - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SharedGalleryImageVersion or the result of cls(response) :rtype: @@ -6161,19 +5624,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_shared_gallery_image_versions_list_request( + _request = build_shared_gallery_image_versions_list_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, shared_to=shared_to, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6184,14 +5646,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SharedGalleryImageVersionList", pipeline_response) @@ -6201,11 +5663,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6217,10 +5679,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions" - } - @distributed_trace def get( self, @@ -6244,7 +5702,6 @@ def get( must be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SharedGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.SharedGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6263,23 +5720,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.SharedGalleryImageVersion] = kwargs.pop("cls", None) - request = build_shared_gallery_image_versions_get_request( + _request = build_shared_gallery_image_versions_get_request( location=location, gallery_unique_name=gallery_unique_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6291,13 +5747,9 @@ def get( deserialized = self._deserialize("SharedGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/sharedGalleries/{galleryUniqueName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore class CommunityGalleriesOperations: @@ -6328,7 +5780,6 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGallery or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGallery :raises ~azure.core.exceptions.HttpResponseError: @@ -6347,21 +5798,20 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGallery] = kwargs.pop("cls", None) - request = build_community_galleries_get_request( + _request = build_community_galleries_get_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6373,13 +5823,9 @@ def get(self, location: str, public_gallery_name: str, **kwargs: Any) -> _models deserialized = self._deserialize("CommunityGallery", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}" - } + return deserialized # type: ignore class CommunityGalleryImagesOperations: @@ -6414,7 +5860,6 @@ def get( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImage :raises ~azure.core.exceptions.HttpResponseError: @@ -6433,22 +5878,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGalleryImage] = kwargs.pop("cls", None) - request = build_community_gallery_images_get_request( + _request = build_community_gallery_images_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6460,13 +5904,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}" - } + return deserialized # type: ignore @distributed_trace def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterable["_models.CommunityGalleryImage"]: @@ -6476,7 +5916,6 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab :type location: str :param public_gallery_name: The public name of the community gallery. Required. :type public_gallery_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImage or the result of cls(response) :rtype: @@ -6500,17 +5939,16 @@ def list(self, location: str, public_gallery_name: str, **kwargs: Any) -> Iterab def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_images_list_request( + _request = build_community_gallery_images_list_request( location=location, public_gallery_name=public_gallery_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6521,14 +5959,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageList", pipeline_response) @@ -6538,11 +5976,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6554,10 +5992,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images" - } - class CommunityGalleryImageVersionsOperations: """ @@ -6601,7 +6035,6 @@ def get( be within the range of a 32-bit integer. Format: :code:``.:code:``.:code:``. Required. :type gallery_image_version_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CommunityGalleryImageVersion or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_07_03.models.CommunityGalleryImageVersion :raises ~azure.core.exceptions.HttpResponseError: @@ -6620,23 +6053,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-07-03")) cls: ClsType[_models.CommunityGalleryImageVersion] = kwargs.pop("cls", None) - request = build_community_gallery_image_versions_get_request( + _request = build_community_gallery_image_versions_get_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, gallery_image_version_name=gallery_image_version_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6648,13 +6080,9 @@ def get( deserialized = self._deserialize("CommunityGalleryImageVersion", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions/{galleryImageVersionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -6668,7 +6096,6 @@ def list( :type public_gallery_name: str :param gallery_image_name: The name of the community gallery image definition. Required. :type gallery_image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CommunityGalleryImageVersion or the result of cls(response) :rtype: @@ -6692,18 +6119,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_community_gallery_image_versions_list_request( + _request = build_community_gallery_image_versions_list_request( location=location, public_gallery_name=public_gallery_name, gallery_image_name=gallery_image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6714,14 +6140,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CommunityGalleryImageVersionList", pipeline_response) @@ -6731,11 +6157,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6746,7 +6172,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/communityGalleries/{publicGalleryName}/images/{galleryImageName}/versions" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_compute_management_client.py index 7ec06e48d9e4..f62c1658c5be 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -150,7 +152,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -229,7 +249,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-09-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -249,7 +269,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_configuration.py index 25ef2a661bdc..09a8bcc5b886 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-09-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_metadata.json index 19630c86e20f..f6ab6d1246fb 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "operations": "Operations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_compute_management_client.py index 10fc35b24662..ba13843ae652 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -152,7 +154,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -231,7 +251,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-09-01" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -251,7 +273,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_configuration.py index e8b565ac2ace..95ade37ac946 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-09-01") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/operations/_operations.py index 98c892634756..69f4417a55db 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -241,7 +241,6 @@ def __init__(self, *args, **kwargs) -> None: def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -265,14 +264,13 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -283,14 +281,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -300,11 +298,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -316,8 +314,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -346,7 +342,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -368,16 +363,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -388,14 +382,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -405,11 +399,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -421,8 +415,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -451,7 +443,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] @@ -474,16 +465,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMa def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -494,14 +484,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -511,11 +501,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -527,8 +517,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -557,7 +545,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -581,16 +568,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -601,14 +587,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -618,11 +604,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,15 +620,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - async def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -671,7 +653,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -681,16 +663,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -710,10 +691,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload async def begin_create_or_update( self, @@ -745,14 +722,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -765,7 +734,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -779,7 +748,7 @@ async def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -791,14 +760,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -811,7 +772,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -823,9 +784,9 @@ async def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet or IO + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -834,17 +795,6 @@ async def begin_create_or_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -879,7 +829,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -889,23 +839,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -934,7 +882,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -944,16 +892,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -965,13 +912,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -1004,14 +947,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1024,7 +959,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -1038,7 +973,7 @@ async def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -1050,14 +985,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1070,7 +997,7 @@ async def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -1082,9 +1009,10 @@ async def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetUpdate or IO + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -1093,17 +1021,6 @@ async def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1138,7 +1055,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1148,17 +1065,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1177,22 +1092,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1202,11 +1116,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1221,14 +1131,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1256,7 +1158,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1265,17 +1167,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -1296,7 +1194,6 @@ async def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -1315,22 +1212,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1342,19 +1238,15 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1385,7 +1277,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1394,16 +1286,15 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1413,11 +1304,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_deallocate( @@ -1449,14 +1336,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1467,7 +1346,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -1483,7 +1362,7 @@ async def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -1491,14 +1370,6 @@ async def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1509,7 +1380,7 @@ async def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -1523,24 +1394,13 @@ async def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1571,7 +1431,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1580,23 +1440,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1624,7 +1480,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -1633,16 +1489,15 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1652,11 +1507,7 @@ async def _delete_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_delete_instances( @@ -1685,14 +1536,6 @@ async def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1703,7 +1546,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -1717,21 +1560,13 @@ async def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1742,7 +1577,7 @@ async def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -1754,23 +1589,12 @@ async def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1801,7 +1625,7 @@ async def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1810,17 +1634,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_instance_view( @@ -1832,7 +1652,6 @@ async def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -1851,21 +1670,20 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1877,13 +1695,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: @@ -1891,7 +1705,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -1915,16 +1728,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1935,14 +1747,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -1952,11 +1764,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1968,17 +1780,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -2002,15 +1809,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.VirtualMachineScaleS def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2021,14 +1827,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -2038,11 +1844,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2054,8 +1860,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2067,7 +1871,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -2091,17 +1894,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2112,14 +1914,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -2129,11 +1931,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2145,10 +1947,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2159,7 +1957,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -2183,17 +1980,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2204,14 +2000,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -2221,11 +2017,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2237,15 +2033,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2276,7 +2068,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2285,16 +2077,15 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2304,11 +2095,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_power_off( @@ -2340,14 +2127,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2358,7 +2137,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -2374,7 +2153,7 @@ async def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -2382,14 +2161,6 @@ async def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2400,7 +2171,7 @@ async def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -2414,24 +2185,13 @@ async def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2462,7 +2222,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2471,23 +2231,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2516,7 +2272,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2524,16 +2280,15 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2543,11 +2298,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_restart( @@ -2572,14 +2323,6 @@ async def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2590,7 +2333,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2603,18 +2346,10 @@ async def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2625,7 +2360,7 @@ async def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -2635,20 +2370,9 @@ async def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2678,7 +2402,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2687,23 +2411,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -2732,7 +2452,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -2740,16 +2460,15 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2759,11 +2478,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_start( @@ -2788,14 +2503,6 @@ async def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2806,7 +2513,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -2819,18 +2526,10 @@ async def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2841,7 +2540,7 @@ async def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -2851,20 +2550,9 @@ async def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -2894,7 +2582,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2903,17 +2591,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -2932,21 +2616,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2960,11 +2643,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async async def begin_reapply( @@ -2977,14 +2656,6 @@ async def begin_reapply( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3011,7 +2682,7 @@ async def begin_reapply( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -3022,23 +2693,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3067,7 +2734,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3075,16 +2742,15 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3094,11 +2760,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_redeploy( @@ -3124,14 +2786,6 @@ async def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3142,7 +2796,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3156,18 +2810,10 @@ async def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3178,7 +2824,7 @@ async def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -3189,20 +2835,9 @@ async def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3232,7 +2867,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3241,23 +2876,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3286,7 +2917,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3294,16 +2925,15 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3313,11 +2943,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_perform_maintenance( @@ -3345,14 +2971,6 @@ async def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3363,7 +2981,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3379,18 +2997,10 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3401,7 +3011,7 @@ async def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -3414,20 +3024,9 @@ async def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3457,7 +3056,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3466,23 +3065,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -3508,7 +3103,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3516,16 +3111,15 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3535,11 +3129,7 @@ async def _update_instances_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_update_instances( @@ -3564,14 +3154,6 @@ async def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3582,7 +3164,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3595,18 +3177,10 @@ async def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3617,7 +3191,7 @@ async def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -3627,20 +3201,9 @@ async def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3670,7 +3233,7 @@ async def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3679,23 +3242,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3724,7 +3283,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3732,16 +3291,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3751,11 +3309,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -3781,14 +3335,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3799,7 +3345,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -3813,18 +3359,10 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3835,7 +3373,7 @@ async def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -3847,20 +3385,9 @@ async def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -3890,7 +3417,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3899,23 +3426,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -3944,7 +3467,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -3952,16 +3475,15 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3971,11 +3493,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage_all( @@ -4001,14 +3519,6 @@ async def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4019,7 +3529,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4033,18 +3543,10 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4055,7 +3557,7 @@ async def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -4066,20 +3568,9 @@ async def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4109,7 +3600,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4118,23 +3609,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -4163,7 +3650,7 @@ async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-retu else: _json = None - request = build_virtual_machine_scale_sets_approve_rolling_upgrade_request( + _request = build_virtual_machine_scale_sets_approve_rolling_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4171,16 +3658,15 @@ async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-retu content_type=content_type, json=_json, content=_content, - template_url=self._approve_rolling_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4193,11 +3679,7 @@ async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-retu response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _approve_rolling_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/approveRollingUpgrade" - } + return cls(pipeline_response, None, response_headers) # type: ignore @overload async def begin_approve_rolling_upgrade( @@ -4223,14 +3705,6 @@ async def begin_approve_rolling_upgrade( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4241,7 +3715,7 @@ async def begin_approve_rolling_upgrade( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -4255,18 +3729,10 @@ async def begin_approve_rolling_upgrade( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4277,7 +3743,7 @@ async def begin_approve_rolling_upgrade( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale @@ -4288,20 +3754,9 @@ async def begin_approve_rolling_upgrade( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4331,7 +3786,7 @@ async def begin_approve_rolling_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4340,20 +3795,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_approve_rolling_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/approveRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async - async def force_recovery_service_fabric_platform_update_domain_walk( + async def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -4379,7 +3830,6 @@ async def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -4398,7 +3848,7 @@ async def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4406,16 +3856,15 @@ async def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4427,13 +3876,9 @@ async def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload async def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -4458,7 +3903,6 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4469,7 +3913,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4482,11 +3926,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4497,7 +3940,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -4508,13 +3951,10 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_09_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_09_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -4542,7 +3982,7 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4550,16 +3990,15 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4569,17 +4008,13 @@ async def convert_to_single_placement_group( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore async def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -4605,7 +4040,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -4613,16 +4048,15 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4632,11 +4066,7 @@ async def _set_orchestration_service_state_initial( # pylint: disable=inconsist raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_set_orchestration_service_state( @@ -4660,14 +4090,6 @@ async def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4678,7 +4100,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4691,18 +4113,10 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4713,7 +4127,7 @@ async def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[None]: """Changes ServiceState property for a given service. @@ -4724,19 +4138,9 @@ async def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -4766,7 +4170,7 @@ async def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4775,20 +4179,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } - -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -4813,7 +4213,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -4839,7 +4239,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -4848,16 +4248,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4877,10 +4276,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -4908,14 +4303,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4929,7 +4316,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4945,18 +4332,10 @@ async def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -4970,7 +4349,7 @@ async def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -4983,20 +4362,9 @@ async def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5030,7 +4398,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5040,24 +4408,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -5083,7 +4449,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -5092,16 +4458,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5121,10 +4486,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload async def begin_update( self, @@ -5152,14 +4513,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5173,7 +4526,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5189,18 +4542,10 @@ async def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5214,7 +4559,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -5227,20 +4572,10 @@ async def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5274,7 +4609,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5284,17 +4619,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -5313,22 +4646,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5338,11 +4670,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -5357,14 +4685,6 @@ async def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5392,7 +4712,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5401,17 +4721,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -5433,7 +4749,6 @@ async def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -5452,23 +4767,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5480,13 +4794,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -5498,7 +4808,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -5522,17 +4831,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5543,14 +4851,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -5560,11 +4868,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5576,12 +4884,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -5618,21 +4922,20 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5642,11 +4945,7 @@ async def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_cancel( @@ -5658,14 +4957,6 @@ async def begin_cancel( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5692,7 +4983,7 @@ async def begin_cancel( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5701,17 +4992,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5730,21 +5017,20 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5754,11 +5040,7 @@ async def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-stat raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_os_upgrade( @@ -5772,14 +5054,6 @@ async def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5806,7 +5080,7 @@ async def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5815,17 +5089,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -5844,21 +5114,20 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5868,11 +5137,7 @@ async def _start_extension_upgrade_initial( # pylint: disable=inconsistent-retu raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start_extension_upgrade( @@ -5886,14 +5151,6 @@ async def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5920,7 +5177,7 @@ async def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -5929,17 +5186,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_latest( @@ -5951,7 +5204,6 @@ async def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -5970,21 +5222,20 @@ async def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5996,16 +5247,12 @@ async def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -6031,7 +5278,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -6057,7 +5304,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6067,16 +5314,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6096,10 +5342,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -6129,14 +5371,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6151,7 +5385,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6168,18 +5402,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6194,7 +5420,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -6208,20 +5434,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6256,7 +5471,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6266,17 +5481,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -6284,7 +5497,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -6310,7 +5523,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6320,16 +5533,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6341,13 +5553,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -6378,14 +5586,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6400,7 +5600,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6417,18 +5617,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6443,7 +5635,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -6457,20 +5649,10 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -6505,7 +5687,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6515,17 +5697,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -6544,23 +5724,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6570,11 +5749,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -6590,14 +5765,6 @@ async def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6626,7 +5793,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -6635,17 +5802,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -6670,7 +5833,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -6689,7 +5851,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6697,16 +5859,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6718,13 +5879,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -6746,7 +5903,6 @@ async def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -6765,23 +5921,22 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6793,13 +5948,9 @@ async def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -6827,7 +5978,9 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -6856,7 +6009,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -6865,16 +6018,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6884,11 +6036,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -6916,14 +6064,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6935,7 +6075,7 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -6950,18 +6090,10 @@ async def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6973,7 +6105,9 @@ async def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -6985,20 +6119,10 @@ async def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -7029,7 +6153,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7038,17 +6162,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7067,22 +6187,21 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7092,11 +6211,7 @@ async def _reimage_all_initial( # pylint: disable=inconsistent-return-statement raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reimage_all( @@ -7111,14 +6226,6 @@ async def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7146,7 +6253,7 @@ async def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7155,17 +6262,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7184,22 +6287,21 @@ async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-retu api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( + _request = build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._approve_rolling_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7212,11 +6314,7 @@ async def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-retu response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _approve_rolling_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/approveRollingUpgrade" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async async def begin_approve_rolling_upgrade( @@ -7230,14 +6328,6 @@ async def begin_approve_rolling_upgrade( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7265,7 +6355,7 @@ async def begin_approve_rolling_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7274,17 +6364,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_approve_rolling_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/approveRollingUpgrade" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -7303,22 +6389,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7328,11 +6413,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -7348,14 +6429,6 @@ async def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7383,7 +6456,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7392,24 +6465,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -7438,7 +6507,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -7449,16 +6518,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7478,10 +6546,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload async def begin_update( self, @@ -7518,14 +6582,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7539,7 +6595,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -7557,7 +6613,7 @@ async def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -7569,14 +6625,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7590,7 +6638,7 @@ async def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -7606,8 +6654,8 @@ async def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM or IO + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -7616,17 +6664,6 @@ async def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -7662,7 +6699,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7672,17 +6709,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -7707,23 +6742,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7733,11 +6767,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -7760,14 +6790,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7796,7 +6818,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -7805,17 +6827,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -7839,7 +6857,6 @@ async def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -7858,23 +6875,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7886,13 +6902,9 @@ async def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace_async async def get_instance_view( @@ -7906,7 +6918,6 @@ async def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7925,22 +6936,21 @@ async def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7952,13 +6962,9 @@ async def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -7987,7 +6993,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -8011,7 +7016,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -8019,12 +7024,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8035,14 +7039,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -8052,11 +7056,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8068,10 +7072,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -8095,23 +7095,22 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8121,11 +7120,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -8151,14 +7146,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8187,7 +7174,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8196,17 +7183,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8225,22 +7208,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8250,11 +7232,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -8268,14 +7246,6 @@ async def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8303,7 +7273,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8312,17 +7282,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8341,22 +7307,21 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8366,11 +7331,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start( @@ -8384,14 +7345,6 @@ async def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8419,7 +7372,7 @@ async def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8428,17 +7381,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8457,22 +7406,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8482,11 +7430,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -8501,14 +7445,6 @@ async def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8536,7 +7472,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8545,17 +7481,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -8580,7 +7512,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -8599,23 +7530,22 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8627,13 +7557,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -8652,22 +7578,21 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8677,11 +7602,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -8695,14 +7616,6 @@ async def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8730,7 +7643,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -8739,17 +7652,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -8763,7 +7672,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -8782,22 +7690,21 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8807,18 +7714,14 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _attach_detach_data_disks_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> Optional[_models.StorageProfile]: error_map = { @@ -8844,7 +7747,7 @@ async def _attach_detach_data_disks_initial( else: _json = self._serialize.body(parameters, "AttachDetachDataDisksRequest") - request = build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( + _request = build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -8853,16 +7756,15 @@ async def _attach_detach_data_disks_initial( content_type=content_type, json=_json, content=_content, - template_url=self._attach_detach_data_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8880,13 +7782,9 @@ async def _attach_detach_data_disks_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, deserialized, response_headers) - - return deserialized + return cls(pipeline_response, deserialized, response_headers) # type: ignore - _attach_detach_data_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/attachDetachDataDisks" - } + return deserialized # type: ignore @overload async def begin_attach_detach_data_disks( @@ -8913,14 +7811,6 @@ async def begin_attach_detach_data_disks( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -8934,7 +7824,7 @@ async def begin_attach_detach_data_disks( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -8949,18 +7839,10 @@ async def begin_attach_detach_data_disks( :type instance_id: str :param parameters: Parameters supplied to the attach and detach data disks operation on a Virtual Machine Scale Sets VM. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -8974,7 +7856,7 @@ async def begin_attach_detach_data_disks( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -8986,20 +7868,10 @@ async def begin_attach_detach_data_disks( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the attach and detach data disks operation on a - Virtual Machine Scale Sets VM. Is either a AttachDetachDataDisksRequest type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Virtual Machine Scale Sets VM. Is either a AttachDetachDataDisksRequest type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or + IO[bytes] :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -9033,7 +7905,7 @@ async def begin_attach_detach_data_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("StorageProfile", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9045,24 +7917,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.StorageProfile].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_attach_detach_data_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/attachDetachDataDisks" - } + return AsyncLROPoller[_models.StorageProfile]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -9088,7 +7958,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -9097,16 +7967,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9120,13 +7989,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -9152,14 +8017,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -9173,7 +8030,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9187,18 +8044,10 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -9212,7 +8061,7 @@ async def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -9224,19 +8073,8 @@ async def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -9270,7 +8108,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9282,17 +8120,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -9320,7 +8156,7 @@ async def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9346,7 +8182,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9355,16 +8191,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9384,10 +8219,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload async def begin_create_or_update( self, @@ -9414,14 +8245,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9435,7 +8258,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9451,18 +8274,10 @@ async def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9476,7 +8291,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -9489,20 +8304,9 @@ async def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9536,7 +8340,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9546,24 +8350,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -9589,7 +8391,7 @@ async def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -9598,16 +8400,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9619,13 +8420,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -9654,14 +8451,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9675,7 +8464,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9691,18 +8480,10 @@ async def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9716,7 +8497,7 @@ async def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -9729,20 +8510,9 @@ async def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -9776,7 +8546,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -9786,17 +8556,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -9815,22 +8583,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9840,11 +8607,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -9859,14 +8622,6 @@ async def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9894,7 +8649,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -9903,17 +8658,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -9935,7 +8686,6 @@ async def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -9954,23 +8704,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9982,13 +8731,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -10002,7 +8747,6 @@ async def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -10021,22 +8765,21 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10048,13 +8791,9 @@ async def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -10084,7 +8823,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -10107,16 +8845,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> AsyncIterable["_mode def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -10127,14 +8864,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -10144,11 +8881,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10160,15 +8897,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - async def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -10194,7 +8927,7 @@ async def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10202,16 +8935,15 @@ async def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10225,13 +8957,9 @@ async def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload async def begin_capture( @@ -10255,14 +8983,6 @@ async def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10275,7 +8995,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10288,18 +9008,10 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10312,7 +9024,7 @@ async def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -10323,19 +9035,9 @@ async def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -10368,7 +9070,7 @@ async def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10380,23 +9082,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return AsyncLROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachine, IO], + parameters: Union[_models.VirtualMachine, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10425,7 +9125,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10435,16 +9135,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10464,10 +9163,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload async def begin_create_or_update( self, @@ -10500,14 +9195,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10520,7 +9207,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10535,7 +9222,7 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -10547,14 +9234,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10567,7 +9246,7 @@ async def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachine, IO], + parameters: Union[_models.VirtualMachine, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10581,8 +9260,8 @@ async def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine or IO + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -10591,17 +9270,6 @@ async def begin_create_or_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10636,7 +9304,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10646,23 +9314,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineUpdate, IO], + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10691,7 +9357,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -10701,16 +9367,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10722,13 +9387,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -10761,14 +9422,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10781,7 +9434,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10795,7 +9448,7 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -10807,14 +9460,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10827,7 +9472,7 @@ async def begin_update( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineUpdate, IO], + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -10840,8 +9485,8 @@ async def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineUpdate or IO + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineUpdate or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -10850,17 +9495,6 @@ async def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachine or the result of cls(response) :rtype: @@ -10895,7 +9529,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -10905,17 +9539,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -10934,22 +9566,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10959,11 +9590,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -10978,14 +9605,6 @@ async def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11013,7 +9632,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11022,17 +9641,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -11055,7 +9670,6 @@ async def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -11074,22 +9688,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11101,13 +9714,9 @@ async def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace_async async def instance_view( @@ -11119,7 +9728,6 @@ async def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -11138,21 +9746,20 @@ async def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11164,13 +9771,9 @@ async def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11189,21 +9792,20 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11213,11 +9815,7 @@ async def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-ret raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_convert_to_managed_disks( @@ -11230,14 +9828,6 @@ async def begin_convert_to_managed_disks( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11264,7 +9854,7 @@ async def begin_convert_to_managed_disks( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11273,17 +9863,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -11302,22 +9888,21 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11327,11 +9912,7 @@ async def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_deallocate( @@ -11347,14 +9928,6 @@ async def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11382,7 +9955,7 @@ async def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11391,17 +9964,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def generalize( # pylint: disable=inconsistent-return-statements @@ -11418,7 +9987,6 @@ async def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -11437,21 +10005,20 @@ async def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11461,11 +10028,7 @@ async def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -11490,7 +10053,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -11513,18 +10075,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11535,14 +10096,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11552,11 +10113,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11568,10 +10129,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -11596,7 +10153,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -11619,18 +10175,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11641,14 +10196,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -11658,11 +10213,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11674,8 +10229,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11687,7 +10240,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] @@ -11710,17 +10262,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11731,14 +10282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -11748,11 +10299,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11764,10 +10315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - async def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -11785,22 +10332,21 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11810,11 +10356,7 @@ async def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_power_off( @@ -11831,14 +10373,6 @@ async def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11866,7 +10400,7 @@ async def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11875,17 +10409,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -11904,21 +10434,20 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11928,11 +10457,7 @@ async def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -11942,14 +10467,6 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11976,7 +10493,7 @@ async def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -11985,17 +10502,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12014,21 +10527,20 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12038,11 +10550,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12052,14 +10560,6 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12086,7 +10586,7 @@ async def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12095,17 +10595,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12124,21 +10620,20 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12148,11 +10643,7 @@ async def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12162,14 +10653,6 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12196,7 +10679,7 @@ async def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: An def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12205,17 +10688,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12234,21 +10713,20 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12258,11 +10736,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -12272,14 +10746,6 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12306,7 +10772,7 @@ async def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12315,23 +10781,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -12360,7 +10822,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12368,16 +10830,15 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12387,11 +10848,7 @@ async def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload async def begin_reimage( @@ -12420,14 +10877,6 @@ async def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12438,7 +10887,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12456,18 +10905,10 @@ async def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12478,7 +10919,7 @@ async def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> AsyncLROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -12493,19 +10934,9 @@ async def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of AsyncLROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -12535,7 +10966,7 @@ async def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12544,17 +10975,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def retrieve_boot_diagnostics_data( @@ -12575,7 +11002,6 @@ async def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12594,22 +11020,21 @@ async def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12621,13 +11046,9 @@ async def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12646,21 +11067,20 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12670,11 +11090,7 @@ async def _perform_maintenance_initial( # pylint: disable=inconsistent-return-s raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_perform_maintenance( @@ -12686,14 +11102,6 @@ async def begin_perform_maintenance( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12720,7 +11128,7 @@ async def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -12729,17 +11137,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -12751,7 +11155,6 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -12770,21 +11173,20 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12794,11 +11196,7 @@ async def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore async def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -12817,21 +11215,20 @@ async def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12845,13 +11242,9 @@ async def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace_async async def begin_assess_patches( @@ -12863,14 +11256,6 @@ async def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -12900,7 +11285,7 @@ async def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12912,23 +11297,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return AsyncLROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -12954,7 +11337,7 @@ async def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -12962,16 +11345,15 @@ async def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12985,13 +11367,9 @@ async def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload async def begin_install_patches( @@ -13016,14 +11394,6 @@ async def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13036,7 +11406,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13049,18 +11419,10 @@ async def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13073,7 +11435,7 @@ async def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -13083,20 +11445,9 @@ async def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -13129,7 +11480,7 @@ async def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13141,23 +11492,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return AsyncLROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _attach_detach_data_disks_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> Optional[_models.StorageProfile]: error_map = { @@ -13183,7 +11532,7 @@ async def _attach_detach_data_disks_initial( else: _json = self._serialize.body(parameters, "AttachDetachDataDisksRequest") - request = build_virtual_machines_attach_detach_data_disks_request( + _request = build_virtual_machines_attach_detach_data_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -13191,16 +11540,15 @@ async def _attach_detach_data_disks_initial( content_type=content_type, json=_json, content=_content, - template_url=self._attach_detach_data_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13218,13 +11566,9 @@ async def _attach_detach_data_disks_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, deserialized, response_headers) + return cls(pipeline_response, deserialized, response_headers) # type: ignore - return deserialized - - _attach_detach_data_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/attachDetachDataDisks" - } + return deserialized # type: ignore @overload async def begin_attach_detach_data_disks( @@ -13248,14 +11592,6 @@ async def begin_attach_detach_data_disks( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -13268,7 +11604,7 @@ async def begin_attach_detach_data_disks( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13281,18 +11617,10 @@ async def begin_attach_detach_data_disks( :type vm_name: str :param parameters: Parameters supplied to the attach and detach data disks operation on the virtual machine. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -13305,7 +11633,7 @@ async def begin_attach_detach_data_disks( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -13315,19 +11643,9 @@ async def begin_attach_detach_data_disks( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the attach and detach data disks operation on the - virtual machine. Is either a AttachDetachDataDisksRequest type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + virtual machine. Is either a AttachDetachDataDisksRequest type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or + IO[bytes] :return: An instance of AsyncLROPoller that returns either StorageProfile or the result of cls(response) :rtype: @@ -13360,7 +11678,7 @@ async def begin_attach_detach_data_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("StorageProfile", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13372,20 +11690,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.StorageProfile].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_attach_detach_data_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/attachDetachDataDisks" - } + return AsyncLROPoller[_models.StorageProfile]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -13410,7 +11730,7 @@ async def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -13418,16 +11738,15 @@ async def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13441,13 +11760,9 @@ async def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload async def begin_run_command( @@ -13470,14 +11785,6 @@ async def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13490,7 +11797,7 @@ async def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -13502,18 +11809,10 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13523,7 +11822,11 @@ async def begin_run_command( @distributed_trace_async async def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -13532,19 +11835,8 @@ async def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either RunCommandResult or the result of cls(response) :rtype: @@ -13577,7 +11869,7 @@ async def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13589,17 +11881,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return AsyncLROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -13638,7 +11928,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -13657,7 +11946,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -13665,16 +11954,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13686,13 +11974,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -13724,7 +12008,6 @@ async def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13743,7 +12026,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -13753,16 +12036,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13774,13 +12056,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -13792,7 +12070,6 @@ async def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13811,21 +12088,20 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13837,13 +12113,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -13851,7 +12123,6 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13870,20 +12141,19 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13895,13 +12165,9 @@ async def list_publishers(self, location: str, **kwargs: Any) -> List[_models.Vi deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -13915,7 +12181,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -13934,22 +12199,21 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13961,13 +12225,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace_async async def list_by_edge_zone( @@ -13979,7 +12239,6 @@ async def list_by_edge_zone( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -13998,21 +12257,20 @@ async def list_by_edge_zone( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14024,13 +12282,9 @@ async def list_by_edge_zone( deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -14071,7 +12325,6 @@ async def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14090,7 +12343,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -14099,16 +12352,15 @@ async def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14120,13 +12372,9 @@ async def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list( @@ -14163,7 +12411,6 @@ async def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14182,7 +12429,7 @@ async def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -14193,16 +12440,15 @@ async def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14214,13 +12460,9 @@ async def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace_async async def list_offers( @@ -14235,7 +12477,6 @@ async def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14254,22 +12495,21 @@ async def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14281,13 +12521,9 @@ async def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace_async async def list_publishers( @@ -14299,7 +12535,6 @@ async def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14318,21 +12553,20 @@ async def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14344,13 +12578,9 @@ async def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace_async async def list_skus( @@ -14367,7 +12597,6 @@ async def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -14386,23 +12615,22 @@ async def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14414,13 +12642,9 @@ async def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -14457,7 +12681,6 @@ async def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -14476,23 +12699,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14504,13 +12726,9 @@ async def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace_async async def list_types( @@ -14522,7 +12740,6 @@ async def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -14541,21 +12758,20 @@ async def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14567,13 +12783,9 @@ async def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace_async async def list_versions( @@ -14601,7 +12813,6 @@ async def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -14620,7 +12831,7 @@ async def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -14629,16 +12840,15 @@ async def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14650,13 +12860,9 @@ async def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -14700,7 +12906,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14711,7 +12916,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14723,11 +12928,10 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14738,7 +12942,7 @@ async def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -14748,12 +12952,8 @@ async def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14781,7 +12981,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -14789,16 +12989,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14810,13 +13009,9 @@ async def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload async def update( @@ -14839,7 +13034,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14850,7 +13044,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14862,11 +13056,10 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14877,7 +13070,7 @@ async def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -14887,12 +13080,8 @@ async def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -14920,7 +13109,7 @@ async def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -14928,16 +13117,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14949,13 +13137,9 @@ async def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -14967,7 +13151,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14986,21 +13169,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15010,11 +13192,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -15024,7 +13202,6 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -15043,21 +13220,20 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15069,13 +13245,9 @@ async def get(self, resource_group_name: str, availability_set_name: str, **kwar deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -15086,7 +13258,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet] @@ -15109,16 +13280,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15129,14 +13299,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -15146,11 +13316,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15162,17 +13332,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet] @@ -15195,16 +13360,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15215,14 +13379,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -15232,11 +13396,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15248,10 +13412,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -15263,7 +13423,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] @@ -15286,17 +13445,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15307,14 +13465,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -15324,11 +13482,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15340,10 +13498,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -15387,7 +13541,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15398,7 +13551,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15411,11 +13564,10 @@ async def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15426,7 +13578,7 @@ async def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -15436,12 +13588,8 @@ async def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15469,7 +13617,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -15477,16 +13625,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15506,10 +13653,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload async def update( self, @@ -15532,7 +13675,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15543,7 +13685,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15556,11 +13698,10 @@ async def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15571,7 +13712,7 @@ async def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -15581,12 +13722,9 @@ async def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15614,7 +13752,7 @@ async def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -15622,16 +13760,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15643,13 +13780,9 @@ async def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -15661,7 +13794,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -15680,21 +13812,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15704,11 +13835,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -15728,7 +13855,6 @@ async def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -15747,22 +13873,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15774,19 +13899,14 @@ async def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -15810,15 +13930,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Proximit def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15829,14 +13948,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -15846,11 +13965,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15862,10 +13981,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -15874,7 +13989,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -15898,16 +14012,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -15918,14 +14031,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -15935,11 +14048,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15951,10 +14064,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -15999,7 +14108,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16010,7 +14118,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16024,11 +14132,10 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16039,7 +14146,7 @@ async def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -16051,12 +14158,8 @@ async def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16084,7 +14187,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -16092,16 +14195,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16121,10 +14223,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload async def update( self, @@ -16146,7 +14244,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16157,7 +14254,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16169,11 +14266,10 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16184,7 +14280,7 @@ async def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -16194,12 +14290,8 @@ async def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16227,7 +14319,7 @@ async def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -16235,16 +14327,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16256,13 +14347,9 @@ async def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -16274,7 +14361,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -16293,21 +14379,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16317,11 +14402,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -16343,7 +14424,6 @@ async def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -16362,22 +14442,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16389,13 +14468,9 @@ async def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -16406,7 +14481,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup] @@ -16429,16 +14503,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16449,14 +14522,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -16466,11 +14539,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16482,16 +14555,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup] @@ -16514,15 +14582,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.Dedicate def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16533,14 +14600,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -16550,11 +14617,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16566,8 +14633,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -16594,7 +14659,7 @@ async def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -16620,7 +14685,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -16629,16 +14694,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16658,10 +14722,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload async def begin_create_or_update( self, @@ -16686,14 +14746,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16707,7 +14759,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16721,18 +14773,10 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16746,7 +14790,7 @@ async def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -16758,19 +14802,8 @@ async def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16804,7 +14837,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16814,24 +14847,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -16857,7 +14888,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -16866,16 +14897,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16887,13 +14917,9 @@ async def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -16919,14 +14945,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16940,7 +14958,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16954,18 +14972,10 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -16979,7 +14989,7 @@ async def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -16991,19 +15001,8 @@ async def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DedicatedHost or the result of cls(response) :rtype: @@ -17037,7 +15036,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -17047,17 +15046,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -17076,22 +15073,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17101,11 +15097,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -17119,14 +15111,6 @@ async def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17154,7 +15138,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17163,17 +15147,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -17197,7 +15177,6 @@ async def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -17216,23 +15195,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17244,13 +15222,9 @@ async def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -17263,7 +15237,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -17286,17 +15259,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17307,14 +15279,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -17324,11 +15296,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17340,10 +15312,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - async def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -17361,22 +15329,21 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17386,11 +15353,7 @@ async def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_restart( @@ -17407,14 +15370,6 @@ async def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17442,7 +15397,7 @@ async def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17451,17 +15406,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore async def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -17480,22 +15431,21 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_redeploy_request( + _request = build_dedicated_hosts_redeploy_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17508,11 +15458,7 @@ async def _redeploy_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/redeploy" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace_async async def begin_redeploy( @@ -17530,14 +15476,6 @@ async def begin_redeploy( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17565,7 +15503,7 @@ async def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -17574,17 +15512,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/redeploy" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -17600,7 +15534,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -17622,18 +15555,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17644,14 +15576,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -17661,11 +15593,11 @@ async def extract_data(pipeline_response): return None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17677,10 +15609,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -17707,7 +15635,6 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -17731,15 +15658,14 @@ def list_by_subscription(self, **kwargs: Any) -> AsyncIterable["_models.SshPubli def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17750,14 +15676,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -17767,11 +15693,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17783,8 +15709,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -17794,7 +15718,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -17818,16 +15741,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17838,14 +15760,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -17855,11 +15777,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17871,10 +15793,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload async def create( self, @@ -17896,7 +15814,6 @@ async def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -17907,7 +15824,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -17919,11 +15836,10 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -17934,7 +15850,7 @@ async def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -17944,12 +15860,8 @@ async def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -17977,7 +15889,7 @@ async def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -17985,16 +15897,15 @@ async def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18014,10 +15925,6 @@ async def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload async def update( self, @@ -18039,7 +15946,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -18050,7 +15956,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18062,11 +15968,10 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -18077,7 +15982,7 @@ async def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -18087,12 +15992,9 @@ async def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -18120,7 +16022,7 @@ async def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -18128,16 +16030,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18149,13 +16050,9 @@ async def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -18167,7 +16064,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18186,21 +16082,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18210,11 +16105,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -18226,7 +16117,6 @@ async def get( :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -18245,21 +16135,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18271,13 +16160,9 @@ async def get( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @overload async def generate_key_pair( @@ -18302,7 +16187,6 @@ async def generate_key_pair( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18313,7 +16197,7 @@ async def generate_key_pair( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18327,11 +16211,10 @@ async def generate_key_pair( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to generate the SSH public key. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18342,7 +16225,7 @@ async def generate_key_pair( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO]] = None, + parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO[bytes]]] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with @@ -18354,13 +16237,9 @@ async def generate_key_pair( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to generate the SSH public key. Is either a - SshGenerateKeyPairInputParameters type or a IO type. Default value is None. + SshGenerateKeyPairInputParameters type or a IO[bytes] type. Default value is None. :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshGenerateKeyPairInputParameters or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + IO[bytes] :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18391,7 +16270,7 @@ async def generate_key_pair( else: _json = None - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -18399,16 +16278,15 @@ async def generate_key_pair( content_type=content_type, json=_json, content=_content, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18420,13 +16298,9 @@ async def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -18450,7 +16324,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -18475,7 +16349,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -18483,16 +16357,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18512,10 +16385,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_create_or_update( self, @@ -18537,14 +16406,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18555,7 +16416,7 @@ async def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18567,18 +16428,10 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18586,7 +16439,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Create or update an image. @@ -18595,19 +16448,8 @@ async def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.Image or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18638,7 +16480,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18648,20 +16490,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -18686,7 +16530,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -18694,16 +16538,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18723,10 +16566,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload async def begin_update( self, @@ -18748,14 +16587,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18766,7 +16597,7 @@ async def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -18778,18 +16609,10 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18797,7 +16620,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Image]: """Update an image. @@ -18806,19 +16633,8 @@ async def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ImageUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -18849,7 +16665,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18859,17 +16675,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -18888,21 +16702,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18912,11 +16725,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -18926,14 +16735,6 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18960,7 +16761,7 @@ async def begin_delete(self, resource_group_name: str, image_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -18969,17 +16770,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -18993,7 +16790,6 @@ async def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -19012,22 +16808,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19039,13 +16834,9 @@ async def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Image"]: @@ -19054,7 +16845,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19076,16 +16866,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19096,14 +16885,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19113,11 +16902,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19129,16 +16918,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -19160,15 +16944,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19179,14 +16962,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -19196,11 +16979,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19212,8 +16995,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -19259,7 +17040,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19270,7 +17050,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19285,11 +17065,10 @@ async def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19300,7 +17079,7 @@ async def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -19312,12 +17091,8 @@ async def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19345,7 +17120,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -19353,16 +17128,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19382,10 +17156,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload async def update( self, @@ -19408,7 +17178,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19419,7 +17188,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19432,11 +17201,10 @@ async def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19447,7 +17215,7 @@ async def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -19457,12 +17225,9 @@ async def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19490,7 +17255,7 @@ async def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -19498,16 +17263,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19519,13 +17283,9 @@ async def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -19544,21 +17304,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19568,11 +17327,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -19585,14 +17340,6 @@ async def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -19619,7 +17366,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -19628,17 +17375,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -19660,7 +17403,6 @@ async def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -19679,22 +17421,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19706,13 +17447,9 @@ async def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: @@ -19720,7 +17457,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -19744,16 +17480,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_model def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19764,14 +17499,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -19781,11 +17516,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19797,17 +17532,12 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -19831,15 +17561,14 @@ def list_all(self, **kwargs: Any) -> AsyncIterable["_models.RestorePointCollecti def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -19850,14 +17579,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -19867,11 +17596,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19883,8 +17612,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -19911,7 +17638,7 @@ async def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -19937,7 +17664,7 @@ async def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -19946,16 +17673,15 @@ async def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19967,13 +17693,9 @@ async def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload async def begin_create( @@ -20000,14 +17722,6 @@ async def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -20020,7 +17734,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20035,18 +17749,10 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -20059,7 +17765,7 @@ async def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -20072,19 +17778,8 @@ async def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint or IO[bytes] :return: An instance of AsyncLROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -20117,7 +17812,7 @@ async def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -20127,17 +17822,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -20156,22 +17849,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20181,11 +17873,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -20199,14 +17887,6 @@ async def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -20234,7 +17914,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -20243,17 +17923,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -20276,7 +17952,6 @@ async def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -20295,23 +17970,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20323,13 +17997,9 @@ async def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -20375,7 +18045,6 @@ async def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20386,7 +18055,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20400,11 +18069,10 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20415,7 +18083,7 @@ async def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -20427,12 +18095,8 @@ async def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20460,7 +18124,7 @@ async def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -20468,16 +18132,15 @@ async def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20497,10 +18160,6 @@ async def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload async def update( self, @@ -20524,7 +18183,6 @@ async def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20535,7 +18193,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20549,11 +18207,10 @@ async def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20564,7 +18221,7 @@ async def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -20575,12 +18232,9 @@ async def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20608,7 +18262,7 @@ async def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -20616,16 +18270,15 @@ async def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20637,13 +18290,9 @@ async def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace_async async def delete( # pylint: disable=inconsistent-return-statements @@ -20658,7 +18307,6 @@ async def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20677,21 +18325,20 @@ async def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20701,11 +18348,7 @@ async def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def get( @@ -20729,7 +18372,6 @@ async def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -20748,22 +18390,21 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20775,13 +18416,9 @@ async def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -20802,7 +18439,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -20826,17 +18462,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20847,14 +18482,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -20864,11 +18499,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20880,10 +18515,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -20897,7 +18528,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -20921,16 +18551,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -20941,14 +18570,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -20958,11 +18587,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20974,10 +18603,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -21004,7 +18629,7 @@ async def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -21030,7 +18655,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21039,16 +18664,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21068,10 +18692,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload async def begin_create_or_update( self, @@ -21098,14 +18718,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21119,7 +18731,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21135,18 +18747,10 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21160,7 +18764,7 @@ async def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -21174,19 +18778,8 @@ async def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21220,7 +18813,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21230,24 +18823,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -21273,7 +18864,7 @@ async def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -21282,16 +18873,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21305,13 +18895,9 @@ async def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -21337,14 +18923,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21358,7 +18936,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21372,18 +18950,10 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21397,7 +18967,7 @@ async def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -21409,19 +18979,8 @@ async def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -21455,7 +19014,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21465,17 +19024,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -21498,22 +19055,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21523,11 +19079,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -21547,14 +19099,6 @@ async def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -21582,7 +19126,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -21591,17 +19135,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -21626,7 +19166,6 @@ async def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -21645,23 +19184,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21673,13 +19211,9 @@ async def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -21692,7 +19226,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_09_01.models.CapacityReservation] @@ -21715,17 +19248,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21736,14 +19268,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -21753,11 +19285,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21769,10 +19301,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -21795,7 +19323,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -21820,23 +19348,22 @@ async def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21850,13 +19377,9 @@ async def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload async def begin_export_request_rate_by_interval( @@ -21878,14 +19401,6 @@ async def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21895,7 +19410,7 @@ async def begin_export_request_rate_by_interval( @overload async def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21904,18 +19419,10 @@ async def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21925,7 +19432,7 @@ async def begin_export_request_rate_by_interval( @distributed_trace_async async def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -21933,19 +19440,9 @@ async def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -21977,7 +19474,7 @@ async def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -21990,20 +19487,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -22028,23 +19523,22 @@ async def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22058,13 +19552,9 @@ async def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload async def begin_export_throttled_requests( @@ -22085,14 +19575,6 @@ async def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22102,7 +19584,7 @@ async def begin_export_throttled_requests( @overload async def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -22110,18 +19592,10 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22131,7 +19605,7 @@ async def begin_export_throttled_requests( @distributed_trace_async async def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -22139,19 +19613,8 @@ async def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of AsyncLROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -22183,7 +19646,7 @@ async def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22196,17 +19659,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return AsyncLROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -22235,7 +19696,6 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -22259,16 +19719,15 @@ def list(self, location: str, **kwargs: Any) -> AsyncIterable["_models.RunComman def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22279,14 +19738,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -22296,11 +19755,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22312,10 +19771,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace_async async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -22324,7 +19779,6 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -22343,21 +19797,20 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22369,20 +19822,16 @@ async def get(self, location: str, command_id: str, **kwargs: Any) -> _models.Ru deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore async def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22408,7 +19857,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -22417,16 +19866,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22446,10 +19894,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -22476,14 +19920,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22497,7 +19933,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22513,18 +19949,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22538,7 +19966,7 @@ async def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -22551,19 +19979,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22597,7 +20014,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22607,24 +20024,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -22650,7 +20065,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -22659,16 +20074,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22680,13 +20094,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -22714,14 +20124,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22735,7 +20137,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22751,18 +20153,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22776,7 +20170,7 @@ async def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -22789,19 +20183,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -22835,7 +20219,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22845,17 +20229,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -22874,22 +20256,21 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22899,11 +20280,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -22918,14 +20295,6 @@ async def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -22953,7 +20322,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -22962,17 +20331,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get_by_virtual_machine( @@ -22994,7 +20359,6 @@ async def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -23013,23 +20377,22 @@ async def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23041,13 +20404,9 @@ async def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -23061,7 +20420,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23085,18 +20443,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23107,14 +20464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -23124,11 +20481,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23140,12 +20497,8 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -23171,7 +20524,7 @@ async def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23197,7 +20550,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23207,16 +20560,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23236,10 +20588,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload async def begin_create_or_update( self, @@ -23268,14 +20616,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23290,7 +20630,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23307,18 +20647,10 @@ async def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23333,7 +20665,7 @@ async def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -23347,19 +20679,8 @@ async def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23394,7 +20715,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23404,17 +20725,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, @@ -23422,7 +20741,7 @@ async def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -23448,7 +20767,7 @@ async def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23458,16 +20777,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23479,13 +20797,9 @@ async def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload async def begin_update( @@ -23515,14 +20829,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23537,7 +20843,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23554,18 +20860,10 @@ async def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23580,7 +20878,7 @@ async def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -23594,19 +20892,9 @@ async def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23641,7 +20929,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23651,17 +20939,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -23680,23 +20966,22 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23706,11 +20991,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -23726,14 +21007,6 @@ async def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23762,7 +21035,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -23771,17 +21044,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace_async async def get( @@ -23806,7 +21075,6 @@ async def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -23825,7 +21093,7 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -23833,16 +21101,15 @@ async def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23854,13 +21121,9 @@ async def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -23882,7 +21145,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -23906,19 +21168,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23929,14 +21190,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -23946,11 +21207,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23961,7 +21222,3 @@ async def get_next(next_link=None): return pipeline_response return AsyncItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_compute_management_client_enums.py index b059cb0717bd..230c82b699fa 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_compute_management_client_enums.py @@ -470,7 +470,6 @@ class PublicIPAddressSkuTier(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Specify public IP sku tier.""" REGIONAL = "Regional" - GLOBAL = "Global" GLOBAL_ENUM = "Global" @@ -516,10 +515,10 @@ class RestorePointEncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk Restore Point is encrypted at rest with Platform managed key.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk Restore Point is encrypted at rest with Customer managed key that can be changed and - #: revoked by a customer.""" + revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk Restore Point is encrypted at rest with 2 layers of encryption. One of the keys is - #: Customer managed and the other key is Platform managed.""" + Customer managed and the other key is Platform managed.""" class RestorePointExpandOptions(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_models_py3.py index 79455b2dcf19..3350e9c748e6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/models/_models_py3.py @@ -9,7 +9,7 @@ import datetime import sys -from typing import Any, Dict, List, Optional, TYPE_CHECKING, Union +from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union from ... import _serialization @@ -17,10 +17,6 @@ from collections.abc import MutableMapping else: from typing import MutableMapping # type: ignore # pylint: disable=ungrouped-imports -if sys.version_info >= (3, 8): - from typing import Literal # pylint: disable=no-name-in-module, ungrouped-imports -else: - from typing_extensions import Literal # type: ignore # pylint: disable=ungrouped-imports if TYPE_CHECKING: # pylint: disable=unused-import,ungrouped-imports @@ -339,7 +335,7 @@ class AutomaticOSUpgradePolicy(_serialization.Model): applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :vartype enable_automatic_os_upgrade: bool :ivar disable_automatic_rollback: Whether OS image rollback feature should be disabled. Default @@ -377,7 +373,7 @@ def __init__( applied to scale set instances in a rolling fashion when a newer version of the OS image becomes available. Default value is false. If this is set to true for Windows based scale sets, `enableAutomaticUpdates - `_ + `_ # pylint: disable=line-too-long is automatically set to false and cannot be set to true. :paramtype enable_automatic_os_upgrade: bool :keyword disable_automatic_rollback: Whether OS image rollback feature should be disabled. @@ -403,7 +399,7 @@ def __init__( class AutomaticOSUpgradeProperties(_serialization.Model): """Describes automatic OS upgrade properties on the image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar automatic_os_upgrade_supported: Specifies whether automatic OS upgrade is supported on the image. Required. @@ -486,7 +482,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -543,7 +539,7 @@ class AvailabilitySet(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -640,7 +636,7 @@ def __init__( class AvailabilitySetListResult(_serialization.Model): """The List Availability Set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of availability sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet] @@ -948,7 +944,7 @@ class CapacityReservation(Resource): # pylint: disable=too-many-instance-attrib Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1074,7 +1070,7 @@ class CapacityReservationGroup(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -1206,7 +1202,7 @@ def __init__(self, **kwargs: Any) -> None: class CapacityReservationGroupListResult(_serialization.Model): """The List capacity reservation group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservation groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup] @@ -1388,7 +1384,7 @@ def __init__( class CapacityReservationListResult(_serialization.Model): """The list capacity reservation operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of capacity reservations. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.CapacityReservation] @@ -1638,7 +1634,7 @@ class DataDisk(_serialization.Model): # pylint: disable=too-many-instance-attri Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar lun: Specifies the logical unit number of the data disk. This value is used to identify data disks within the VM and therefore must be unique for each data disk attached to a VM. @@ -1837,7 +1833,7 @@ def __init__(self, **kwargs: Any) -> None: class DataDisksToAttach(_serialization.Model): """Describes the data disk to be attached. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_id: ID of the managed data disk. Required. :vartype disk_id: str @@ -1873,7 +1869,7 @@ def __init__(self, *, disk_id: str, lun: Optional[int] = None, **kwargs: Any) -> class DataDisksToDetach(_serialization.Model): """Describes the data disk to be detached. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar disk_id: ID of the managed data disk. Required. :vartype disk_id: str @@ -1915,7 +1911,7 @@ class DedicatedHost(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2095,7 +2091,7 @@ class DedicatedHostGroup(Resource): # pylint: disable=too-many-instance-attribu Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2223,7 +2219,7 @@ def __init__( class DedicatedHostGroupListResult(_serialization.Model): """The List Dedicated Host Group with resource group response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated host groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup] @@ -2256,7 +2252,7 @@ def __init__( self.next_link = next_link -class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): +class DedicatedHostGroupPropertiesAdditionalCapabilities(_serialization.Model): # pylint: disable=name-too-long """Enables or disables a capability on the dedicated host group. Minimum api-version: 2022-03-01. :ivar ultra_ssd_enabled: The flag that enables or disables a capability to have UltraSSD @@ -2471,7 +2467,7 @@ def __init__( class DedicatedHostListResult(_serialization.Model): """The list dedicated host operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of dedicated hosts. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -2749,17 +2745,6 @@ class DiskEncryptionSetParameters(SubResource): :vartype id: str """ - _attribute_map = { - "id": {"key": "id", "type": "str"}, - } - - def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin - """ - :keyword id: Resource Id. - :paramtype id: str - """ - super().__init__(id=id, **kwargs) - class DiskEncryptionSettings(_serialization.Model): """Describes a Encryption Settings for a Disk. @@ -3166,7 +3151,7 @@ class Image(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -3344,7 +3329,7 @@ def __init__( class ImageDataDisk(ImageDisk): """Describes a data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_09_01.models.SubResource @@ -3495,7 +3480,7 @@ def __init__( class ImageListResult(_serialization.Model): """The List Image operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of Images. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.Image] @@ -3529,7 +3514,7 @@ def __init__(self, *, value: List["_models.Image"], next_link: Optional[str] = N class ImageOSDisk(ImageDisk): """Describes an Operating System disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar snapshot: The snapshot. :vartype snapshot: ~azure.mgmt.compute.v2023_09_01.models.SubResource @@ -3667,7 +3652,7 @@ class ImageReference(SubResource): time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :vartype version: str :ivar exact_version: Specifies in decimal numbers, the version of platform image or marketplace @@ -3726,7 +3711,7 @@ def __init__( after deploy time even if a new version becomes available. Please do not use field 'version' for gallery image deployment, gallery image should always use 'id' field for deployment, to use 'latest' version of gallery image, just set - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/images/{imageName}' # pylint: disable=line-too-long in the 'id' field without version input. :paramtype version: str :keyword shared_gallery_image_id: Specified the shared gallery image unique id for vm @@ -3948,7 +3933,7 @@ def __init__( class KeyVaultKeyReference(_serialization.Model): """Describes a reference to Key Vault Key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar key_url: The URL referencing a key encryption key in Key Vault. Required. :vartype key_url: str @@ -3981,7 +3966,7 @@ def __init__(self, *, key_url: str, source_vault: "_models.SubResource", **kwarg class KeyVaultSecretReference(_serialization.Model): """Describes a reference to Key Vault Secret. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar secret_url: The URL referencing a secret in a Key Vault. Required. :vartype secret_url: str @@ -4286,7 +4271,7 @@ def __init__( self.automatic_by_platform_settings = automatic_by_platform_settings -class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class LinuxVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Linux patch settings. @@ -4331,7 +4316,7 @@ def __init__( class ListUsagesResult(_serialization.Model): """The List Usages operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of compute resource usages. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.Usage] @@ -4365,7 +4350,7 @@ def __init__(self, *, value: List["_models.Usage"], next_link: Optional[str] = N class LogAnalyticsInputBase(_serialization.Model): """Api input base class for LogAnalytics Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -4719,7 +4704,7 @@ def __init__( class OrchestrationServiceStateInput(_serialization.Model): """The input for OrchestrationServiceState. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar service_name: The name of the service. Required. Known values are: "AutomaticRepairs" and "DummyOrchestrationServiceName". @@ -4795,7 +4780,7 @@ class OSDisk(_serialization.Model): # pylint: disable=too-many-instance-attribu information about disks, see `About disks and VHDs for Azure virtual machines `_. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar os_type: This property allows you to specify the type of the OS that is included in the disk if creating a VM from user-image or a specialized VHD. Possible values are: **Windows,** @@ -4945,7 +4930,7 @@ def __init__( class OSDiskImage(_serialization.Model): """Contains the os disk image information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar operating_system: The operating system of the osDiskImage. Required. Known values are: "Windows" and "Linux". @@ -5482,7 +5467,7 @@ class ProximityPlacementGroup(Resource): # pylint: disable=too-many-instance-at Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -5591,7 +5576,7 @@ def __init__( class ProximityPlacementGroupListResult(_serialization.Model): """The List Proximity Placement Group operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of proximity placement groups. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup] @@ -5651,17 +5636,6 @@ class ProximityPlacementGroupUpdate(UpdateResource): :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: Resource tags. - :paramtype tags: dict[str, str] - """ - super().__init__(tags=tags, **kwargs) - class ProxyAgentSettings(_serialization.Model): """Specifies ProxyAgent settings while creating the virtual machine. Minimum api-version: @@ -5752,7 +5726,7 @@ class PublicIPAddressSku(_serialization.Model): :ivar name: Specify public IP sku name. Known values are: "Basic" and "Standard". :vartype name: str or ~azure.mgmt.compute.v2023_09_01.models.PublicIPAddressSkuName - :ivar tier: Specify public IP sku tier. Known values are: "Regional", "Global", and "Global". + :ivar tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :vartype tier: str or ~azure.mgmt.compute.v2023_09_01.models.PublicIPAddressSkuTier """ @@ -5771,8 +5745,7 @@ def __init__( """ :keyword name: Specify public IP sku name. Known values are: "Basic" and "Standard". :paramtype name: str or ~azure.mgmt.compute.v2023_09_01.models.PublicIPAddressSkuName - :keyword tier: Specify public IP sku tier. Known values are: "Regional", "Global", and - "Global". + :keyword tier: Specify public IP sku tier. Known values are: "Regional" and "Global". :paramtype tier: str or ~azure.mgmt.compute.v2023_09_01.models.PublicIPAddressSkuTier """ super().__init__(**kwargs) @@ -5783,7 +5756,7 @@ def __init__( class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar publisher: The publisher ID. Required. :vartype publisher: str @@ -5854,7 +5827,7 @@ def __init__(self, **kwargs: Any) -> None: class RequestRateByIntervalInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getRequestRateByInterval Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -6198,7 +6171,7 @@ class RestorePointCollection(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -6988,7 +6961,7 @@ class RollingUpgradeStatusInfo(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7051,7 +7024,7 @@ def __init__(self, *, location: str, tags: Optional[Dict[str, str]] = None, **kw class RunCommandDocumentBase(_serialization.Model): """Describes the properties of a Run Command metadata. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7114,7 +7087,7 @@ def __init__( class RunCommandDocument(RunCommandDocumentBase): """Describes the properties of a Run Command. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar schema: The VM run command schema. Required. :vartype schema: str @@ -7188,7 +7161,7 @@ def __init__( class RunCommandInput(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar command_id: The run command id. Required. :vartype command_id: str @@ -7235,7 +7208,7 @@ def __init__( class RunCommandInputParameter(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7268,7 +7241,7 @@ def __init__(self, *, name: str, value: str, **kwargs: Any) -> None: class RunCommandListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.RunCommandDocumentBase] @@ -7341,7 +7314,7 @@ def __init__(self, *, client_id: Optional[str] = None, object_id: Optional[str] class RunCommandParameterDefinition(_serialization.Model): """Describes the properties of a run command parameter. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The run command parameter name. Required. :vartype name: str @@ -7516,7 +7489,7 @@ class SecurityPostureReference(_serialization.Model): api-version: 2023-03-01. :ivar id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :vartype id: str :ivar exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7538,7 +7511,7 @@ def __init__( ) -> None: """ :keyword id: The security posture reference id in the form of - /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. + /CommunityGalleries/{communityGalleryName}/securityPostures/{securityPostureName}/versions/{major.minor.patch}|{major.*}|latest. # pylint: disable=line-too-long :paramtype id: str :keyword exclude_extensions: List of virtual machine extensions to exclude when applying the Security Posture. @@ -7626,7 +7599,7 @@ class ServiceArtifactReference(_serialization.Model): machines in the scale set when using 'latest' image version. Minimum api-version: 2022-11-01. :ivar id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :vartype id: str """ @@ -7637,7 +7610,7 @@ class ServiceArtifactReference(_serialization.Model): def __init__(self, *, id: Optional[str] = None, **kwargs: Any) -> None: # pylint: disable=redefined-builtin """ :keyword id: The service artifact reference id in the form of - /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. + /subscriptions/{subscriptionId}/resourceGroups/{resourceGroup}/providers/Microsoft.Compute/galleries/{galleryName}/serviceArtifacts/{serviceArtifactName}/vmArtifactsProfiles/{vmArtifactsProfilesName}. # pylint: disable=line-too-long :paramtype id: str """ super().__init__(**kwargs) @@ -7803,7 +7776,7 @@ def __init__(self, *, path: Optional[str] = None, key_data: Optional[str] = None class SshPublicKeyGenerateKeyPairResult(_serialization.Model): """Response from generation of an SSH key pair. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar private_key: Private key portion of the key pair used to authenticate to a virtual machine through ssh. The private key is returned in RFC3447 format and should be treated as a @@ -7813,7 +7786,7 @@ class SshPublicKeyGenerateKeyPairResult(_serialization.Model): through ssh. The public key is in ssh-rsa format. Required. :vartype public_key: str :ivar id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :vartype id: str """ @@ -7842,7 +7815,7 @@ def __init__( machine through ssh. The public key is in ssh-rsa format. Required. :paramtype public_key: str :keyword id: The ARM resource id in the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{SshPublicKeyName}. # pylint: disable=line-too-long Required. :paramtype id: str """ @@ -7857,7 +7830,7 @@ class SshPublicKeyResource(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -7913,7 +7886,7 @@ def __init__( class SshPublicKeysGroupListResult(_serialization.Model): """The list SSH public keys operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of SSH public keys. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource] @@ -8153,7 +8126,7 @@ def __init__( class ThrottledRequestsInput(LogAnalyticsInputBase): """Api request input for LogAnalytics getThrottledRequests Api. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics Api writes output logs to. Required. @@ -8174,67 +8147,6 @@ class ThrottledRequestsInput(LogAnalyticsInputBase): :vartype group_by_user_agent: bool """ - _validation = { - "blob_container_sas_uri": {"required": True}, - "from_time": {"required": True}, - "to_time": {"required": True}, - } - - _attribute_map = { - "blob_container_sas_uri": {"key": "blobContainerSasUri", "type": "str"}, - "from_time": {"key": "fromTime", "type": "iso-8601"}, - "to_time": {"key": "toTime", "type": "iso-8601"}, - "group_by_throttle_policy": {"key": "groupByThrottlePolicy", "type": "bool"}, - "group_by_operation_name": {"key": "groupByOperationName", "type": "bool"}, - "group_by_resource_name": {"key": "groupByResourceName", "type": "bool"}, - "group_by_client_application_id": {"key": "groupByClientApplicationId", "type": "bool"}, - "group_by_user_agent": {"key": "groupByUserAgent", "type": "bool"}, - } - - def __init__( - self, - *, - blob_container_sas_uri: str, - from_time: datetime.datetime, - to_time: datetime.datetime, - group_by_throttle_policy: Optional[bool] = None, - group_by_operation_name: Optional[bool] = None, - group_by_resource_name: Optional[bool] = None, - group_by_client_application_id: Optional[bool] = None, - group_by_user_agent: Optional[bool] = None, - **kwargs: Any - ) -> None: - """ - :keyword blob_container_sas_uri: SAS Uri of the logging blob container to which LogAnalytics - Api writes output logs to. Required. - :paramtype blob_container_sas_uri: str - :keyword from_time: From time of the query. Required. - :paramtype from_time: ~datetime.datetime - :keyword to_time: To time of the query. Required. - :paramtype to_time: ~datetime.datetime - :keyword group_by_throttle_policy: Group query result by Throttle Policy applied. - :paramtype group_by_throttle_policy: bool - :keyword group_by_operation_name: Group query result by Operation Name. - :paramtype group_by_operation_name: bool - :keyword group_by_resource_name: Group query result by Resource Name. - :paramtype group_by_resource_name: bool - :keyword group_by_client_application_id: Group query result by Client Application ID. - :paramtype group_by_client_application_id: bool - :keyword group_by_user_agent: Group query result by User Agent. - :paramtype group_by_user_agent: bool - """ - super().__init__( - blob_container_sas_uri=blob_container_sas_uri, - from_time=from_time, - to_time=to_time, - group_by_throttle_policy=group_by_throttle_policy, - group_by_operation_name=group_by_operation_name, - group_by_resource_name=group_by_resource_name, - group_by_client_application_id=group_by_client_application_id, - group_by_user_agent=group_by_user_agent, - **kwargs - ) - class UefiSettings(_serialization.Model): """Specifies the security settings like secure boot and vTPM used while creating the virtual @@ -8303,7 +8215,7 @@ def __init__(self, **kwargs: Any) -> None: self.location = None -class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): +class UpgradeOperationHistoricalStatusInfoProperties(_serialization.Model): # pylint: disable=name-too-long """Describes each OS upgrade on the Virtual Machine Scale Set. Variables are only populated by the server, and will be ignored when sending a request. @@ -8446,7 +8358,7 @@ class Usage(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar unit: An enum describing the unit of usage measurement. Required. Default value is "Count". @@ -8664,7 +8576,7 @@ class VirtualMachine(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9156,7 +9068,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineCaptureParameters(_serialization.Model): """Capture Virtual Machine parameters. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar vhd_prefix: The captured virtual hard disk's name prefix. Required. :vartype vhd_prefix: str @@ -9404,7 +9316,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): +class VirtualMachineExtensionHandlerInstanceView(_serialization.Model): # pylint: disable=name-too-long """The instance view of a virtual machine extension handler. :ivar type: Specifies the type of the extension; an example is "CustomScriptExtension". @@ -9448,7 +9360,7 @@ class VirtualMachineExtensionImage(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9763,7 +9675,7 @@ class VirtualMachineIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_09_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_09_01.models.UserAssignedIdentitiesValue] """ @@ -9795,7 +9707,7 @@ def __init__( :paramtype type: str or ~azure.mgmt.compute.v2023_09_01.models.ResourceIdentityType :keyword user_assigned_identities: The list of user identities associated with the Virtual Machine. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_09_01.models.UserAssignedIdentitiesValue] """ @@ -9809,7 +9721,7 @@ def __init__( class VirtualMachineImageResource(SubResource): """Virtual machine image resource information. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -9872,7 +9784,7 @@ def __init__( class VirtualMachineImage(VirtualMachineImageResource): # pylint: disable=too-many-instance-attributes """Describes a Virtual Machine Image. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -10035,7 +9947,7 @@ def __init__(self, *, name: Optional[str] = None, value: Optional[str] = None, * class VirtualMachineInstallPatchesParameters(_serialization.Model): """Input for InstallPatches as directly received by the API. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar maximum_duration: Specifies the maximum amount of time that the operation will run. It must be an ISO 8601-compliant duration string such as PT4H (4 hours). @@ -10363,7 +10275,7 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No class VirtualMachineListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machines. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -10396,10 +10308,12 @@ def __init__( self.next_link = next_link -class VirtualMachineNetworkInterfaceConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineNetworkInterfaceConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine network interface configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network interface configuration name. Required. :vartype name: str @@ -10537,7 +10451,7 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -10557,10 +10471,10 @@ def __init__(self, *, dns_servers: Optional[List[str]] = None, **kwargs: Any) -> self.dns_servers = dns_servers -class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): +class VirtualMachineNetworkInterfaceIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -10726,10 +10640,10 @@ def __init__( self.configuration_statuses = None -class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -10829,10 +10743,10 @@ def __init__( self.public_ip_allocation_method = public_ip_allocation_method -class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): +class VirtualMachinePublicIPAddressDnsSettingsConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label prefix of the PublicIPAddress resources that will be created. The generated name label is the concatenation of the domain name label and vm @@ -10930,7 +10844,7 @@ class VirtualMachineRunCommand(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11269,7 +11183,7 @@ def __init__( class VirtualMachineRunCommandsListResult(_serialization.Model): """The List run command operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of run commands. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand] @@ -11489,7 +11403,7 @@ class VirtualMachineScaleSet(Resource): # pylint: disable=too-many-instance-att Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -11769,7 +11683,7 @@ def __init__( # pylint: disable=too-many-locals class VirtualMachineScaleSetDataDisk(_serialization.Model): """Describes a virtual machine scale set data disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -12045,10 +11959,10 @@ def __init__( self.protected_settings_from_key_vault = protected_settings_from_key_vault -class VirtualMachineScaleSetExtensionListResult(_serialization.Model): +class VirtualMachineScaleSetExtensionListResult(_serialization.Model): # pylint: disable=name-too-long """The List VM scale set extension operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of VM scale set extensions. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension] @@ -12306,7 +12220,7 @@ class VirtualMachineScaleSetIdentity(_serialization.Model): :vartype type: str or ~azure.mgmt.compute.v2023_09_01.models.ResourceIdentityType :ivar user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_09_01.models.UserAssignedIdentitiesValue] """ @@ -12340,7 +12254,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the virtual machine scale set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_09_01.models.UserAssignedIdentitiesValue] """ @@ -12394,7 +12308,7 @@ def __init__(self, *, statuses: Optional[List["_models.InstanceViewStatus"]] = N self.orchestration_services = None -class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): +class VirtualMachineScaleSetInstanceViewStatusesSummary(_serialization.Model): # pylint: disable=name-too-long """Instance view statuses summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -12421,7 +12335,7 @@ def __init__(self, **kwargs: Any) -> None: class VirtualMachineScaleSetIPConfiguration(_serialization.Model): """Describes a virtual machine scale set network profile's IP configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The IP configuration name. Required. :vartype name: str @@ -12570,10 +12484,10 @@ def __init__(self, *, ip_tag_type: Optional[str] = None, tag: Optional[str] = No self.tag = tag -class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): +class VirtualMachineScaleSetListOSUpgradeHistory(_serialization.Model): # pylint: disable=name-too-long """List of Virtual Machine Scale Set OS Upgrade History operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of OS upgrades performed on the virtual machine scale set. Required. :vartype value: @@ -12615,7 +12529,7 @@ def __init__( class VirtualMachineScaleSetListResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet] @@ -12651,7 +12565,7 @@ def __init__( class VirtualMachineScaleSetListSkusResult(_serialization.Model): """The Virtual Machine Scale Set List Skus operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of skus available for the virtual machine scale set. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetSku] @@ -12687,7 +12601,7 @@ def __init__( class VirtualMachineScaleSetListWithLinkResult(_serialization.Model): """The List Virtual Machine operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet] @@ -12720,7 +12634,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): +class VirtualMachineScaleSetManagedDiskParameters(_serialization.Model): # pylint: disable=name-too-long """Describes the parameters of a ScaleSet managed disk. :ivar storage_account_type: Specifies the storage account type for the managed disk. NOTE: @@ -12771,10 +12685,12 @@ def __init__( self.security_profile = security_profile -class VirtualMachineScaleSetNetworkConfiguration(_serialization.Model): # pylint: disable=too-many-instance-attributes +class VirtualMachineScaleSetNetworkConfiguration( + _serialization.Model +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The network configuration name. Required. :vartype name: str @@ -12902,7 +12818,7 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetNetworkConfigurationDnsSettings(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. :ivar dns_servers: List of DNS servers IP addresses. @@ -12927,7 +12843,7 @@ class VirtualMachineScaleSetNetworkProfile(_serialization.Model): :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_09_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -12958,7 +12874,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_09_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -12977,7 +12893,7 @@ def __init__( class VirtualMachineScaleSetOSDisk(_serialization.Model): # pylint: disable=too-many-instance-attributes """Describes a virtual machine scale set operating system disk. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The disk name. :vartype name: str @@ -13271,10 +13187,10 @@ def __init__( self.require_guest_provision_signal = require_guest_provision_signal -class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The publicIP address configuration name. Required. :vartype name: str @@ -13363,10 +13279,12 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings(_serialization.Model): +class VirtualMachineScaleSetPublicIPAddressConfigurationDnsSettings( + _serialization.Model +): # pylint: disable=name-too-long """Describes a virtual machines scale sets network configuration's DNS settings. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar domain_name_label: The Domain name label.The concatenation of the domain name label and vm index will be the domain name labels of the PublicIPAddress resources that will be created. @@ -13413,7 +13331,7 @@ def __init__( self.domain_name_label_scope = domain_name_label_scope -class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): +class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters): # pylint: disable=name-too-long """Describes a Virtual Machine Scale Set VM Reimage Parameters. :ivar temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This temp @@ -13427,33 +13345,6 @@ class VirtualMachineScaleSetVMReimageParameters(VirtualMachineReimageParameters) :vartype os_profile: ~azure.mgmt.compute.v2023_09_01.models.OSProfileProvisioningData """ - _attribute_map = { - "temp_disk": {"key": "tempDisk", "type": "bool"}, - "exact_version": {"key": "exactVersion", "type": "str"}, - "os_profile": {"key": "osProfile", "type": "OSProfileProvisioningData"}, - } - - def __init__( - self, - *, - temp_disk: Optional[bool] = None, - exact_version: Optional[str] = None, - os_profile: Optional["_models.OSProfileProvisioningData"] = None, - **kwargs: Any - ) -> None: - """ - :keyword temp_disk: Specifies whether to reimage temp disk. Default value: false. Note: This - temp disk reimage parameter is only supported for VM/VMSS with Ephemeral OS disk. - :paramtype temp_disk: bool - :keyword exact_version: Specifies in decimal number, the version the OS disk should be reimaged - to. If exact version is not provided, the OS disk is reimaged to the existing version of OS - Disk. - :paramtype exact_version: str - :keyword os_profile: Specifies information required for reimaging the non-ephemeral OS disk. - :paramtype os_profile: ~azure.mgmt.compute.v2023_09_01.models.OSProfileProvisioningData - """ - super().__init__(temp_disk=temp_disk, exact_version=exact_version, os_profile=os_profile, **kwargs) - class VirtualMachineScaleSetReimageParameters(VirtualMachineScaleSetVMReimageParameters): """Describes a Virtual Machine Scale Set VM Reimage Parameters. @@ -13814,7 +13705,7 @@ def __init__( self.resiliency_policy = resiliency_policy -class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdateIPConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile's IP configuration. NOTE: The subnet of a scale set may be modified as long as the original subnet and the new subnet are in the same virtual network. @@ -13928,7 +13819,7 @@ def __init__( class VirtualMachineScaleSetUpdateNetworkConfiguration( _serialization.Model -): # pylint: disable=too-many-instance-attributes +): # pylint: disable=too-many-instance-attributes,name-too-long """Describes a virtual machine scale set network profile's network configurations. :ivar name: The network configuration name. @@ -14054,12 +13945,12 @@ def __init__( self.auxiliary_sku = auxiliary_sku -class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateNetworkProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set network profile. :ivar health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :vartype health_probe: ~azure.mgmt.compute.v2023_09_01.models.ApiEntityReference :ivar network_interface_configurations: The list of network configurations. :vartype network_interface_configurations: @@ -14092,7 +13983,7 @@ def __init__( """ :keyword health_probe: A reference to a load balancer probe used to determine the health of an instance in the virtual machine scale set. The reference will be in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Network/loadBalancers/{loadBalancerName}/probes/{probeName}'. # pylint: disable=line-too-long :paramtype health_probe: ~azure.mgmt.compute.v2023_09_01.models.ApiEntityReference :keyword network_interface_configurations: The list of network configurations. :paramtype network_interface_configurations: @@ -14250,7 +14141,7 @@ def __init__( self.secrets = secrets -class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): +class VirtualMachineScaleSetUpdatePublicIPAddressConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machines scale set IP Configuration's PublicIPAddress configuration. :ivar name: The publicIP address configuration name. @@ -14310,7 +14201,7 @@ def __init__( self.delete_option = delete_option -class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): +class VirtualMachineScaleSetUpdateStorageProfile(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set storage profile. :ivar image_reference: The image reference. @@ -14476,7 +14367,7 @@ class VirtualMachineScaleSetVM(Resource): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -14901,7 +14792,7 @@ def __init__( self.provision_after_extensions = provision_after_extensions -class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsListResult(_serialization.Model): # pylint: disable=name-too-long """The List VMSS VM Extension operation response. :ivar value: The list of VMSS VM extensions. @@ -14924,7 +14815,7 @@ def __init__( self.value = value -class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): +class VirtualMachineScaleSetVMExtensionsSummary(_serialization.Model): # pylint: disable=name-too-long """Extensions summary for virtual machines of a virtual machine scale set. Variables are only populated by the server, and will be ignored when sending a request. @@ -15107,10 +14998,10 @@ def __init__(self, *, instance_ids: Optional[List[str]] = None, **kwargs: Any) - self.instance_ids = instance_ids -class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): +class VirtualMachineScaleSetVMInstanceRequiredIDs(_serialization.Model): # pylint: disable=name-too-long """Specifies a list of virtual machine instance IDs from the VM scale set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar instance_ids: The virtual machine scale set instance ids. Required. :vartype instance_ids: list[str] @@ -15174,7 +15065,7 @@ class VirtualMachineScaleSetVMInstanceView(_serialization.Model): # pylint: dis :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :vartype computer_name: str :ivar os_name: The Operating System running on the hybrid machine. :vartype os_name: str @@ -15259,7 +15150,7 @@ def __init__( :code:`
    `:code:`
    ` **Max-length (Windows):** 15 characters :code:`
    `:code:`
    ` **Max-length (Linux):** 64 characters. :code:`
    `:code:`
    ` For naming conventions and restrictions see `Azure infrastructure services implementation guidelines - `_. + `_. # pylint: disable=line-too-long :paramtype computer_name: str :keyword os_name: The Operating System running on the hybrid machine. :paramtype os_name: str @@ -15291,7 +15182,7 @@ def __init__( class VirtualMachineScaleSetVMListResult(_serialization.Model): """The List Virtual Machine Scale Set VMs operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: The list of virtual machine scale sets VMs. Required. :vartype value: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM] @@ -15324,7 +15215,7 @@ def __init__( self.next_link = next_link -class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): +class VirtualMachineScaleSetVMNetworkProfileConfiguration(_serialization.Model): # pylint: disable=name-too-long """Describes a virtual machine scale set VM network profile. :ivar network_interface_configurations: The list of network configurations. @@ -16199,14 +16090,14 @@ def __init__( class VMGalleryApplication(_serialization.Model): """Specifies the required information to reference a compute gallery application version. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar tags: Optional, Specifies a passthrough value for more generic context. :vartype tags: str :ivar order: Optional, Specifies the order in which the packages have to be installed. :vartype order: int :ivar package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :vartype package_reference_id: str :ivar configuration_reference: Optional, Specifies the uri to an azure blob that will replace @@ -16251,7 +16142,7 @@ def __init__( :paramtype order: int :keyword package_reference_id: Specifies the GalleryApplicationVersion resource id on the form of - /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. + /subscriptions/{SubscriptionId}/resourceGroups/{ResourceGroupName}/providers/Microsoft.Compute/galleries/{galleryName}/applications/{application}/versions/{version}. # pylint: disable=line-too-long Required. :paramtype package_reference_id: str :keyword configuration_reference: Optional, Specifies the uri to an azure blob that will @@ -16307,7 +16198,7 @@ def __init__( self.next_link = next_link -class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): +class VMScaleSetConvertToSinglePlacementGroupInput(_serialization.Model): # pylint: disable=name-too-long """VMScaleSetConvertToSinglePlacementGroupInput. :ivar active_placement_group_id: Id of the placement group in which you want future virtual @@ -16531,7 +16422,7 @@ def __init__( self.max_patch_publish_date = max_patch_publish_date -class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): +class WindowsVMGuestPatchAutomaticByPlatformSettings(_serialization.Model): # pylint: disable=name-too-long """Specifies additional settings to be applied when patch mode AutomaticByPlatform is selected in Windows patch settings. diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/operations/_operations.py index e6cdc9e2b392..fb07b81e6815 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_09_01/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -112,7 +112,7 @@ def build_virtual_machine_sizes_list_request(location: str, subscription_id: str return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_by_location_request( +def build_virtual_machine_scale_sets_list_by_location_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -142,7 +142,7 @@ def build_virtual_machine_scale_sets_list_by_location_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_create_or_update_request( +def build_virtual_machine_scale_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -186,7 +186,7 @@ def build_virtual_machine_scale_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_request( +def build_virtual_machine_scale_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -230,7 +230,7 @@ def build_virtual_machine_scale_sets_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_request( +def build_virtual_machine_scale_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -268,7 +268,7 @@ def build_virtual_machine_scale_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_request( +def build_virtual_machine_scale_sets_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -306,7 +306,7 @@ def build_virtual_machine_scale_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_deallocate_request( +def build_virtual_machine_scale_sets_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -347,7 +347,7 @@ def build_virtual_machine_scale_sets_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_delete_instances_request( +def build_virtual_machine_scale_sets_delete_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -388,7 +388,7 @@ def build_virtual_machine_scale_sets_delete_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_instance_view_request( +def build_virtual_machine_scale_sets_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -419,7 +419,7 @@ def build_virtual_machine_scale_sets_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_request( +def build_virtual_machine_scale_sets_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -449,7 +449,9 @@ def build_virtual_machine_scale_sets_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_scale_sets_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -475,7 +477,7 @@ def build_virtual_machine_scale_sets_list_all_request(subscription_id: str, **kw return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_list_skus_request( +def build_virtual_machine_scale_sets_list_skus_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -506,7 +508,7 @@ def build_virtual_machine_scale_sets_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_get_os_upgrade_history_request( +def build_virtual_machine_scale_sets_get_os_upgrade_history_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -537,7 +539,7 @@ def build_virtual_machine_scale_sets_get_os_upgrade_history_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_power_off_request( +def build_virtual_machine_scale_sets_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -578,7 +580,7 @@ def build_virtual_machine_scale_sets_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_restart_request( +def build_virtual_machine_scale_sets_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -612,7 +614,7 @@ def build_virtual_machine_scale_sets_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_start_request( +def build_virtual_machine_scale_sets_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -646,7 +648,7 @@ def build_virtual_machine_scale_sets_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reapply_request( +def build_virtual_machine_scale_sets_reapply_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -677,7 +679,7 @@ def build_virtual_machine_scale_sets_reapply_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_redeploy_request( +def build_virtual_machine_scale_sets_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -711,7 +713,7 @@ def build_virtual_machine_scale_sets_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_perform_maintenance_request( +def build_virtual_machine_scale_sets_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -745,7 +747,7 @@ def build_virtual_machine_scale_sets_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_update_instances_request( +def build_virtual_machine_scale_sets_update_instances_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -779,7 +781,7 @@ def build_virtual_machine_scale_sets_update_instances_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_request( +def build_virtual_machine_scale_sets_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -813,7 +815,7 @@ def build_virtual_machine_scale_sets_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_reimage_all_request( +def build_virtual_machine_scale_sets_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -847,7 +849,7 @@ def build_virtual_machine_scale_sets_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_approve_rolling_upgrade_request( +def build_virtual_machine_scale_sets_approve_rolling_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -881,7 +883,7 @@ def build_virtual_machine_scale_sets_approve_rolling_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( +def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, @@ -924,7 +926,7 @@ def build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_upda return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( +def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -958,7 +960,7 @@ def build_virtual_machine_scale_sets_convert_to_single_placement_group_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_sets_set_orchestration_service_state_request( +def build_virtual_machine_scale_sets_set_orchestration_service_state_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -992,7 +994,7 @@ def build_virtual_machine_scale_sets_set_orchestration_service_state_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_create_or_update_request( +def build_virtual_machine_scale_set_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1027,7 +1029,7 @@ def build_virtual_machine_scale_set_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_update_request( +def build_virtual_machine_scale_set_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1062,7 +1064,7 @@ def build_virtual_machine_scale_set_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_delete_request( +def build_virtual_machine_scale_set_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1094,7 +1096,7 @@ def build_virtual_machine_scale_set_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_get_request( +def build_virtual_machine_scale_set_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, @@ -1134,7 +1136,7 @@ def build_virtual_machine_scale_set_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_extensions_list_request( +def build_virtual_machine_scale_set_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1165,7 +1167,7 @@ def build_virtual_machine_scale_set_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( +def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1196,7 +1198,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_cancel_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1227,7 +1229,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( +def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1258,7 +1260,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_req return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( +def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1289,7 +1291,7 @@ def build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( +def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1330,7 +1332,7 @@ def build_virtual_machine_scale_set_vm_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_update_request( +def build_virtual_machine_scale_set_vm_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1371,7 +1373,7 @@ def build_virtual_machine_scale_set_vm_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_delete_request( +def build_virtual_machine_scale_set_vm_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1409,7 +1411,7 @@ def build_virtual_machine_scale_set_vm_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_get_request( +def build_virtual_machine_scale_set_vm_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1451,7 +1453,7 @@ def build_virtual_machine_scale_set_vm_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_extensions_list_request( +def build_virtual_machine_scale_set_vm_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1491,7 +1493,7 @@ def build_virtual_machine_scale_set_vm_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_request( +def build_virtual_machine_scale_set_vms_reimage_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1526,7 +1528,7 @@ def build_virtual_machine_scale_set_vms_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_reimage_all_request( +def build_virtual_machine_scale_set_vms_reimage_all_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1558,7 +1560,7 @@ def build_virtual_machine_scale_set_vms_reimage_all_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( +def build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1590,7 +1592,7 @@ def build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_deallocate_request( +def build_virtual_machine_scale_set_vms_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1622,7 +1624,7 @@ def build_virtual_machine_scale_set_vms_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_update_request( +def build_virtual_machine_scale_set_vms_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1668,7 +1670,7 @@ def build_virtual_machine_scale_set_vms_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_delete_request( +def build_virtual_machine_scale_set_vms_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1708,7 +1710,7 @@ def build_virtual_machine_scale_set_vms_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_request( +def build_virtual_machine_scale_set_vms_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1748,7 +1750,7 @@ def build_virtual_machine_scale_set_vms_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_get_instance_view_request( +def build_virtual_machine_scale_set_vms_get_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1780,7 +1782,7 @@ def build_virtual_machine_scale_set_vms_get_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_list_request( +def build_virtual_machine_scale_set_vms_list_request( # pylint: disable=name-too-long resource_group_name: str, virtual_machine_scale_set_name: str, subscription_id: str, @@ -1826,7 +1828,7 @@ def build_virtual_machine_scale_set_vms_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_power_off_request( +def build_virtual_machine_scale_set_vms_power_off_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -1866,7 +1868,7 @@ def build_virtual_machine_scale_set_vms_power_off_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_restart_request( +def build_virtual_machine_scale_set_vms_restart_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1898,7 +1900,7 @@ def build_virtual_machine_scale_set_vms_restart_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_start_request( +def build_virtual_machine_scale_set_vms_start_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1930,7 +1932,7 @@ def build_virtual_machine_scale_set_vms_start_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_redeploy_request( +def build_virtual_machine_scale_set_vms_redeploy_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1962,7 +1964,7 @@ def build_virtual_machine_scale_set_vms_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( +def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -2004,7 +2006,7 @@ def build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_perform_maintenance_request( +def build_virtual_machine_scale_set_vms_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2036,7 +2038,7 @@ def build_virtual_machine_scale_set_vms_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_simulate_eviction_request( +def build_virtual_machine_scale_set_vms_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2068,7 +2070,7 @@ def build_virtual_machine_scale_set_vms_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( +def build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2103,7 +2105,7 @@ def build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vms_run_command_request( +def build_virtual_machine_scale_set_vms_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2138,7 +2140,7 @@ def build_virtual_machine_scale_set_vms_run_command_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_create_or_update_request( +def build_virtual_machine_extensions_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2173,7 +2175,7 @@ def build_virtual_machine_extensions_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_update_request( +def build_virtual_machine_extensions_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2208,7 +2210,7 @@ def build_virtual_machine_extensions_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_delete_request( +def build_virtual_machine_extensions_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2240,7 +2242,7 @@ def build_virtual_machine_extensions_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_get_request( +def build_virtual_machine_extensions_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, vm_extension_name: str, @@ -2280,7 +2282,7 @@ def build_virtual_machine_extensions_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extensions_list_request( +def build_virtual_machine_extensions_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2313,7 +2315,9 @@ def build_virtual_machine_extensions_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_by_location_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machines_list_by_location_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -2375,7 +2379,7 @@ def build_virtual_machines_capture_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_create_or_update_request( +def build_virtual_machines_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -2539,7 +2543,7 @@ def build_virtual_machines_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_instance_view_request( +def build_virtual_machines_instance_view_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2570,7 +2574,7 @@ def build_virtual_machines_instance_view_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_convert_to_managed_disks_request( +def build_virtual_machines_convert_to_managed_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2601,7 +2605,7 @@ def build_virtual_machines_convert_to_managed_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_deallocate_request( +def build_virtual_machines_deallocate_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, hibernate: Optional[bool] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2634,7 +2638,7 @@ def build_virtual_machines_deallocate_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_generalize_request( +def build_virtual_machines_generalize_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2741,7 +2745,7 @@ def build_virtual_machines_list_all_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_list_available_sizes_request( +def build_virtual_machines_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -2963,7 +2967,7 @@ def build_virtual_machines_reimage_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_retrieve_boot_diagnostics_data_request( +def build_virtual_machines_retrieve_boot_diagnostics_data_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, @@ -3003,7 +3007,7 @@ def build_virtual_machines_retrieve_boot_diagnostics_data_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_perform_maintenance_request( +def build_virtual_machines_perform_maintenance_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3034,7 +3038,7 @@ def build_virtual_machines_perform_maintenance_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_simulate_eviction_request( +def build_virtual_machines_simulate_eviction_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3065,7 +3069,7 @@ def build_virtual_machines_simulate_eviction_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_assess_patches_request( +def build_virtual_machines_assess_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3096,7 +3100,7 @@ def build_virtual_machines_assess_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_install_patches_request( +def build_virtual_machines_install_patches_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3130,7 +3134,7 @@ def build_virtual_machines_install_patches_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_attach_detach_data_disks_request( +def build_virtual_machines_attach_detach_data_disks_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3164,7 +3168,7 @@ def build_virtual_machines_attach_detach_data_disks_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machines_run_command_request( +def build_virtual_machines_run_command_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3232,7 +3236,7 @@ def build_virtual_machine_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_request( +def build_virtual_machine_images_list_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, @@ -3280,7 +3284,7 @@ def build_virtual_machine_images_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_offers_request( +def build_virtual_machine_images_list_offers_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3311,7 +3315,7 @@ def build_virtual_machine_images_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_publishers_request( +def build_virtual_machine_images_list_publishers_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3340,7 +3344,7 @@ def build_virtual_machine_images_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_skus_request( +def build_virtual_machine_images_list_skus_request( # pylint: disable=name-too-long location: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3372,7 +3376,7 @@ def build_virtual_machine_images_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_list_by_edge_zone_request( +def build_virtual_machine_images_list_by_edge_zone_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3403,7 +3407,7 @@ def build_virtual_machine_images_list_by_edge_zone_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_get_request( +def build_virtual_machine_images_edge_zone_get_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3445,7 +3449,7 @@ def build_virtual_machine_images_edge_zone_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_request( +def build_virtual_machine_images_edge_zone_list_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, @@ -3495,7 +3499,7 @@ def build_virtual_machine_images_edge_zone_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_offers_request( +def build_virtual_machine_images_edge_zone_list_offers_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3527,7 +3531,7 @@ def build_virtual_machine_images_edge_zone_list_offers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_publishers_request( +def build_virtual_machine_images_edge_zone_list_publishers_request( # pylint: disable=name-too-long location: str, edge_zone: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3558,7 +3562,7 @@ def build_virtual_machine_images_edge_zone_list_publishers_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_images_edge_zone_list_skus_request( +def build_virtual_machine_images_edge_zone_list_skus_request( # pylint: disable=name-too-long location: str, edge_zone: str, publisher_name: str, offer: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3591,7 +3595,7 @@ def build_virtual_machine_images_edge_zone_list_skus_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_get_request( +def build_virtual_machine_extension_images_get_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, version: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3624,7 +3628,7 @@ def build_virtual_machine_extension_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_types_request( +def build_virtual_machine_extension_images_list_types_request( # pylint: disable=name-too-long location: str, publisher_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3655,7 +3659,7 @@ def build_virtual_machine_extension_images_list_types_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_extension_images_list_versions_request( +def build_virtual_machine_extension_images_list_versions_request( # pylint: disable=name-too-long location: str, publisher_name: str, type: str, @@ -3701,7 +3705,7 @@ def build_virtual_machine_extension_images_list_versions_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_create_or_update_request( +def build_availability_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3831,7 +3835,7 @@ def build_availability_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_by_subscription_request( +def build_availability_sets_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3887,7 +3891,7 @@ def build_availability_sets_list_request(resource_group_name: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_availability_sets_list_available_sizes_request( +def build_availability_sets_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, availability_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3918,7 +3922,7 @@ def build_availability_sets_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_create_or_update_request( +def build_proximity_placement_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3954,7 +3958,7 @@ def build_proximity_placement_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_update_request( +def build_proximity_placement_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -3990,7 +3994,7 @@ def build_proximity_placement_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_delete_request( +def build_proximity_placement_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4023,7 +4027,7 @@ def build_proximity_placement_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_get_request( +def build_proximity_placement_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, proximity_placement_group_name: str, subscription_id: str, @@ -4065,7 +4069,9 @@ def build_proximity_placement_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_proximity_placement_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4091,7 +4097,7 @@ def build_proximity_placement_groups_list_by_subscription_request(subscription_i return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_proximity_placement_groups_list_by_resource_group_request( +def build_proximity_placement_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4121,7 +4127,7 @@ def build_proximity_placement_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_create_or_update_request( +def build_dedicated_host_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4155,7 +4161,7 @@ def build_dedicated_host_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_update_request( +def build_dedicated_host_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4189,7 +4195,7 @@ def build_dedicated_host_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_delete_request( +def build_dedicated_host_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4258,7 +4264,7 @@ def build_dedicated_host_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_resource_group_request( +def build_dedicated_host_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4288,7 +4294,9 @@ def build_dedicated_host_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_host_groups_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_dedicated_host_groups_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4312,7 +4320,7 @@ def build_dedicated_host_groups_list_by_subscription_request(subscription_id: st return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_create_or_update_request( +def build_dedicated_hosts_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4454,7 +4462,7 @@ def build_dedicated_hosts_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_by_host_group_request( +def build_dedicated_hosts_list_by_host_group_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4549,7 +4557,7 @@ def build_dedicated_hosts_redeploy_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_dedicated_hosts_list_available_sizes_request( +def build_dedicated_hosts_list_available_sizes_request( # pylint: disable=name-too-long resource_group_name: str, host_group_name: str, host_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4581,7 +4589,9 @@ def build_dedicated_hosts_list_available_sizes_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_ssh_public_keys_list_by_subscription_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -4605,7 +4615,7 @@ def build_ssh_public_keys_list_by_subscription_request(subscription_id: str, **k return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_list_by_resource_group_request( +def build_ssh_public_keys_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4765,7 +4775,7 @@ def build_ssh_public_keys_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_ssh_public_keys_generate_key_pair_request( +def build_ssh_public_keys_generate_key_pair_request( # pylint: disable=name-too-long resource_group_name: str, ssh_public_key_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4931,7 +4941,7 @@ def build_images_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_images_list_by_resource_group_request( +def build_images_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -4985,7 +4995,7 @@ def build_images_list_request(subscription_id: str, **kwargs: Any) -> HttpReques return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_create_or_update_request( +def build_restore_point_collections_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5021,7 +5031,7 @@ def build_restore_point_collections_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_update_request( +def build_restore_point_collections_update_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5057,7 +5067,7 @@ def build_restore_point_collections_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_delete_request( +def build_restore_point_collections_delete_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5090,7 +5100,7 @@ def build_restore_point_collections_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_get_request( +def build_restore_point_collections_get_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, subscription_id: str, @@ -5130,7 +5140,7 @@ def build_restore_point_collections_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_request( +def build_restore_point_collections_list_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5160,7 +5170,9 @@ def build_restore_point_collections_list_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_restore_point_collections_list_all_request(subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_restore_point_collections_list_all_request( # pylint: disable=name-too-long + subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5307,7 +5319,7 @@ def build_restore_points_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_create_or_update_request( +def build_capacity_reservation_groups_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5343,7 +5355,7 @@ def build_capacity_reservation_groups_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_update_request( +def build_capacity_reservation_groups_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5379,7 +5391,7 @@ def build_capacity_reservation_groups_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_delete_request( +def build_capacity_reservation_groups_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5412,7 +5424,7 @@ def build_capacity_reservation_groups_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_get_request( +def build_capacity_reservation_groups_get_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, @@ -5452,7 +5464,7 @@ def build_capacity_reservation_groups_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_resource_group_request( +def build_capacity_reservation_groups_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, *, @@ -5488,7 +5500,7 @@ def build_capacity_reservation_groups_list_by_resource_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservation_groups_list_by_subscription_request( +def build_capacity_reservation_groups_list_by_subscription_request( # pylint: disable=name-too-long subscription_id: str, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, @@ -5521,7 +5533,7 @@ def build_capacity_reservation_groups_list_by_subscription_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_create_or_update_request( +def build_capacity_reservations_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5562,7 +5574,7 @@ def build_capacity_reservations_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_update_request( +def build_capacity_reservations_update_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5603,7 +5615,7 @@ def build_capacity_reservations_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_delete_request( +def build_capacity_reservations_delete_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, @@ -5683,7 +5695,7 @@ def build_capacity_reservations_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_capacity_reservations_list_by_capacity_reservation_group_request( +def build_capacity_reservations_list_by_capacity_reservation_group_request( # pylint: disable=name-too-long resource_group_name: str, capacity_reservation_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5716,7 +5728,7 @@ def build_capacity_reservations_list_by_capacity_reservation_group_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_request_rate_by_interval_request( +def build_log_analytics_export_request_rate_by_interval_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5749,7 +5761,7 @@ def build_log_analytics_export_request_rate_by_interval_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_log_analytics_export_throttled_requests_request( +def build_log_analytics_export_throttled_requests_request( # pylint: disable=name-too-long location: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5782,7 +5794,9 @@ def build_log_analytics_export_throttled_requests_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_request(location: str, subscription_id: str, **kwargs: Any) -> HttpRequest: +def build_virtual_machine_run_commands_list_request( # pylint: disable=name-too-long + location: str, subscription_id: str, **kwargs: Any +) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) _params = case_insensitive_dict(kwargs.pop("params", {}) or {}) @@ -5809,7 +5823,7 @@ def build_virtual_machine_run_commands_list_request(location: str, subscription_ return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_request( +def build_virtual_machine_run_commands_get_request( # pylint: disable=name-too-long location: str, command_id: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5840,7 +5854,7 @@ def build_virtual_machine_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_create_or_update_request( +def build_virtual_machine_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5875,7 +5889,7 @@ def build_virtual_machine_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_update_request( +def build_virtual_machine_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5910,7 +5924,7 @@ def build_virtual_machine_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_delete_request( +def build_virtual_machine_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -5942,7 +5956,7 @@ def build_virtual_machine_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_get_by_virtual_machine_request( +def build_virtual_machine_run_commands_get_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, run_command_name: str, @@ -5982,7 +5996,7 @@ def build_virtual_machine_run_commands_get_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_run_commands_list_by_virtual_machine_request( +def build_virtual_machine_run_commands_list_by_virtual_machine_request( # pylint: disable=name-too-long resource_group_name: str, vm_name: str, subscription_id: str, *, expand: Optional[str] = None, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -6015,7 +6029,7 @@ def build_virtual_machine_run_commands_list_by_virtual_machine_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( +def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6056,7 +6070,7 @@ def build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_update_request( +def build_virtual_machine_scale_set_vm_run_commands_update_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6097,7 +6111,7 @@ def build_virtual_machine_scale_set_vm_run_commands_update_request( return HttpRequest(method="PATCH", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_delete_request( +def build_virtual_machine_scale_set_vm_run_commands_delete_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6135,7 +6149,7 @@ def build_virtual_machine_scale_set_vm_run_commands_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_get_request( +def build_virtual_machine_scale_set_vm_run_commands_get_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6177,7 +6191,7 @@ def build_virtual_machine_scale_set_vm_run_commands_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_virtual_machine_scale_set_vm_run_commands_list_request( +def build_virtual_machine_scale_set_vm_run_commands_list_request( # pylint: disable=name-too-long resource_group_name: str, vm_scale_set_name: str, instance_id: str, @@ -6241,7 +6255,6 @@ def __init__(self, *args, **kwargs): def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: """Gets a list of compute operations. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ComputeOperationValue or the result of cls(response) :rtype: @@ -6265,14 +6278,13 @@ def list(self, **kwargs: Any) -> Iterable["_models.ComputeOperationValue"]: def prepare_request(next_link=None): if not next_link: - request = build_operations_list_request( + _request = build_operations_list_request( api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6283,14 +6295,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ComputeOperationListResult", pipeline_response) @@ -6300,11 +6312,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6316,8 +6328,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/providers/Microsoft.Compute/operations"} - class UsageOperations: """ @@ -6346,7 +6356,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: :param location: The location for which resource usage is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Usage or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.Usage] :raises ~azure.core.exceptions.HttpResponseError: @@ -6368,16 +6377,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.Usage"]: def prepare_request(next_link=None): if not next_link: - request = build_usage_list_request( + _request = build_usage_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6388,14 +6396,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ListUsagesResult", pipeline_response) @@ -6405,11 +6413,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6421,8 +6429,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/usages"} - class VirtualMachineSizesOperations: """ @@ -6451,7 +6457,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -6473,16 +6478,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.VirtualMachine def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_sizes_list_request( + _request = build_virtual_machine_sizes_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6493,14 +6497,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -6510,11 +6514,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6526,8 +6530,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/vmSizes"} - class VirtualMachineScaleSetsOperations: # pylint: disable=too-many-public-methods """ @@ -6556,7 +6558,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which VM scale sets under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6580,16 +6581,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_by_location_request( + _request = build_virtual_machine_scale_sets_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6600,14 +6600,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -6617,11 +6617,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6633,15 +6633,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachineScaleSets" - } - def _create_or_update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -6670,7 +6666,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSet") - request = build_virtual_machine_scale_sets_create_or_update_request( + _request = build_virtual_machine_scale_sets_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6680,16 +6676,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6709,10 +6704,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } - @overload def begin_create_or_update( self, @@ -6744,14 +6735,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6764,7 +6747,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -6778,7 +6761,7 @@ def begin_create_or_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -6790,14 +6773,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6810,7 +6785,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSet, IO], + parameters: Union[_models.VirtualMachineScaleSet, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -6822,9 +6797,9 @@ def begin_create_or_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet or IO + :param parameters: The scale set object. Is either a VirtualMachineScaleSet type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -6833,17 +6808,6 @@ def begin_create_or_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -6878,7 +6842,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6888,23 +6852,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -6933,7 +6895,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetUpdate") - request = build_virtual_machine_scale_sets_update_request( + _request = build_virtual_machine_scale_sets_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -6943,16 +6905,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6964,13 +6925,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -7003,14 +6960,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7023,7 +6972,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -7037,7 +6986,7 @@ def begin_update( :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str :param parameters: The scale set object. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -7049,14 +6998,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7069,7 +7010,7 @@ def begin_update( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VirtualMachineScaleSetUpdate, IO], + parameters: Union[_models.VirtualMachineScaleSetUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -7081,9 +7022,10 @@ def begin_update( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set to create or update. Required. :type vm_scale_set_name: str - :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a IO - type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetUpdate or IO + :param parameters: The scale set object. Is either a VirtualMachineScaleSetUpdate type or a + IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetUpdate or + IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -7092,17 +7034,6 @@ def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7137,7 +7068,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -7147,17 +7078,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[_models.VirtualMachineScaleSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7176,22 +7105,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_delete_request( + _request = build_virtual_machine_scale_sets_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7201,11 +7129,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -7220,14 +7144,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7255,7 +7171,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7264,17 +7180,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -7295,7 +7207,6 @@ def get( UserData property of the VM scale set that was provided by the user during the VM scale set Create/Update operation. "userData" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetVMScaleSets - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSet :raises ~azure.core.exceptions.HttpResponseError: @@ -7314,22 +7225,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSet] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_request( + _request = build_virtual_machine_scale_sets_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7341,19 +7251,15 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}" - } + return deserialized # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7384,7 +7290,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_deallocate_request( + _request = build_virtual_machine_scale_sets_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7393,16 +7299,15 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7412,11 +7317,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_deallocate( @@ -7448,14 +7349,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7466,7 +7359,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, hibernate: Optional[bool] = None, content_type: str = "application/json", @@ -7482,7 +7375,7 @@ def begin_deallocate( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. @@ -7490,14 +7383,6 @@ def begin_deallocate( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7508,7 +7393,7 @@ def begin_deallocate( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -7522,24 +7407,13 @@ def begin_deallocate( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword hibernate: Optional parameter to hibernate a virtual machine from the VM scale set. (This feature is available for VMSS with Flexible OrchestrationMode only). Default value is None. :paramtype hibernate: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7570,7 +7444,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7579,23 +7453,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _delete_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7623,7 +7493,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_delete_instances_request( + _request = build_virtual_machine_scale_sets_delete_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -7632,16 +7502,15 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._delete_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7651,11 +7520,7 @@ def _delete_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_delete_instances( @@ -7684,14 +7549,6 @@ def begin_delete_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7702,7 +7559,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, force_deletion: Optional[bool] = None, content_type: str = "application/json", @@ -7716,21 +7573,13 @@ def begin_delete_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7741,7 +7590,7 @@ def begin_delete_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -7753,23 +7602,12 @@ def begin_delete_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :keyword force_deletion: Optional parameter to force delete virtual machines from the VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -7800,7 +7638,7 @@ def begin_delete_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -7809,17 +7647,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_instance_view( @@ -7831,7 +7665,6 @@ def get_instance_view( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -7850,21 +7683,20 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_get_instance_view_request( + _request = build_virtual_machine_scale_sets_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7876,13 +7708,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: @@ -7890,7 +7718,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -7914,16 +7741,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Vir def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_request( + _request = build_virtual_machine_scale_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -7934,14 +7760,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response) @@ -7951,11 +7777,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -7967,17 +7793,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this till nextLink is null to fetch all the VM Scale Sets. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSet or the result of cls(response) :rtype: @@ -8001,15 +7822,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.VirtualMachineScaleSet"]: def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_all_request( + _request = build_virtual_machine_scale_sets_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8020,14 +7840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response) @@ -8037,11 +7857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8053,8 +7873,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"} - @distributed_trace def list_skus( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8066,7 +7884,6 @@ def list_skus( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetSku or the result of cls(response) :rtype: @@ -8090,17 +7907,16 @@ def list_skus( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_list_skus_request( + _request = build_virtual_machine_scale_sets_list_skus_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8111,14 +7927,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response) @@ -8128,11 +7944,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8144,10 +7960,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus" - } - @distributed_trace def get_os_upgrade_history( self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8158,7 +7970,6 @@ def get_os_upgrade_history( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either UpgradeOperationHistoricalStatusInfo or the result of cls(response) :rtype: @@ -8182,17 +7993,16 @@ def get_os_upgrade_history( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( + _request = build_virtual_machine_scale_sets_get_os_upgrade_history_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_os_upgrade_history.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -8203,14 +8013,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response) @@ -8220,11 +8030,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8236,15 +8046,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - get_os_upgrade_history.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8275,7 +8081,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_power_off_request( + _request = build_virtual_machine_scale_sets_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8284,16 +8090,15 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8303,11 +8108,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_power_off( @@ -8339,14 +8140,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8357,7 +8150,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, skip_shutdown: bool = False, content_type: str = "application/json", @@ -8373,7 +8166,7 @@ def begin_power_off( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. @@ -8381,14 +8174,6 @@ def begin_power_off( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8399,7 +8184,7 @@ def begin_power_off( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, *, skip_shutdown: bool = False, **kwargs: Any @@ -8413,24 +8198,13 @@ def begin_power_off( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :keyword skip_shutdown: The parameter to request non-graceful VM shutdown. True value for this flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8461,7 +8235,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8470,23 +8244,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8515,7 +8285,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_restart_request( + _request = build_virtual_machine_scale_sets_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8523,16 +8293,15 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8542,11 +8311,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_restart( @@ -8571,14 +8336,6 @@ def begin_restart( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8589,7 +8346,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8602,18 +8359,10 @@ def begin_restart( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8624,7 +8373,7 @@ def begin_restart( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Restarts one or more virtual machines in a VM scale set. @@ -8634,20 +8383,9 @@ def begin_restart( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8677,7 +8415,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8686,23 +8424,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -8731,7 +8465,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_start_request( + _request = build_virtual_machine_scale_sets_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -8739,16 +8473,15 @@ def _start_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8758,11 +8491,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_start( @@ -8787,14 +8516,6 @@ def begin_start( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8805,7 +8526,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -8818,18 +8539,10 @@ def begin_start( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -8840,7 +8553,7 @@ def begin_start( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Starts one or more virtual machines in a VM scale set. @@ -8850,20 +8563,9 @@ def begin_start( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -8893,7 +8595,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -8902,17 +8604,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -8931,21 +8629,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_reapply_request( + _request = build_virtual_machine_scale_sets_reapply_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -8959,11 +8656,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -8974,14 +8667,6 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9008,7 +8693,7 @@ def begin_reapply(self, resource_group_name: str, vm_scale_set_name: str, **kwar def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -9019,23 +8704,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9064,7 +8745,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_redeploy_request( + _request = build_virtual_machine_scale_sets_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9072,16 +8753,15 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9091,11 +8771,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_redeploy( @@ -9121,14 +8797,6 @@ def begin_redeploy( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9139,7 +8807,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9153,18 +8821,10 @@ def begin_redeploy( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9175,7 +8835,7 @@ def begin_redeploy( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node, @@ -9186,20 +8846,9 @@ def begin_redeploy( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9229,7 +8878,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9238,23 +8887,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9283,7 +8928,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme else: _json = None - request = build_virtual_machine_scale_sets_perform_maintenance_request( + _request = build_virtual_machine_scale_sets_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9291,16 +8936,15 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme content_type=content_type, json=_json, content=_content, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9310,11 +8954,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_perform_maintenance( @@ -9342,14 +8982,6 @@ def begin_perform_maintenance( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9360,7 +8992,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9376,18 +9008,10 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9398,7 +9022,7 @@ def begin_perform_maintenance( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances @@ -9411,20 +9035,9 @@ def begin_perform_maintenance( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9454,7 +9067,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9463,23 +9076,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_instances_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -9505,7 +9114,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements else: _json = self._serialize.body(vm_instance_i_ds, "VirtualMachineScaleSetVMInstanceRequiredIDs") - request = build_virtual_machine_scale_sets_update_instances_request( + _request = build_virtual_machine_scale_sets_update_instances_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9513,16 +9122,15 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._update_instances_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9532,11 +9140,7 @@ def _update_instances_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _update_instances_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_update_instances( @@ -9561,14 +9165,6 @@ def begin_update_instances( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9579,7 +9175,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: IO, + vm_instance_i_ds: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -9592,18 +9188,10 @@ def begin_update_instances( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Required. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9614,7 +9202,7 @@ def begin_update_instances( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO], + vm_instance_i_ds: Union[_models.VirtualMachineScaleSetVMInstanceRequiredIDs, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model. @@ -9624,20 +9212,9 @@ def begin_update_instances( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO type. Required. + either a VirtualMachineScaleSetVMInstanceRequiredIDs type or a IO[bytes] type. Required. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9667,7 +9244,7 @@ def begin_update_instances( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9676,23 +9253,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_instances.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9721,7 +9294,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_request( + _request = build_virtual_machine_scale_sets_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9729,16 +9302,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9748,11 +9320,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -9778,14 +9346,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9796,7 +9356,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[IO] = None, + vm_scale_set_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -9810,18 +9370,10 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None. - :type vm_scale_set_reimage_input: IO + :type vm_scale_set_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -9832,7 +9384,7 @@ def begin_reimage( self, resource_group_name: str, vm_scale_set_name: str, - vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO]] = None, + vm_scale_set_reimage_input: Optional[Union[_models.VirtualMachineScaleSetReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which @@ -9844,20 +9396,9 @@ def begin_reimage( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Is either a - VirtualMachineScaleSetReimageParameters type or a IO type. Default value is None. + VirtualMachineScaleSetReimageParameters type or a IO[bytes] type. Default value is None. :type vm_scale_set_reimage_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -9887,7 +9428,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -9896,23 +9437,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -9941,7 +9478,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_sets_reimage_all_request( + _request = build_virtual_machine_scale_sets_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -9949,16 +9486,15 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -9968,11 +9504,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage_all( @@ -9998,14 +9530,6 @@ def begin_reimage_all( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10016,7 +9540,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10030,18 +9554,10 @@ def begin_reimage_all( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10052,7 +9568,7 @@ def begin_reimage_all( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This @@ -10063,20 +9579,9 @@ def begin_reimage_all( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10106,7 +9611,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10115,23 +9620,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -10160,7 +9661,7 @@ def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-sta else: _json = None - request = build_virtual_machine_scale_sets_approve_rolling_upgrade_request( + _request = build_virtual_machine_scale_sets_approve_rolling_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10168,16 +9669,15 @@ def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-sta content_type=content_type, json=_json, content=_content, - template_url=self._approve_rolling_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10190,11 +9690,7 @@ def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-sta response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _approve_rolling_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/approveRollingUpgrade" - } + return cls(pipeline_response, None, response_headers) # type: ignore @overload def begin_approve_rolling_upgrade( @@ -10220,14 +9716,6 @@ def begin_approve_rolling_upgrade( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10238,7 +9726,7 @@ def begin_approve_rolling_upgrade( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[IO] = None, + vm_instance_i_ds: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -10252,18 +9740,10 @@ def begin_approve_rolling_upgrade( :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default value is None. - :type vm_instance_i_ds: IO + :type vm_instance_i_ds: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10274,7 +9754,7 @@ def begin_approve_rolling_upgrade( self, resource_group_name: str, vm_scale_set_name: str, - vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO]] = None, + vm_instance_i_ds: Optional[Union[_models.VirtualMachineScaleSetVMInstanceIDs, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Approve upgrade on deferred rolling upgrades for OS disks in the virtual machines in a VM scale @@ -10285,20 +9765,9 @@ def begin_approve_rolling_upgrade( :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Is - either a VirtualMachineScaleSetVMInstanceIDs type or a IO type. Default value is None. + either a VirtualMachineScaleSetVMInstanceIDs type or a IO[bytes] type. Default value is None. :type vm_instance_i_ds: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceIDs or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10328,7 +9797,7 @@ def begin_approve_rolling_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10337,20 +9806,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_approve_rolling_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/approveRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace - def force_recovery_service_fabric_platform_update_domain_walk( + def force_recovery_service_fabric_platform_update_domain_walk( # pylint: disable=name-too-long self, resource_group_name: str, vm_scale_set_name: str, @@ -10376,7 +9841,6 @@ def force_recovery_service_fabric_platform_update_domain_walk( :keyword placement_group_id: The placement group id for which the manual recovery walk is requested. Default value is None. :paramtype placement_group_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RecoveryWalkResponse or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RecoveryWalkResponse :raises ~azure.core.exceptions.HttpResponseError: @@ -10395,7 +9859,7 @@ def force_recovery_service_fabric_platform_update_domain_walk( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RecoveryWalkResponse] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( + _request = build_virtual_machine_scale_sets_force_recovery_service_fabric_platform_update_domain_walk_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10403,16 +9867,15 @@ def force_recovery_service_fabric_platform_update_domain_walk( zone=zone, placement_group_id=placement_group_id, api_version=api_version, - template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10424,13 +9887,9 @@ def force_recovery_service_fabric_platform_update_domain_walk( deserialized = self._deserialize("RecoveryWalkResponse", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - force_recovery_service_fabric_platform_update_domain_walk.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk" - } + return deserialized # type: ignore @overload def convert_to_single_placement_group( # pylint: disable=inconsistent-return-statements @@ -10455,7 +9914,6 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10466,7 +9924,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10479,11 +9937,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10494,7 +9951,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO], + parameters: Union[_models.VMScaleSetConvertToSinglePlacementGroupInput, IO[bytes]], **kwargs: Any ) -> None: """Converts SinglePlacementGroup property to false for a existing virtual machine scale set. @@ -10505,13 +9962,10 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st Required. :type vm_scale_set_name: str :param parameters: The input object for ConvertToSinglePlacementGroup API. Is either a - VMScaleSetConvertToSinglePlacementGroupInput type or a IO type. Required. + VMScaleSetConvertToSinglePlacementGroupInput type or a IO[bytes] type. Required. :type parameters: - ~azure.mgmt.compute.v2023_09_01.models.VMScaleSetConvertToSinglePlacementGroupInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + ~azure.mgmt.compute.v2023_09_01.models.VMScaleSetConvertToSinglePlacementGroupInput or + IO[bytes] :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -10539,7 +9993,7 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st else: _json = self._serialize.body(parameters, "VMScaleSetConvertToSinglePlacementGroupInput") - request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( + _request = build_virtual_machine_scale_sets_convert_to_single_placement_group_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10547,16 +10001,15 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st content_type=content_type, json=_json, content=_content, - template_url=self.convert_to_single_placement_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10566,17 +10019,13 @@ def convert_to_single_placement_group( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - convert_to_single_placement_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/convertToSinglePlacementGroup" - } + return cls(pipeline_response, None, {}) # type: ignore def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> None: error_map = { @@ -10602,7 +10051,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re else: _json = self._serialize.body(parameters, "OrchestrationServiceStateInput") - request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( + _request = build_virtual_machine_scale_sets_set_orchestration_service_state_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, @@ -10610,16 +10059,15 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re content_type=content_type, json=_json, content=_content, - template_url=self._set_orchestration_service_state_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10629,11 +10077,7 @@ def _set_orchestration_service_state_initial( # pylint: disable=inconsistent-re raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _set_orchestration_service_state_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_set_orchestration_service_state( @@ -10657,14 +10101,6 @@ def begin_set_orchestration_service_state( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10675,7 +10111,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10688,18 +10124,10 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -10710,7 +10138,7 @@ def begin_set_orchestration_service_state( self, resource_group_name: str, vm_scale_set_name: str, - parameters: Union[_models.OrchestrationServiceStateInput, IO], + parameters: Union[_models.OrchestrationServiceStateInput, IO[bytes]], **kwargs: Any ) -> LROPoller[None]: """Changes ServiceState property for a given service. @@ -10721,19 +10149,9 @@ def begin_set_orchestration_service_state( Required. :type vm_scale_set_name: str :param parameters: The input object for SetOrchestrationServiceState API. Is either a - OrchestrationServiceStateInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.OrchestrationServiceStateInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + OrchestrationServiceStateInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.OrchestrationServiceStateInput or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -10763,7 +10181,7 @@ def begin_set_orchestration_service_state( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -10772,20 +10190,16 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_set_orchestration_service_state.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/setOrchestrationServiceState" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore -class VirtualMachineScaleSetExtensionsOperations: +class VirtualMachineScaleSetExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -10810,7 +10224,7 @@ def _create_or_update_initial( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -10836,7 +10250,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtension") - request = build_virtual_machine_scale_set_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -10845,16 +10259,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -10874,10 +10287,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_create_or_update( self, @@ -10905,14 +10314,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10926,7 +10327,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -10942,18 +10343,10 @@ def begin_create_or_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -10967,7 +10360,7 @@ def begin_create_or_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to create or update an extension. @@ -10980,20 +10373,9 @@ def begin_create_or_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Create VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11027,7 +10409,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11037,24 +10419,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetExtension: error_map = { @@ -11080,7 +10460,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetExtensionUpdate") - request = build_virtual_machine_scale_set_extensions_update_request( + _request = build_virtual_machine_scale_set_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, @@ -11089,16 +10469,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11118,10 +10497,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } - @overload def begin_update( self, @@ -11149,14 +10524,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11170,7 +10537,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -11186,18 +10553,10 @@ def begin_update( :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11211,7 +10570,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetExtension]: """The operation to update an extension. @@ -11224,20 +10583,10 @@ def begin_update( :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str :param extension_parameters: Parameters supplied to the Update VM scale set Extension - operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11271,7 +10620,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -11281,17 +10630,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, vmss_extension_name: str, **kwargs: Any @@ -11310,22 +10657,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_delete_request( + _request = build_virtual_machine_scale_set_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11335,11 +10681,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -11354,14 +10696,6 @@ def begin_delete( :type vm_scale_set_name: str :param vmss_extension_name: The name of the VM scale set extension. Required. :type vmss_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11389,7 +10723,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11398,17 +10732,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -11430,7 +10760,6 @@ def get( :type vmss_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -11449,23 +10778,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_extensions_get_request( + _request = build_virtual_machine_scale_set_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, vmss_extension_name=vmss_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11477,13 +10805,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions/{vmssExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -11495,7 +10819,6 @@ def list( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set containing the extension. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetExtension or the result of cls(response) :rtype: @@ -11519,17 +10842,16 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_extensions_list_request( + _request = build_virtual_machine_scale_set_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -11540,14 +10862,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetExtensionListResult", pipeline_response) @@ -11557,11 +10879,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11573,12 +10895,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensions" - } - -class VirtualMachineScaleSetRollingUpgradesOperations: +class VirtualMachineScaleSetRollingUpgradesOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -11615,21 +10933,20 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_cancel_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._cancel_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11639,11 +10956,7 @@ def _cancel_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _cancel_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -11653,14 +10966,6 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11687,7 +10992,7 @@ def begin_cancel(self, resource_group_name: str, vm_scale_set_name: str, **kwarg def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11696,17 +11001,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_cancel.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/cancel" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11725,21 +11026,20 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_os_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_os_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11749,11 +11049,7 @@ def _start_os_upgrade_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_os_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_os_upgrade( @@ -11767,14 +11063,6 @@ def begin_start_os_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11801,7 +11089,7 @@ def begin_start_os_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11810,17 +11098,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_os_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, **kwargs: Any @@ -11839,21 +11123,20 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_start_extension_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_extension_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11863,11 +11146,7 @@ def _start_extension_upgrade_initial( # pylint: disable=inconsistent-return-sta raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_extension_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start_extension_upgrade( @@ -11881,14 +11160,6 @@ def begin_start_extension_upgrade( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -11915,7 +11186,7 @@ def begin_start_extension_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -11924,17 +11195,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start_extension_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/extensionRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_latest( @@ -11946,7 +11213,6 @@ def get_latest( :type resource_group_name: str :param vm_scale_set_name: The name of the VM scale set. Required. :type vm_scale_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RollingUpgradeStatusInfo or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RollingUpgradeStatusInfo :raises ~azure.core.exceptions.HttpResponseError: @@ -11965,21 +11231,20 @@ def get_latest( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RollingUpgradeStatusInfo] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( + _request = build_virtual_machine_scale_set_rolling_upgrades_get_latest_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_latest.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -11991,16 +11256,12 @@ def get_latest( deserialized = self._deserialize("RollingUpgradeStatusInfo", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_latest.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/rollingUpgrades/latest" - } + return deserialized # type: ignore -class VirtualMachineScaleSetVMExtensionsOperations: +class VirtualMachineScaleSetVMExtensionsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -12026,7 +11287,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -12052,7 +11313,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtension") - request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12062,16 +11323,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12091,10 +11351,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -12124,14 +11380,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12146,7 +11394,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12163,18 +11411,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12189,7 +11429,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to create or update the VMSS VM extension. @@ -12203,20 +11443,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtension type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtension type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12251,7 +11480,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12261,17 +11490,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -12279,7 +11506,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineScaleSetVMExtension: error_map = { @@ -12305,7 +11532,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineScaleSetVMExtensionUpdate") - request = build_virtual_machine_scale_set_vm_extensions_update_request( + _request = build_virtual_machine_scale_set_vm_extensions_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12315,16 +11542,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12336,13 +11562,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -12373,14 +11595,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12395,7 +11609,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -12412,18 +11626,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12438,7 +11644,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineScaleSetVMExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineScaleSetVMExtension]: """The operation to update the VMSS VM extension. @@ -12452,20 +11658,10 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineScaleSetVMExtensionUpdate type or a IO[bytes] type. + Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: @@ -12500,7 +11696,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -12510,17 +11706,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVMExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineScaleSetVMExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, vm_extension_name: str, **kwargs: Any @@ -12539,23 +11733,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_delete_request( + _request = build_virtual_machine_scale_set_vm_extensions_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12565,11 +11758,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -12585,14 +11774,6 @@ def begin_delete( :type instance_id: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12621,7 +11802,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -12630,17 +11811,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -12665,7 +11842,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -12684,7 +11860,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_get_request( + _request = build_virtual_machine_scale_set_vm_extensions_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12692,16 +11868,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12713,13 +11888,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVMExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -12741,7 +11912,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -12760,23 +11930,22 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_extensions_list_request( + _request = build_virtual_machine_scale_set_vm_extensions_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12788,13 +11957,9 @@ def list( deserialized = self._deserialize("VirtualMachineScaleSetVMExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/extensions" - } + return deserialized # type: ignore class VirtualMachineScaleSetVMsOperations: # pylint: disable=too-many-public-methods @@ -12822,7 +11987,9 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> None: error_map = { @@ -12851,7 +12018,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machine_scale_set_vms_reimage_request( + _request = build_virtual_machine_scale_set_vms_reimage_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -12860,16 +12027,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -12879,11 +12045,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -12911,14 +12073,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12930,7 +12084,7 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[IO] = None, + vm_scale_set_vm_reimage_input: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -12945,18 +12099,10 @@ def begin_reimage( :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. Default value is None. - :type vm_scale_set_vm_reimage_input: IO + :type vm_scale_set_vm_reimage_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -12968,7 +12114,9 @@ def begin_reimage( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - vm_scale_set_vm_reimage_input: Optional[Union[_models.VirtualMachineScaleSetVMReimageParameters, IO]] = None, + vm_scale_set_vm_reimage_input: Optional[ + Union[_models.VirtualMachineScaleSetVMReimageParameters, IO[bytes]] + ] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a specific virtual machine in a VM scale set. @@ -12980,20 +12128,10 @@ def begin_reimage( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param vm_scale_set_vm_reimage_input: Parameters for the Reimaging Virtual machine in ScaleSet. - Is either a VirtualMachineScaleSetVMReimageParameters type or a IO type. Default value is None. + Is either a VirtualMachineScaleSetVMReimageParameters type or a IO[bytes] type. Default value + is None. :type vm_scale_set_vm_reimage_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMReimageParameters or IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -13024,7 +12162,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13033,17 +12171,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_all_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13062,22 +12196,21 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_reimage_all_request( + _request = build_virtual_machine_scale_set_vms_reimage_all_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reimage_all_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13087,11 +12220,7 @@ def _reimage_all_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_all_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reimage_all( @@ -13106,14 +12235,6 @@ def begin_reimage_all( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13141,7 +12262,7 @@ def begin_reimage_all( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13150,17 +12271,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage_all.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/reimageall" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13179,22 +12296,21 @@ def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-sta api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( + _request = build_virtual_machine_scale_set_vms_approve_rolling_upgrade_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._approve_rolling_upgrade_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13207,11 +12323,7 @@ def _approve_rolling_upgrade_initial( # pylint: disable=inconsistent-return-sta response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _approve_rolling_upgrade_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/approveRollingUpgrade" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace def begin_approve_rolling_upgrade( @@ -13225,14 +12337,6 @@ def begin_approve_rolling_upgrade( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13260,7 +12364,7 @@ def begin_approve_rolling_upgrade( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13269,17 +12373,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_approve_rolling_upgrade.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/approveRollingUpgrade" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -13298,22 +12398,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_deallocate_request( + _request = build_virtual_machine_scale_set_vms_deallocate_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13323,11 +12422,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -13343,14 +12438,6 @@ def begin_deallocate( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13378,7 +12465,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13387,24 +12474,20 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _update_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -13433,7 +12516,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineScaleSetVM") - request = build_virtual_machine_scale_set_vms_update_request( + _request = build_virtual_machine_scale_set_vms_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -13444,16 +12527,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13473,10 +12555,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } - @overload def begin_update( self, @@ -13513,14 +12591,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13534,7 +12604,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -13552,7 +12622,7 @@ def begin_update( :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -13564,14 +12634,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13585,7 +12647,7 @@ def begin_update( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.VirtualMachineScaleSetVM, IO], + parameters: Union[_models.VirtualMachineScaleSetVM, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -13601,8 +12663,8 @@ def begin_update( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Update Virtual Machine Scale Sets VM operation. - Is either a VirtualMachineScaleSetVM type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM or IO + Is either a VirtualMachineScaleSetVM type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -13611,17 +12673,6 @@ def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -13657,7 +12708,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -13667,17 +12718,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineScaleSetVM].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[_models.VirtualMachineScaleSetVM]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -13702,23 +12751,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_delete_request( + _request = build_virtual_machine_scale_set_vms_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13728,11 +12776,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -13755,14 +12799,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete a virtual machine from a VM scale set. (Feature in Preview). Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -13791,7 +12827,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -13800,17 +12836,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -13834,7 +12866,6 @@ def get( the instance view of the virtual machine. 'UserData' will retrieve the UserData of the virtual machine. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVM or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVM :raises ~azure.core.exceptions.HttpResponseError: @@ -13853,23 +12884,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVM] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_request( + _request = build_virtual_machine_scale_set_vms_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13881,13 +12911,9 @@ def get( deserialized = self._deserialize("VirtualMachineScaleSetVM", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}" - } + return deserialized # type: ignore @distributed_trace def get_instance_view( @@ -13901,7 +12927,6 @@ def get_instance_view( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineScaleSetVMInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineScaleSetVMInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -13920,22 +12945,21 @@ def get_instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineScaleSetVMInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_get_instance_view_request( + _request = build_virtual_machine_scale_set_vms_get_instance_view_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -13947,13 +12971,9 @@ def get_instance_view( deserialized = self._deserialize("VirtualMachineScaleSetVMInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/instanceView" - } + return deserialized # type: ignore @distributed_trace def list( @@ -13982,7 +13002,6 @@ def list( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineScaleSetVM or the result of cls(response) :rtype: @@ -14006,7 +13025,7 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vms_list_request( + _request = build_virtual_machine_scale_set_vms_list_request( resource_group_name=resource_group_name, virtual_machine_scale_set_name=virtual_machine_scale_set_name, subscription_id=self._config.subscription_id, @@ -14014,12 +13033,11 @@ def prepare_request(next_link=None): select=select, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -14030,14 +13048,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineScaleSetVMListResult", pipeline_response) @@ -14047,11 +13065,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14063,10 +13081,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{virtualMachineScaleSetName}/virtualMachines" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, @@ -14090,23 +13104,22 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_power_off_request( + _request = build_virtual_machine_scale_set_vms_power_off_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14116,11 +13129,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -14146,14 +13155,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14182,7 +13183,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14191,17 +13192,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/poweroff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14220,22 +13217,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_restart_request( + _request = build_virtual_machine_scale_set_vms_restart_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14245,11 +13241,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -14263,14 +13255,6 @@ def begin_restart( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14298,7 +13282,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14307,17 +13291,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14336,22 +13316,21 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_start_request( + _request = build_virtual_machine_scale_set_vms_start_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14361,11 +13340,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start( @@ -14379,14 +13354,6 @@ def begin_start( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14414,7 +13381,7 @@ def begin_start( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14423,17 +13390,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14452,22 +13415,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_redeploy_request( + _request = build_virtual_machine_scale_set_vms_redeploy_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14477,11 +13439,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy( @@ -14496,14 +13454,6 @@ def begin_redeploy( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14531,7 +13481,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14540,17 +13490,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -14575,7 +13521,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -14594,23 +13539,22 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machine_scale_set_vms_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14622,13 +13566,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, **kwargs: Any @@ -14647,22 +13587,21 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_perform_maintenance_request( + _request = build_virtual_machine_scale_set_vms_perform_maintenance_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14672,11 +13611,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance( @@ -14690,14 +13625,6 @@ def begin_perform_maintenance( :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -14725,7 +13652,7 @@ def begin_perform_maintenance( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -14734,17 +13661,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -14758,7 +13681,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type vm_scale_set_name: str :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -14777,22 +13699,21 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vms_simulate_eviction_request( + _request = build_virtual_machine_scale_set_vms_simulate_eviction_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14802,18 +13723,14 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _attach_detach_data_disks_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> Optional[_models.StorageProfile]: error_map = { @@ -14839,7 +13756,7 @@ def _attach_detach_data_disks_initial( else: _json = self._serialize.body(parameters, "AttachDetachDataDisksRequest") - request = build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( + _request = build_virtual_machine_scale_set_vms_attach_detach_data_disks_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -14848,16 +13765,15 @@ def _attach_detach_data_disks_initial( content_type=content_type, json=_json, content=_content, - template_url=self._attach_detach_data_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -14875,13 +13791,9 @@ def _attach_detach_data_disks_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, deserialized, response_headers) - - return deserialized + return cls(pipeline_response, deserialized, response_headers) # type: ignore - _attach_detach_data_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/attachDetachDataDisks" - } + return deserialized # type: ignore @overload def begin_attach_detach_data_disks( @@ -14908,14 +13820,6 @@ def begin_attach_detach_data_disks( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -14928,7 +13832,7 @@ def begin_attach_detach_data_disks( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -14943,18 +13847,10 @@ def begin_attach_detach_data_disks( :type instance_id: str :param parameters: Parameters supplied to the attach and detach data disks operation on a Virtual Machine Scale Sets VM. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -14967,7 +13863,7 @@ def begin_attach_detach_data_disks( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from a virtual machine in a VM scale set. @@ -14979,20 +13875,10 @@ def begin_attach_detach_data_disks( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the attach and detach data disks operation on a - Virtual Machine Scale Sets VM. Is either a AttachDetachDataDisksRequest type or a IO type. - Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Virtual Machine Scale Sets VM. Is either a AttachDetachDataDisksRequest type or a IO[bytes] + type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or + IO[bytes] :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -15025,7 +13911,7 @@ def begin_attach_detach_data_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("StorageProfile", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15037,24 +13923,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.StorageProfile].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_attach_detach_data_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/attachDetachDataDisks" - } + return LROPoller[_models.StorageProfile]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { @@ -15080,7 +13964,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machine_scale_set_vms_run_command_request( + _request = build_virtual_machine_scale_set_vms_run_command_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -15089,16 +13973,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15112,13 +13995,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -15144,14 +14023,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -15164,7 +14035,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15178,18 +14049,10 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -15202,7 +14065,7 @@ def begin_run_command( resource_group_name: str, vm_scale_set_name: str, instance_id: str, - parameters: Union[_models.RunCommandInput, IO], + parameters: Union[_models.RunCommandInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on a virtual machine in a VM scale set. @@ -15214,19 +14077,8 @@ def begin_run_command( :param instance_id: The instance ID of the virtual machine. Required. :type instance_id: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -15259,7 +14111,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15271,17 +14123,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualmachines/{instanceId}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineExtensionsOperations: @@ -15309,7 +14159,7 @@ def _create_or_update_initial( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -15335,7 +14185,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtension") - request = build_virtual_machine_extensions_create_or_update_request( + _request = build_virtual_machine_extensions_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -15344,16 +14194,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15373,10 +14222,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } - @overload def begin_create_or_update( self, @@ -15403,14 +14248,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15424,7 +14261,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15440,18 +14277,10 @@ def begin_create_or_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15465,7 +14294,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtension, IO], + extension_parameters: Union[_models.VirtualMachineExtension, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to create or update the extension. @@ -15478,20 +14307,9 @@ def begin_create_or_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Create Virtual Machine Extension - operation. Is either a VirtualMachineExtension type or a IO type. Required. + operation. Is either a VirtualMachineExtension type or a IO[bytes] type. Required. :type extension_parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtension or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15525,7 +14343,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15535,24 +14353,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineExtension: error_map = { @@ -15578,7 +14394,7 @@ def _update_initial( else: _json = self._serialize.body(extension_parameters, "VirtualMachineExtensionUpdate") - request = build_virtual_machine_extensions_update_request( + _request = build_virtual_machine_extensions_update_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, @@ -15587,16 +14403,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15608,13 +14423,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -15643,14 +14454,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15664,7 +14467,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: IO, + extension_parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -15680,18 +14483,10 @@ def begin_update( :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension operation. Required. - :type extension_parameters: IO + :type extension_parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15705,7 +14500,7 @@ def begin_update( resource_group_name: str, vm_name: str, vm_extension_name: str, - extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO], + extension_parameters: Union[_models.VirtualMachineExtensionUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineExtension]: """The operation to update the extension. @@ -15718,20 +14513,9 @@ def begin_update( :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str :param extension_parameters: Parameters supplied to the Update Virtual Machine Extension - operation. Is either a VirtualMachineExtensionUpdate type or a IO type. Required. + operation. Is either a VirtualMachineExtensionUpdate type or a IO[bytes] type. Required. :type extension_parameters: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionUpdate or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineExtension or the result of cls(response) :rtype: @@ -15765,7 +14549,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -15775,17 +14559,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineExtension].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[_models.VirtualMachineExtension]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, vm_extension_name: str, **kwargs: Any @@ -15804,22 +14586,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_delete_request( + _request = build_virtual_machine_extensions_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15829,11 +14610,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -15848,14 +14625,6 @@ def begin_delete( :type vm_name: str :param vm_extension_name: The name of the virtual machine extension. Required. :type vm_extension_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -15883,7 +14652,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -15892,17 +14661,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -15924,7 +14689,6 @@ def get( :type vm_extension_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtension or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtension :raises ~azure.core.exceptions.HttpResponseError: @@ -15943,23 +14707,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtension] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_get_request( + _request = build_virtual_machine_extensions_get_request( resource_group_name=resource_group_name, vm_name=vm_name, vm_extension_name=vm_extension_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -15971,13 +14734,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtension", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions/{vmExtensionName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -15991,7 +14750,6 @@ def list( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionsListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionsListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -16010,22 +14768,21 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtensionsListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_extensions_list_request( + _request = build_virtual_machine_extensions_list_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16037,13 +14794,9 @@ def list( deserialized = self._deserialize("VirtualMachineExtensionsListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/extensions" - } + return deserialized # type: ignore class VirtualMachinesOperations: # pylint: disable=too-many-public-methods @@ -16073,7 +14826,6 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi :param location: The location for which virtual machines under the subscription are queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -16095,16 +14847,15 @@ def list_by_location(self, location: str, **kwargs: Any) -> Iterable["_models.Vi def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_by_location_request( + _request = build_virtual_machines_list_by_location_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_location.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -16115,14 +14866,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -16132,11 +14883,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16148,15 +14899,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_location.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/virtualMachines" - } - def _capture_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineCaptureResult]: error_map = { @@ -16182,7 +14929,7 @@ def _capture_initial( else: _json = self._serialize.body(parameters, "VirtualMachineCaptureParameters") - request = build_virtual_machines_capture_request( + _request = build_virtual_machines_capture_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -16190,16 +14937,15 @@ def _capture_initial( content_type=content_type, json=_json, content=_content, - template_url=self._capture_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16213,13 +14959,9 @@ def _capture_initial( deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _capture_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return deserialized # type: ignore @overload def begin_capture( @@ -16243,14 +14985,6 @@ def begin_capture( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16263,7 +14997,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -16276,18 +15010,10 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16300,7 +15026,7 @@ def begin_capture( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineCaptureParameters, IO], + parameters: Union[_models.VirtualMachineCaptureParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineCaptureResult]: """Captures the VM by copying virtual hard disks of the VM and outputs a template that can be used @@ -16311,19 +15037,9 @@ def begin_capture( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Capture Virtual Machine operation. Is either a - VirtualMachineCaptureParameters type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineCaptureParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineCaptureParameters type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineCaptureParameters or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineCaptureResult or the result of cls(response) :rtype: @@ -16356,7 +15072,7 @@ def begin_capture( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineCaptureResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16368,23 +15084,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineCaptureResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_capture.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/capture" - } + return LROPoller[_models.VirtualMachineCaptureResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _create_or_update_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachine, IO], + parameters: Union[_models.VirtualMachine, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16413,7 +15127,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "VirtualMachine") - request = build_virtual_machines_create_or_update_request( + _request = build_virtual_machines_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -16423,16 +15137,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16452,10 +15165,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } - @overload def begin_create_or_update( self, @@ -16488,14 +15197,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16507,7 +15208,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16522,7 +15223,7 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -16534,14 +15235,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16553,7 +15246,7 @@ def begin_create_or_update( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachine, IO], + parameters: Union[_models.VirtualMachine, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16567,8 +15260,8 @@ def begin_create_or_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Create Virtual Machine operation. Is either a - VirtualMachine type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine or IO + VirtualMachine type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -16577,17 +15270,6 @@ def begin_create_or_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16621,7 +15303,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16631,23 +15313,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineUpdate, IO], + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16676,7 +15356,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "VirtualMachineUpdate") - request = build_virtual_machines_update_request( + _request = build_virtual_machines_update_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -16686,16 +15366,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16707,13 +15386,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -16746,14 +15421,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16765,7 +15432,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16779,7 +15446,7 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -16791,14 +15458,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16810,7 +15469,7 @@ def begin_update( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.VirtualMachineUpdate, IO], + parameters: Union[_models.VirtualMachineUpdate, IO[bytes]], *, if_match: Optional[str] = None, if_none_match: Optional[str] = None, @@ -16823,8 +15482,8 @@ def begin_update( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Update Virtual Machine operation. Is either a - VirtualMachineUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineUpdate or IO + VirtualMachineUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineUpdate or IO[bytes] :keyword if_match: The ETag of the transformation. Omit this value to always overwrite the current resource. Specify the last-seen ETag value to prevent accidentally overwriting concurrent changes. Default value is None. @@ -16833,17 +15492,6 @@ def begin_update( updating an existing record set. Other values will result in error from server as they are not supported. Default value is None. :paramtype if_none_match: str - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachine or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] @@ -16877,7 +15525,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -16887,17 +15535,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachine].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[_models.VirtualMachine]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, force_deletion: Optional[bool] = None, **kwargs: Any @@ -16916,22 +15562,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_delete_request( + _request = build_virtual_machines_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, force_deletion=force_deletion, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -16941,11 +15586,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -16960,14 +15601,6 @@ def begin_delete( :keyword force_deletion: Optional parameter to force delete virtual machines. Default value is None. :paramtype force_deletion: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -16995,7 +15628,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17004,17 +15637,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -17037,7 +15666,6 @@ def get( part of the VM model view that was provided by the user during the VM Create/Update operation. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachine or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachine :raises ~azure.core.exceptions.HttpResponseError: @@ -17056,22 +15684,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachine] = kwargs.pop("cls", None) - request = build_virtual_machines_get_request( + _request = build_virtual_machines_get_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17083,13 +15710,9 @@ def get( deserialized = self._deserialize("VirtualMachine", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}" - } + return deserialized # type: ignore @distributed_trace def instance_view( @@ -17101,7 +15724,6 @@ def instance_view( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineInstanceView or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstanceView :raises ~azure.core.exceptions.HttpResponseError: @@ -17120,21 +15742,20 @@ def instance_view( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineInstanceView] = kwargs.pop("cls", None) - request = build_virtual_machines_instance_view_request( + _request = build_virtual_machines_instance_view_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.instance_view.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17146,13 +15767,9 @@ def instance_view( deserialized = self._deserialize("VirtualMachineInstanceView", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - instance_view.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/instanceView" - } + return deserialized # type: ignore def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17171,21 +15788,20 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_convert_to_managed_disks_request( + _request = build_virtual_machines_convert_to_managed_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._convert_to_managed_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17195,11 +15811,7 @@ def _convert_to_managed_disks_initial( # pylint: disable=inconsistent-return-st raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _convert_to_managed_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17210,14 +15822,6 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17244,7 +15848,7 @@ def begin_convert_to_managed_disks(self, resource_group_name: str, vm_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17253,17 +15857,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_convert_to_managed_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/convertToManagedDisks" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _deallocate_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, hibernate: Optional[bool] = None, **kwargs: Any @@ -17282,22 +15882,21 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_deallocate_request( + _request = build_virtual_machines_deallocate_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, hibernate=hibernate, api_version=api_version, - template_url=self._deallocate_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17307,11 +15906,7 @@ def _deallocate_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _deallocate_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_deallocate( @@ -17327,14 +15922,6 @@ def begin_deallocate( :keyword hibernate: Optional parameter to hibernate a virtual machine. (Feature in Preview). Default value is None. :paramtype hibernate: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17362,7 +15949,7 @@ def begin_deallocate( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17371,17 +15958,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_deallocate.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/deallocate" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def generalize( # pylint: disable=inconsistent-return-statements @@ -17398,7 +15981,6 @@ def generalize( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -17417,21 +15999,20 @@ def generalize( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_generalize_request( + _request = build_virtual_machines_generalize_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.generalize.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17441,11 +16022,7 @@ def generalize( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - generalize.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/generalize" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def list( @@ -17470,7 +16047,6 @@ def list( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypeForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -17492,18 +16068,17 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_request( + _request = build_virtual_machines_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, filter=filter, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17514,14 +16089,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -17531,11 +16106,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17547,10 +16122,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines" - } - @distributed_trace def list_all( self, @@ -17575,7 +16146,6 @@ def list_all( run time status of all Virtual Machines, this can only be specified if a valid $filter option is specified. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForListVMs - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachine or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachine] :raises ~azure.core.exceptions.HttpResponseError: @@ -17597,18 +16167,17 @@ def list_all( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_all_request( + _request = build_virtual_machines_list_all_request( subscription_id=self._config.subscription_id, status_only=status_only, filter=filter, expand=expand, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17619,14 +16188,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineListResult", pipeline_response) @@ -17636,11 +16205,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17652,8 +16221,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachines"} - @distributed_trace def list_available_sizes( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17665,7 +16232,6 @@ def list_available_sizes( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -17687,17 +16253,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machines_list_available_sizes_request( + _request = build_virtual_machines_list_available_sizes_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -17708,14 +16273,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -17725,11 +16290,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17741,10 +16306,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/vmSizes" - } - def _power_off_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, *, skip_shutdown: bool = False, **kwargs: Any ) -> None: @@ -17762,22 +16323,21 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_power_off_request( + _request = build_virtual_machines_power_off_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, skip_shutdown=skip_shutdown, api_version=api_version, - template_url=self._power_off_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17787,11 +16347,7 @@ def _power_off_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _power_off_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_power_off( @@ -17808,14 +16364,6 @@ def begin_power_off( flag indicates non-graceful shutdown whereas false indicates otherwise. Default value for this flag is false if not specified. Default value is False. :paramtype skip_shutdown: bool - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17843,7 +16391,7 @@ def begin_power_off( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17852,17 +16400,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_power_off.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/powerOff" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reapply_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17881,21 +16425,20 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_reapply_request( + _request = build_virtual_machines_reapply_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._reapply_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -17905,11 +16448,7 @@ def _reapply_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reapply_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -17919,14 +16458,6 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -17953,7 +16484,7 @@ def begin_reapply(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -17962,17 +16493,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reapply.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reapply" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -17991,21 +16518,20 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_restart_request( + _request = build_virtual_machines_restart_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18015,11 +16541,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18029,14 +16551,6 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18063,7 +16577,7 @@ def begin_restart(self, resource_group_name: str, vm_name: str, **kwargs: Any) - def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18072,17 +16586,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _start_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18101,21 +16611,20 @@ def _start_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_start_request( + _request = build_virtual_machines_start_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._start_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18125,11 +16634,7 @@ def _start_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _start_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18139,14 +16644,6 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18173,7 +16670,7 @@ def begin_start(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18182,17 +16679,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_start.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/start" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18211,21 +16704,20 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_redeploy_request( + _request = build_virtual_machines_redeploy_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18235,11 +16727,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18249,14 +16737,6 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18283,7 +16763,7 @@ def begin_redeploy(self, resource_group_name: str, vm_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18292,23 +16772,19 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _reimage_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> None: error_map = { @@ -18337,7 +16813,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements else: _json = None - request = build_virtual_machines_reimage_request( + _request = build_virtual_machines_reimage_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18345,16 +16821,15 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements content_type=content_type, json=_json, content=_content, - template_url=self._reimage_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18364,11 +16839,7 @@ def _reimage_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _reimage_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return cls(pipeline_response, None, {}) # type: ignore @overload def begin_reimage( @@ -18397,14 +16868,6 @@ def begin_reimage( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18415,7 +16878,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -18433,18 +16896,10 @@ def begin_reimage( :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18455,7 +16910,7 @@ def begin_reimage( self, resource_group_name: str, vm_name: str, - parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO]] = None, + parameters: Optional[Union[_models.VirtualMachineReimageParameters, IO[bytes]]] = None, **kwargs: Any ) -> LROPoller[None]: """Reimages (upgrade the operating system) a virtual machine which don't have a ephemeral OS disk, @@ -18470,19 +16925,9 @@ def begin_reimage( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Reimage Virtual Machine operation. Is either a - VirtualMachineReimageParameters type or a IO type. Default value is None. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineReimageParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + VirtualMachineReimageParameters type or a IO[bytes] type. Default value is None. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineReimageParameters or + IO[bytes] :return: An instance of LROPoller that returns either None or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[None] :raises ~azure.core.exceptions.HttpResponseError: @@ -18512,7 +16957,7 @@ def begin_reimage( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18521,17 +16966,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_reimage.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/reimage" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def retrieve_boot_diagnostics_data( @@ -18552,7 +16993,6 @@ def retrieve_boot_diagnostics_data( with a value between 1 to 1440 minutes. **Note:** If not specified, SAS URIs will be generated with a default expiration duration of 120 minutes. Default value is None. :paramtype sas_uri_expiration_time_in_minutes: int - :keyword callable cls: A custom type or function that will be passed the direct response :return: RetrieveBootDiagnosticsDataResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RetrieveBootDiagnosticsDataResult :raises ~azure.core.exceptions.HttpResponseError: @@ -18571,22 +17011,21 @@ def retrieve_boot_diagnostics_data( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RetrieveBootDiagnosticsDataResult] = kwargs.pop("cls", None) - request = build_virtual_machines_retrieve_boot_diagnostics_data_request( + _request = build_virtual_machines_retrieve_boot_diagnostics_data_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, sas_uri_expiration_time_in_minutes=sas_uri_expiration_time_in_minutes, api_version=api_version, - template_url=self.retrieve_boot_diagnostics_data.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18598,13 +17037,9 @@ def retrieve_boot_diagnostics_data( deserialized = self._deserialize("RetrieveBootDiagnosticsDataResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - retrieve_boot_diagnostics_data.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/retrieveBootDiagnosticsData" - } + return deserialized # type: ignore def _perform_maintenance_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18623,21 +17058,20 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_perform_maintenance_request( + _request = build_virtual_machines_perform_maintenance_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._perform_maintenance_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18647,11 +17081,7 @@ def _perform_maintenance_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _perform_maintenance_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kwargs: Any) -> LROPoller[None]: @@ -18661,14 +17091,6 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -18695,7 +17117,7 @@ def begin_perform_maintenance(self, resource_group_name: str, vm_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -18704,17 +17126,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_perform_maintenance.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/performMaintenance" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def simulate_eviction( # pylint: disable=inconsistent-return-statements @@ -18726,7 +17144,6 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -18745,21 +17162,20 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machines_simulate_eviction_request( + _request = build_virtual_machines_simulate_eviction_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.simulate_eviction.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18769,11 +17185,7 @@ def simulate_eviction( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - simulate_eviction.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/simulateEviction" - } + return cls(pipeline_response, None, {}) # type: ignore def _assess_patches_initial( self, resource_group_name: str, vm_name: str, **kwargs: Any @@ -18792,21 +17204,20 @@ def _assess_patches_initial( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[Optional[_models.VirtualMachineAssessPatchesResult]] = kwargs.pop("cls", None) - request = build_virtual_machines_assess_patches_request( + _request = build_virtual_machines_assess_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._assess_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18820,13 +17231,9 @@ def _assess_patches_initial( deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _assess_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return deserialized # type: ignore @distributed_trace def begin_assess_patches( @@ -18838,14 +17245,6 @@ def begin_assess_patches( :type resource_group_name: str :param vm_name: The name of the virtual machine. Required. :type vm_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineAssessPatchesResult or the result of cls(response) :rtype: @@ -18875,7 +17274,7 @@ def begin_assess_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineAssessPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -18887,23 +17286,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineAssessPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_assess_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/assessPatches" - } + return LROPoller[_models.VirtualMachineAssessPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _install_patches_initial( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> Optional[_models.VirtualMachineInstallPatchesResult]: error_map = { @@ -18929,7 +17326,7 @@ def _install_patches_initial( else: _json = self._serialize.body(install_patches_input, "VirtualMachineInstallPatchesParameters") - request = build_virtual_machines_install_patches_request( + _request = build_virtual_machines_install_patches_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -18937,16 +17334,15 @@ def _install_patches_initial( content_type=content_type, json=_json, content=_content, - template_url=self._install_patches_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -18960,13 +17356,9 @@ def _install_patches_initial( deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _install_patches_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return deserialized # type: ignore @overload def begin_install_patches( @@ -18991,14 +17383,6 @@ def begin_install_patches( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19011,7 +17395,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: IO, + install_patches_input: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19024,18 +17408,10 @@ def begin_install_patches( :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Required. - :type install_patches_input: IO + :type install_patches_input: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19048,7 +17424,7 @@ def begin_install_patches( self, resource_group_name: str, vm_name: str, - install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO], + install_patches_input: Union[_models.VirtualMachineInstallPatchesParameters, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineInstallPatchesResult]: """Installs patches on the VM. @@ -19058,20 +17434,9 @@ def begin_install_patches( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param install_patches_input: Input for InstallPatches as directly received by the API. Is - either a VirtualMachineInstallPatchesParameters type or a IO type. Required. + either a VirtualMachineInstallPatchesParameters type or a IO[bytes] type. Required. :type install_patches_input: - ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstallPatchesParameters or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineInstallPatchesParameters or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineInstallPatchesResult or the result of cls(response) :rtype: @@ -19104,7 +17469,7 @@ def begin_install_patches( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineInstallPatchesResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19116,23 +17481,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineInstallPatchesResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_install_patches.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/installPatches" - } + return LROPoller[_models.VirtualMachineInstallPatchesResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _attach_detach_data_disks_initial( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> Optional[_models.StorageProfile]: error_map = { @@ -19158,7 +17521,7 @@ def _attach_detach_data_disks_initial( else: _json = self._serialize.body(parameters, "AttachDetachDataDisksRequest") - request = build_virtual_machines_attach_detach_data_disks_request( + _request = build_virtual_machines_attach_detach_data_disks_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -19166,16 +17529,15 @@ def _attach_detach_data_disks_initial( content_type=content_type, json=_json, content=_content, - template_url=self._attach_detach_data_disks_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19193,13 +17555,9 @@ def _attach_detach_data_disks_initial( response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, deserialized, response_headers) + return cls(pipeline_response, deserialized, response_headers) # type: ignore - return deserialized - - _attach_detach_data_disks_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/attachDetachDataDisks" - } + return deserialized # type: ignore @overload def begin_attach_detach_data_disks( @@ -19223,14 +17581,6 @@ def begin_attach_detach_data_disks( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -19242,7 +17592,7 @@ def begin_attach_detach_data_disks( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19255,18 +17605,10 @@ def begin_attach_detach_data_disks( :type vm_name: str :param parameters: Parameters supplied to the attach and detach data disks operation on the virtual machine. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -19278,7 +17620,7 @@ def begin_attach_detach_data_disks( self, resource_group_name: str, vm_name: str, - parameters: Union[_models.AttachDetachDataDisksRequest, IO], + parameters: Union[_models.AttachDetachDataDisksRequest, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.StorageProfile]: """Attach and detach data disks to/from the virtual machine. @@ -19288,19 +17630,9 @@ def begin_attach_detach_data_disks( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the attach and detach data disks operation on the - virtual machine. Is either a AttachDetachDataDisksRequest type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + virtual machine. Is either a AttachDetachDataDisksRequest type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AttachDetachDataDisksRequest or + IO[bytes] :return: An instance of LROPoller that returns either StorageProfile or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.StorageProfile] @@ -19332,7 +17664,7 @@ def begin_attach_detach_data_disks( def get_long_running_output(pipeline_response): deserialized = self._deserialize("StorageProfile", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19344,20 +17676,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.StorageProfile].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_attach_detach_data_disks.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/attachDetachDataDisks" - } + return LROPoller[_models.StorageProfile]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _run_command_initial( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> Optional[_models.RunCommandResult]: error_map = { 401: ClientAuthenticationError, @@ -19382,7 +17716,7 @@ def _run_command_initial( else: _json = self._serialize.body(parameters, "RunCommandInput") - request = build_virtual_machines_run_command_request( + _request = build_virtual_machines_run_command_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, @@ -19390,16 +17724,15 @@ def _run_command_initial( content_type=content_type, json=_json, content=_content, - template_url=self._run_command_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19413,13 +17746,9 @@ def _run_command_initial( deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _run_command_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return deserialized # type: ignore @overload def begin_run_command( @@ -19442,14 +17771,6 @@ def begin_run_command( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -19461,7 +17782,7 @@ def begin_run_command( self, resource_group_name: str, vm_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -19473,18 +17794,10 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -19493,7 +17806,11 @@ def begin_run_command( @distributed_trace def begin_run_command( - self, resource_group_name: str, vm_name: str, parameters: Union[_models.RunCommandInput, IO], **kwargs: Any + self, + resource_group_name: str, + vm_name: str, + parameters: Union[_models.RunCommandInput, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.RunCommandResult]: """Run command on the VM. @@ -19502,19 +17819,8 @@ def begin_run_command( :param vm_name: The name of the virtual machine. Required. :type vm_name: str :param parameters: Parameters supplied to the Run command operation. Is either a - RunCommandInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RunCommandInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RunCommandInput or IO[bytes] :return: An instance of LROPoller that returns either RunCommandResult or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RunCommandResult] @@ -19546,7 +17852,7 @@ def begin_run_command( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RunCommandResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -19558,17 +17864,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RunCommandResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_run_command.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommand" - } + return LROPoller[_models.RunCommandResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineImagesOperations: @@ -19607,7 +17911,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -19626,7 +17929,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_get_request( + _request = build_virtual_machine_images_get_request( location=location, publisher_name=publisher_name, offer=offer, @@ -19634,16 +17937,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19655,13 +17957,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -19693,7 +17991,6 @@ def list( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19712,7 +18009,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_request( + _request = build_virtual_machine_images_list_request( location=location, publisher_name=publisher_name, offer=offer, @@ -19722,16 +18019,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19743,13 +18039,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -19761,7 +18053,6 @@ def list_offers( :type location: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19780,21 +18071,20 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_offers_request( + _request = build_virtual_machine_images_list_offers_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19806,13 +18096,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualMachineImageResource]: @@ -19820,7 +18106,6 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM :param location: The name of a supported Azure region. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19839,20 +18124,19 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_publishers_request( + _request = build_virtual_machine_images_list_publishers_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19864,13 +18148,9 @@ def list_publishers(self, location: str, **kwargs: Any) -> List[_models.VirtualM deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -19884,7 +18164,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -19903,22 +18182,21 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_skus_request( + _request = build_virtual_machine_images_list_skus_request( location=location, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19930,13 +18208,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore @distributed_trace def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _models.VmImagesInEdgeZoneListResult: @@ -19946,7 +18220,6 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VmImagesInEdgeZoneListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VmImagesInEdgeZoneListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -19965,21 +18238,20 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VmImagesInEdgeZoneListResult] = kwargs.pop("cls", None) - request = build_virtual_machine_images_list_by_edge_zone_request( + _request = build_virtual_machine_images_list_by_edge_zone_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_edge_zone.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -19991,13 +18263,9 @@ def list_by_edge_zone(self, location: str, edge_zone: str, **kwargs: Any) -> _mo deserialized = self._deserialize("VmImagesInEdgeZoneListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_by_edge_zone.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/vmimages" - } + return deserialized # type: ignore class VirtualMachineImagesEdgeZoneOperations: @@ -20038,7 +18306,6 @@ def get( :type skus: str :param version: A valid image SKU version. Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImage :raises ~azure.core.exceptions.HttpResponseError: @@ -20057,7 +18324,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineImage] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_get_request( + _request = build_virtual_machine_images_edge_zone_get_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -20066,16 +18333,15 @@ def get( version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20087,13 +18353,9 @@ def get( deserialized = self._deserialize("VirtualMachineImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -20130,7 +18392,6 @@ def list( :keyword orderby: Specifies the order of the results returned. Formatted as an OData query. Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -20149,7 +18410,7 @@ def list( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_request( + _request = build_virtual_machine_images_edge_zone_list_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, @@ -20160,16 +18421,15 @@ def list( top=top, orderby=orderby, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20181,13 +18441,9 @@ def list( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus/{skus}/versions" - } + return deserialized # type: ignore @distributed_trace def list_offers( @@ -20202,7 +18458,6 @@ def list_offers( :type edge_zone: str :param publisher_name: A valid image publisher. Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -20221,22 +18476,21 @@ def list_offers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_offers_request( + _request = build_virtual_machine_images_edge_zone_list_offers_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_offers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20248,13 +18502,9 @@ def list_offers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_offers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers" - } + return deserialized # type: ignore @distributed_trace def list_publishers( @@ -20266,7 +18516,6 @@ def list_publishers( :type location: str :param edge_zone: The name of the edge zone. Required. :type edge_zone: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -20285,21 +18534,20 @@ def list_publishers( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_publishers_request( + _request = build_virtual_machine_images_edge_zone_list_publishers_request( location=location, edge_zone=edge_zone, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_publishers.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20311,13 +18559,9 @@ def list_publishers( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_publishers.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers" - } + return deserialized # type: ignore @distributed_trace def list_skus( @@ -20334,7 +18578,6 @@ def list_skus( :type publisher_name: str :param offer: A valid image publisher offer. Required. :type offer: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineImageResource or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineImageResource] :raises ~azure.core.exceptions.HttpResponseError: @@ -20353,23 +18596,22 @@ def list_skus( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineImageResource]] = kwargs.pop("cls", None) - request = build_virtual_machine_images_edge_zone_list_skus_request( + _request = build_virtual_machine_images_edge_zone_list_skus_request( location=location, edge_zone=edge_zone, publisher_name=publisher_name, offer=offer, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_skus.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20381,13 +18623,9 @@ def list_skus( deserialized = self._deserialize("[VirtualMachineImageResource]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_skus.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/edgeZones/{edgeZone}/publishers/{publisherName}/artifacttypes/vmimage/offers/{offer}/skus" - } + return deserialized # type: ignore class VirtualMachineExtensionImagesOperations: @@ -20424,7 +18662,6 @@ def get( :type type: str :param version: Required. :type version: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineExtensionImage or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage :raises ~azure.core.exceptions.HttpResponseError: @@ -20443,23 +18680,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineExtensionImage] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_get_request( + _request = build_virtual_machine_extension_images_get_request( location=location, publisher_name=publisher_name, type=type, version=version, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20471,13 +18707,9 @@ def get( deserialized = self._deserialize("VirtualMachineExtensionImage", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions/{version}" - } + return deserialized # type: ignore @distributed_trace def list_types( @@ -20489,7 +18721,6 @@ def list_types( :type location: str :param publisher_name: Required. :type publisher_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -20508,21 +18739,20 @@ def list_types( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_types_request( + _request = build_virtual_machine_extension_images_list_types_request( location=location, publisher_name=publisher_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_types.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20534,13 +18764,9 @@ def list_types( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_types.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types" - } + return deserialized # type: ignore @distributed_trace def list_versions( @@ -20568,7 +18794,6 @@ def list_versions( :paramtype top: int :keyword orderby: Default value is None. :paramtype orderby: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: list of VirtualMachineExtensionImage or the result of cls(response) :rtype: list[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineExtensionImage] :raises ~azure.core.exceptions.HttpResponseError: @@ -20587,7 +18812,7 @@ def list_versions( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[List[_models.VirtualMachineExtensionImage]] = kwargs.pop("cls", None) - request = build_virtual_machine_extension_images_list_versions_request( + _request = build_virtual_machine_extension_images_list_versions_request( location=location, publisher_name=publisher_name, type=type, @@ -20596,16 +18821,15 @@ def list_versions( top=top, orderby=orderby, api_version=api_version, - template_url=self.list_versions.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20617,13 +18841,9 @@ def list_versions( deserialized = self._deserialize("[VirtualMachineExtensionImage]", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - list_versions.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/publishers/{publisherName}/artifacttypes/vmextension/types/{type}/versions" - } + return deserialized # type: ignore class AvailabilitySetsOperations: @@ -20667,7 +18887,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20678,7 +18897,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20690,11 +18909,10 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20705,7 +18923,7 @@ def create_or_update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySet, IO], + parameters: Union[_models.AvailabilitySet, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Create or update an availability set. @@ -20715,12 +18933,8 @@ def create_or_update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Create Availability Set operation. Is either a - AvailabilitySet type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySet type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20748,7 +18962,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "AvailabilitySet") - request = build_availability_sets_create_or_update_request( + _request = build_availability_sets_create_or_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -20756,16 +18970,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20777,13 +18990,9 @@ def create_or_update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @overload def update( @@ -20806,7 +19015,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20817,7 +19025,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -20829,11 +19037,10 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20844,7 +19051,7 @@ def update( self, resource_group_name: str, availability_set_name: str, - parameters: Union[_models.AvailabilitySetUpdate, IO], + parameters: Union[_models.AvailabilitySetUpdate, IO[bytes]], **kwargs: Any ) -> _models.AvailabilitySet: """Update an availability set. @@ -20854,12 +19061,8 @@ def update( :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str :param parameters: Parameters supplied to the Update Availability Set operation. Is either a - AvailabilitySetUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + AvailabilitySetUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySetUpdate or IO[bytes] :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -20887,7 +19090,7 @@ def update( else: _json = self._serialize.body(parameters, "AvailabilitySetUpdate") - request = build_availability_sets_update_request( + _request = build_availability_sets_update_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, @@ -20895,16 +19098,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20916,13 +19118,9 @@ def update( deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -20934,7 +19132,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -20953,21 +19150,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_availability_sets_delete_request( + _request = build_availability_sets_delete_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -20977,11 +19173,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, availability_set_name: str, **kwargs: Any) -> _models.AvailabilitySet: @@ -20991,7 +19183,6 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: AvailabilitySet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet :raises ~azure.core.exceptions.HttpResponseError: @@ -21010,21 +19201,20 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.AvailabilitySet] = kwargs.pop("cls", None) - request = build_availability_sets_get_request( + _request = build_availability_sets_get_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21036,13 +19226,9 @@ def get(self, resource_group_name: str, availability_set_name: str, **kwargs: An deserialized = self._deserialize("AvailabilitySet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription( @@ -21053,7 +19239,6 @@ def list_by_subscription( :keyword expand: The expand expression to apply to the operation. Allowed values are 'instanceView'. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -21075,16 +19260,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_by_subscription_request( + _request = build_availability_sets_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21095,14 +19279,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -21112,11 +19296,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21128,17 +19312,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.AvailabilitySet"]: """Lists all availability sets in a resource group. :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either AvailabilitySet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.AvailabilitySet] :raises ~azure.core.exceptions.HttpResponseError: @@ -21160,16 +19339,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Ava def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_request( + _request = build_availability_sets_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21180,14 +19358,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("AvailabilitySetListResult", pipeline_response) @@ -21197,11 +19375,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21213,10 +19391,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets" - } - @distributed_trace def list_available_sizes( self, resource_group_name: str, availability_set_name: str, **kwargs: Any @@ -21228,7 +19402,6 @@ def list_available_sizes( :type resource_group_name: str :param availability_set_name: The name of the availability set. Required. :type availability_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineSize or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.VirtualMachineSize] :raises ~azure.core.exceptions.HttpResponseError: @@ -21250,17 +19423,16 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_availability_sets_list_available_sizes_request( + _request = build_availability_sets_list_available_sizes_request( resource_group_name=resource_group_name, availability_set_name=availability_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21271,14 +19443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineSizeListResult", pipeline_response) @@ -21288,11 +19460,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21304,10 +19476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/availabilitySets/{availabilitySetName}/vmSizes" - } - class ProximityPlacementGroupsOperations: """ @@ -21351,7 +19519,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21362,7 +19529,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21375,11 +19542,10 @@ def create_or_update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21390,7 +19556,7 @@ def create_or_update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroup, IO], + parameters: Union[_models.ProximityPlacementGroup, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Create or update a proximity placement group. @@ -21400,12 +19566,8 @@ def create_or_update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Create Proximity Placement Group operation. Is - either a ProximityPlacementGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup or IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21433,7 +19595,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroup") - request = build_proximity_placement_groups_create_or_update_request( + _request = build_proximity_placement_groups_create_or_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -21441,16 +19603,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21470,10 +19631,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } - @overload def update( self, @@ -21496,7 +19653,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21507,7 +19663,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21520,11 +19676,10 @@ def update( :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21535,7 +19690,7 @@ def update( self, resource_group_name: str, proximity_placement_group_name: str, - parameters: Union[_models.ProximityPlacementGroupUpdate, IO], + parameters: Union[_models.ProximityPlacementGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.ProximityPlacementGroup: """Update a proximity placement group. @@ -21545,12 +19700,9 @@ def update( :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str :param parameters: Parameters supplied to the Update Proximity Placement Group operation. Is - either a ProximityPlacementGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a ProximityPlacementGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroupUpdate or + IO[bytes] :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21578,7 +19730,7 @@ def update( else: _json = self._serialize.body(parameters, "ProximityPlacementGroupUpdate") - request = build_proximity_placement_groups_update_request( + _request = build_proximity_placement_groups_update_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, @@ -21586,16 +19738,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21607,13 +19758,9 @@ def update( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -21625,7 +19772,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param proximity_placement_group_name: The name of the proximity placement group. Required. :type proximity_placement_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -21644,21 +19790,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_delete_request( + _request = build_proximity_placement_groups_delete_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21668,11 +19813,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -21692,7 +19833,6 @@ def get( :keyword include_colocation_status: includeColocationStatus=true enables fetching the colocation status of all the resources in the proximity placement group. Default value is None. :paramtype include_colocation_status: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: ProximityPlacementGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.ProximityPlacementGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21711,22 +19851,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.ProximityPlacementGroup] = kwargs.pop("cls", None) - request = build_proximity_placement_groups_get_request( + _request = build_proximity_placement_groups_get_request( resource_group_name=resource_group_name, proximity_placement_group_name=proximity_placement_group_name, subscription_id=self._config.subscription_id, include_colocation_status=include_colocation_status, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21738,19 +19877,14 @@ def get( deserialized = self._deserialize("ProximityPlacementGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups/{proximityPlacementGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlacementGroup"]: """Lists all proximity placement groups in a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -21774,15 +19908,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.ProximityPlac def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_subscription_request( + _request = build_proximity_placement_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21793,14 +19926,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -21810,11 +19943,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21826,10 +19959,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/proximityPlacementGroups" - } - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -21838,7 +19967,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either ProximityPlacementGroup or the result of cls(response) :rtype: @@ -21862,16 +19990,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_proximity_placement_groups_list_by_resource_group_request( + _request = build_proximity_placement_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -21882,14 +20009,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ProximityPlacementGroupListResult", pipeline_response) @@ -21899,11 +20026,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -21915,10 +20042,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/proximityPlacementGroups" - } - class DedicatedHostGroupsOperations: """ @@ -21963,7 +20086,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -21974,7 +20096,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -21988,11 +20110,10 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22003,7 +20124,7 @@ def create_or_update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroup, IO], + parameters: Union[_models.DedicatedHostGroup, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Create or update a dedicated host group. For details of Dedicated Host and Dedicated Host @@ -22015,12 +20136,8 @@ def create_or_update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Create Dedicated Host Group. Is either a - DedicatedHostGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + DedicatedHostGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22048,7 +20165,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "DedicatedHostGroup") - request = build_dedicated_host_groups_create_or_update_request( + _request = build_dedicated_host_groups_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -22056,16 +20173,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22085,10 +20201,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } - @overload def update( self, @@ -22110,7 +20222,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22121,7 +20232,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22133,11 +20244,10 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22148,7 +20258,7 @@ def update( self, resource_group_name: str, host_group_name: str, - parameters: Union[_models.DedicatedHostGroupUpdate, IO], + parameters: Union[_models.DedicatedHostGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHostGroup: """Update an dedicated host group. @@ -22158,12 +20268,8 @@ def update( :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str :param parameters: Parameters supplied to the Update Dedicated Host Group operation. Is either - a DedicatedHostGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + a DedicatedHostGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroupUpdate or IO[bytes] :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22191,7 +20297,7 @@ def update( else: _json = self._serialize.body(parameters, "DedicatedHostGroupUpdate") - request = build_dedicated_host_groups_update_request( + _request = build_dedicated_host_groups_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, @@ -22199,16 +20305,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22220,13 +20325,9 @@ def update( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -22238,7 +20339,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -22257,21 +20357,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_delete_request( + _request = build_dedicated_host_groups_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22281,11 +20380,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -22307,7 +20402,6 @@ def get( not supported for dedicated host group. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHostGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -22326,22 +20420,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.DedicatedHostGroup] = kwargs.pop("cls", None) - request = build_dedicated_host_groups_get_request( + _request = build_dedicated_host_groups_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22353,13 +20446,9 @@ def get( deserialized = self._deserialize("DedicatedHostGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: @@ -22368,7 +20457,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -22390,16 +20478,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_resource_group_request( + _request = build_dedicated_host_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22410,14 +20497,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -22427,11 +20514,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22443,16 +20530,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups" - } - @distributed_trace def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHostGroup"]: """Lists all of the dedicated host groups in the subscription. Use the nextLink property in the response to get the next page of dedicated host groups. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHostGroup or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHostGroup] :raises ~azure.core.exceptions.HttpResponseError: @@ -22474,15 +20556,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.DedicatedHost def prepare_request(next_link=None): if not next_link: - request = build_dedicated_host_groups_list_by_subscription_request( + _request = build_dedicated_host_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -22493,14 +20574,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostGroupListResult", pipeline_response) @@ -22510,11 +20591,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22526,8 +20607,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/hostGroups"} - class DedicatedHostsOperations: """ @@ -22554,7 +20633,7 @@ def _create_or_update_initial( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -22580,7 +20659,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "DedicatedHost") - request = build_dedicated_hosts_create_or_update_request( + _request = build_dedicated_hosts_create_or_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -22589,16 +20668,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22618,10 +20696,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } - @overload def begin_create_or_update( self, @@ -22646,14 +20720,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22666,7 +20732,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22680,18 +20746,10 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22704,7 +20762,7 @@ def begin_create_or_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHost, IO], + parameters: Union[_models.DedicatedHost, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Create or update a dedicated host . @@ -22716,19 +20774,8 @@ def begin_create_or_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Create Dedicated Host. Is either a DedicatedHost - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22761,7 +20808,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -22771,24 +20818,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> _models.DedicatedHost: error_map = { @@ -22814,7 +20859,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "DedicatedHostUpdate") - request = build_dedicated_hosts_update_request( + _request = build_dedicated_hosts_update_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, @@ -22823,16 +20868,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -22844,13 +20888,9 @@ def _update_initial( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -22876,14 +20916,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22896,7 +20928,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -22910,18 +20942,10 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22934,7 +20958,7 @@ def begin_update( resource_group_name: str, host_group_name: str, host_name: str, - parameters: Union[_models.DedicatedHostUpdate, IO], + parameters: Union[_models.DedicatedHostUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DedicatedHost]: """Update a dedicated host . @@ -22946,19 +20970,8 @@ def begin_update( :param host_name: The name of the dedicated host . Required. :type host_name: str :param parameters: Parameters supplied to the Update Dedicated Host operation. Is either a - DedicatedHostUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DedicatedHostUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHostUpdate or IO[bytes] :return: An instance of LROPoller that returns either DedicatedHost or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] @@ -22991,7 +21004,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -23001,17 +21014,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DedicatedHost].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[_models.DedicatedHost]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -23030,22 +21041,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_delete_request( + _request = build_dedicated_hosts_delete_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23055,11 +21065,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -23073,14 +21079,6 @@ def begin_delete( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23108,7 +21106,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23117,17 +21115,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -23151,7 +21145,6 @@ def get( the list of instance views of the dedicated host. 'UserData' is not supported for dedicated host. Known values are: "instanceView" and "userData". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.InstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: DedicatedHost or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.DedicatedHost :raises ~azure.core.exceptions.HttpResponseError: @@ -23170,23 +21163,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.DedicatedHost] = kwargs.pop("cls", None) - request = build_dedicated_hosts_get_request( + _request = build_dedicated_hosts_get_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23198,13 +21190,9 @@ def get( deserialized = self._deserialize("DedicatedHost", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}" - } + return deserialized # type: ignore @distributed_trace def list_by_host_group( @@ -23217,7 +21205,6 @@ def list_by_host_group( :type resource_group_name: str :param host_group_name: The name of the dedicated host group. Required. :type host_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DedicatedHost or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.DedicatedHost] :raises ~azure.core.exceptions.HttpResponseError: @@ -23239,17 +21226,16 @@ def list_by_host_group( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_by_host_group_request( + _request = build_dedicated_hosts_list_by_host_group_request( resource_group_name=resource_group_name, host_group_name=host_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_host_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23260,14 +21246,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostListResult", pipeline_response) @@ -23277,11 +21263,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23293,10 +21279,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_host_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts" - } - def _restart_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any ) -> None: @@ -23314,22 +21296,21 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_restart_request( + _request = build_dedicated_hosts_restart_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._restart_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23339,11 +21320,7 @@ def _restart_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _restart_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_restart( @@ -23360,14 +21337,6 @@ def begin_restart( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23395,7 +21364,7 @@ def begin_restart( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23404,17 +21373,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_restart.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/restart" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore def _redeploy_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, host_group_name: str, host_name: str, **kwargs: Any @@ -23433,22 +21398,21 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_dedicated_hosts_redeploy_request( + _request = build_dedicated_hosts_redeploy_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._redeploy_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23461,11 +21425,7 @@ def _redeploy_initial( # pylint: disable=inconsistent-return-statements response_headers["Location"] = self._deserialize("str", response.headers.get("Location")) if cls: - return cls(pipeline_response, None, response_headers) - - _redeploy_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/redeploy" - } + return cls(pipeline_response, None, response_headers) # type: ignore @distributed_trace def begin_redeploy( @@ -23483,14 +21443,6 @@ def begin_redeploy( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -23518,7 +21470,7 @@ def begin_redeploy( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -23527,17 +21479,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_redeploy.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/redeploy" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_available_sizes( @@ -23553,7 +21501,6 @@ def list_available_sizes( :type host_group_name: str :param host_name: The name of the dedicated host. Required. :type host_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -23575,18 +21522,17 @@ def list_available_sizes( def prepare_request(next_link=None): if not next_link: - request = build_dedicated_hosts_list_available_sizes_request( + _request = build_dedicated_hosts_list_available_sizes_request( resource_group_name=resource_group_name, host_group_name=host_group_name, host_name=host_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_available_sizes.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23597,14 +21543,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DedicatedHostSizeListResult", pipeline_response) @@ -23614,11 +21560,11 @@ def extract_data(pipeline_response): return None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23630,10 +21576,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_available_sizes.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/hostGroups/{hostGroupName}/hosts/{hostName}/hostSizes" - } - class SshPublicKeysOperations: """ @@ -23660,7 +21602,6 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR """Lists all of the SSH public keys in the subscription. Use the nextLink property in the response to get the next page of SSH public keys. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -23684,15 +21625,14 @@ def list_by_subscription(self, **kwargs: Any) -> Iterable["_models.SshPublicKeyR def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_subscription_request( + _request = build_ssh_public_keys_list_by_subscription_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23703,14 +21643,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -23720,11 +21660,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23736,8 +21676,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/sshPublicKeys"} - @distributed_trace def list_by_resource_group( self, resource_group_name: str, **kwargs: Any @@ -23747,7 +21685,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either SshPublicKeyResource or the result of cls(response) :rtype: @@ -23771,16 +21708,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_ssh_public_keys_list_by_resource_group_request( + _request = build_ssh_public_keys_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -23791,14 +21727,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SshPublicKeysGroupListResult", pipeline_response) @@ -23808,11 +21744,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23824,10 +21760,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys" - } - @overload def create( self, @@ -23849,7 +21781,6 @@ def create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -23860,7 +21791,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -23872,11 +21803,10 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -23887,7 +21817,7 @@ def create( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyResource, IO], + parameters: Union[_models.SshPublicKeyResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Creates a new SSH public key resource. @@ -23897,12 +21827,8 @@ def create( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to create the SSH public key. Is either a - SshPublicKeyResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource or IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -23930,7 +21856,7 @@ def create( else: _json = self._serialize.body(parameters, "SshPublicKeyResource") - request = build_ssh_public_keys_create_request( + _request = build_ssh_public_keys_create_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -23938,16 +21864,15 @@ def create( content_type=content_type, json=_json, content=_content, - template_url=self.create.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -23967,10 +21892,6 @@ def create( return deserialized # type: ignore - create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } - @overload def update( self, @@ -23992,7 +21913,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -24003,7 +21923,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24015,11 +21935,10 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -24030,7 +21949,7 @@ def update( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Union[_models.SshPublicKeyUpdateResource, IO], + parameters: Union[_models.SshPublicKeyUpdateResource, IO[bytes]], **kwargs: Any ) -> _models.SshPublicKeyResource: """Updates a new SSH public key resource. @@ -24040,12 +21959,9 @@ def update( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to update the SSH public key. Is either a - SshPublicKeyUpdateResource type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyUpdateResource or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + SshPublicKeyUpdateResource type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyUpdateResource or + IO[bytes] :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -24073,7 +21989,7 @@ def update( else: _json = self._serialize.body(parameters, "SshPublicKeyUpdateResource") - request = build_ssh_public_keys_update_request( + _request = build_ssh_public_keys_update_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -24081,16 +21997,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24102,13 +22017,9 @@ def update( deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -24120,7 +22031,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -24139,21 +22049,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_ssh_public_keys_delete_request( + _request = build_ssh_public_keys_delete_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24163,11 +22072,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) -> _models.SshPublicKeyResource: @@ -24177,7 +22082,6 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) :type resource_group_name: str :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyResource or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyResource :raises ~azure.core.exceptions.HttpResponseError: @@ -24196,21 +22100,20 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.SshPublicKeyResource] = kwargs.pop("cls", None) - request = build_ssh_public_keys_get_request( + _request = build_ssh_public_keys_get_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24222,13 +22125,9 @@ def get(self, resource_group_name: str, ssh_public_key_name: str, **kwargs: Any) deserialized = self._deserialize("SshPublicKeyResource", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}" - } + return deserialized # type: ignore @overload def generate_key_pair( @@ -24253,7 +22152,6 @@ def generate_key_pair( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -24264,7 +22162,7 @@ def generate_key_pair( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Optional[IO] = None, + parameters: Optional[IO[bytes]] = None, *, content_type: str = "application/json", **kwargs: Any @@ -24278,11 +22176,10 @@ def generate_key_pair( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to generate the SSH public key. Default value is None. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -24293,7 +22190,7 @@ def generate_key_pair( self, resource_group_name: str, ssh_public_key_name: str, - parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO]] = None, + parameters: Optional[Union[_models.SshGenerateKeyPairInputParameters, IO[bytes]]] = None, **kwargs: Any ) -> _models.SshPublicKeyGenerateKeyPairResult: """Generates and returns a public/private key pair and populates the SSH public key resource with @@ -24305,13 +22202,9 @@ def generate_key_pair( :param ssh_public_key_name: The name of the SSH public key. Required. :type ssh_public_key_name: str :param parameters: Parameters supplied to generate the SSH public key. Is either a - SshGenerateKeyPairInputParameters type or a IO type. Default value is None. + SshGenerateKeyPairInputParameters type or a IO[bytes] type. Default value is None. :type parameters: ~azure.mgmt.compute.v2023_09_01.models.SshGenerateKeyPairInputParameters or - IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + IO[bytes] :return: SshPublicKeyGenerateKeyPairResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.SshPublicKeyGenerateKeyPairResult :raises ~azure.core.exceptions.HttpResponseError: @@ -24342,7 +22235,7 @@ def generate_key_pair( else: _json = None - request = build_ssh_public_keys_generate_key_pair_request( + _request = build_ssh_public_keys_generate_key_pair_request( resource_group_name=resource_group_name, ssh_public_key_name=ssh_public_key_name, subscription_id=self._config.subscription_id, @@ -24350,16 +22243,15 @@ def generate_key_pair( content_type=content_type, json=_json, content=_content, - template_url=self.generate_key_pair.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24371,13 +22263,9 @@ def generate_key_pair( deserialized = self._deserialize("SshPublicKeyGenerateKeyPairResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - generate_key_pair.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/sshPublicKeys/{sshPublicKeyName}/generateKeyPair" - } + return deserialized # type: ignore class ImagesOperations: @@ -24401,7 +22289,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -24426,7 +22314,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "Image") - request = build_images_create_or_update_request( + _request = build_images_create_or_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -24434,16 +22322,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24463,10 +22350,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_create_or_update( self, @@ -24488,14 +22371,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24506,7 +22381,7 @@ def begin_create_or_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24518,18 +22393,10 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24537,7 +22404,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO], **kwargs: Any + self, resource_group_name: str, image_name: str, parameters: Union[_models.Image, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Image]: """Create or update an image. @@ -24546,19 +22413,8 @@ def begin_create_or_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Create Image operation. Is either a Image type or - a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.Image or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.Image or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24589,7 +22445,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24599,20 +22455,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Image: error_map = { 401: ClientAuthenticationError, @@ -24637,7 +22495,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "ImageUpdate") - request = build_images_update_request( + _request = build_images_update_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, @@ -24645,16 +22503,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24674,10 +22531,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } - @overload def begin_update( self, @@ -24699,14 +22552,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24717,7 +22562,7 @@ def begin_update( self, resource_group_name: str, image_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -24729,18 +22574,10 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24748,7 +22585,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, image_name: str, parameters: Union[_models.ImageUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + image_name: str, + parameters: Union[_models.ImageUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Image]: """Update an image. @@ -24757,19 +22598,8 @@ def begin_update( :param image_name: The name of the image. Required. :type image_name: str :param parameters: Parameters supplied to the Update Image operation. Is either a ImageUpdate - type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ImageUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ImageUpdate or IO[bytes] :return: An instance of LROPoller that returns either Image or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -24800,7 +22630,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -24810,17 +22640,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Image].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[_models.Image]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, image_name: str, **kwargs: Any @@ -24839,21 +22667,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_images_delete_request( + _request = build_images_delete_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24863,11 +22690,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) -> LROPoller[None]: @@ -24877,14 +22700,6 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) :type resource_group_name: str :param image_name: The name of the image. Required. :type image_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -24911,7 +22726,7 @@ def begin_delete(self, resource_group_name: str, image_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -24920,17 +22735,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -24944,7 +22755,6 @@ def get( :type image_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Image or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.Image :raises ~azure.core.exceptions.HttpResponseError: @@ -24963,22 +22773,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.Image] = kwargs.pop("cls", None) - request = build_images_get_request( + _request = build_images_get_request( resource_group_name=resource_group_name, image_name=image_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -24990,13 +22799,9 @@ def get( deserialized = self._deserialize("Image", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images/{imageName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Image"]: @@ -25005,7 +22810,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -25027,16 +22831,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_images_list_by_resource_group_request( + _request = build_images_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25047,14 +22850,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -25064,11 +22867,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25080,16 +22883,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/images" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Image"]: """Gets the list of Images in the subscription. Use nextLink property in the response to get the next page of Images. Do this till nextLink is null to fetch all the Images. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Image or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.Image] :raises ~azure.core.exceptions.HttpResponseError: @@ -25111,15 +22909,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Image"]: def prepare_request(next_link=None): if not next_link: - request = build_images_list_request( + _request = build_images_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25130,14 +22927,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ImageListResult", pipeline_response) @@ -25147,11 +22944,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25163,8 +22960,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/images"} - class RestorePointCollectionsOperations: """ @@ -25210,7 +23005,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25221,7 +23015,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25236,11 +23030,10 @@ def create_or_update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25251,7 +23044,7 @@ def create_or_update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollection, IO], + parameters: Union[_models.RestorePointCollection, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to create or update the restore point collection. Please refer to @@ -25263,12 +23056,8 @@ def create_or_update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Create or Update restore point collection - operation. Is either a RestorePointCollection type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + operation. Is either a RestorePointCollection type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection or IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25296,7 +23085,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "RestorePointCollection") - request = build_restore_point_collections_create_or_update_request( + _request = build_restore_point_collections_create_or_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -25304,16 +23093,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25333,10 +23121,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } - @overload def update( self, @@ -25359,7 +23143,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25370,7 +23153,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25383,11 +23166,10 @@ def update( :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25398,7 +23180,7 @@ def update( self, resource_group_name: str, restore_point_collection_name: str, - parameters: Union[_models.RestorePointCollectionUpdate, IO], + parameters: Union[_models.RestorePointCollectionUpdate, IO[bytes]], **kwargs: Any ) -> _models.RestorePointCollection: """The operation to update the restore point collection. @@ -25408,12 +23190,9 @@ def update( :param restore_point_collection_name: The name of the restore point collection. Required. :type restore_point_collection_name: str :param parameters: Parameters supplied to the Update restore point collection operation. Is - either a RestorePointCollectionUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a RestorePointCollectionUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionUpdate or + IO[bytes] :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25441,7 +23220,7 @@ def update( else: _json = self._serialize.body(parameters, "RestorePointCollectionUpdate") - request = build_restore_point_collections_update_request( + _request = build_restore_point_collections_update_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, @@ -25449,16 +23228,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25470,13 +23248,9 @@ def update( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, **kwargs: Any @@ -25495,21 +23269,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_point_collections_delete_request( + _request = build_restore_point_collections_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25519,11 +23292,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -25536,14 +23305,6 @@ def begin_delete( :type resource_group_name: str :param restore_point_collection_name: The name of the Restore Point Collection. Required. :type restore_point_collection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -25570,7 +23331,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -25579,17 +23340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -25611,7 +23368,6 @@ def get( Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollectionExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePointCollection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePointCollection :raises ~azure.core.exceptions.HttpResponseError: @@ -25630,22 +23386,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RestorePointCollection] = kwargs.pop("cls", None) - request = build_restore_point_collections_get_request( + _request = build_restore_point_collections_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25657,13 +23412,9 @@ def get( deserialized = self._deserialize("RestorePointCollection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}" - } + return deserialized # type: ignore @distributed_trace def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: @@ -25671,7 +23422,6 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -25695,16 +23445,15 @@ def list(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Res def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_request( + _request = build_restore_point_collections_list_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25715,14 +23464,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -25732,11 +23481,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25748,17 +23497,12 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections" - } - @distributed_trace def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: """Gets the list of restore point collections in the subscription. Use nextLink property in the response to get the next page of restore point collections. Do this till nextLink is not null to fetch all the restore point collections. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RestorePointCollection or the result of cls(response) :rtype: @@ -25782,15 +23526,14 @@ def list_all(self, **kwargs: Any) -> Iterable["_models.RestorePointCollection"]: def prepare_request(next_link=None): if not next_link: - request = build_restore_point_collections_list_all_request( + _request = build_restore_point_collections_list_all_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_all.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -25801,14 +23544,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RestorePointCollectionListResult", pipeline_response) @@ -25818,11 +23561,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25834,8 +23577,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_all.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/restorePointCollections"} - class RestorePointsOperations: """ @@ -25862,7 +23603,7 @@ def _create_initial( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> _models.RestorePoint: error_map = { @@ -25888,7 +23629,7 @@ def _create_initial( else: _json = self._serialize.body(parameters, "RestorePoint") - request = build_restore_points_create_request( + _request = build_restore_points_create_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, @@ -25897,16 +23638,15 @@ def _create_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -25918,13 +23658,9 @@ def _create_initial( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _create_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore @overload def begin_create( @@ -25951,14 +23687,6 @@ def begin_create( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -25971,7 +23699,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -25986,18 +23714,10 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -26010,7 +23730,7 @@ def begin_create( resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, - parameters: Union[_models.RestorePoint, IO], + parameters: Union[_models.RestorePoint, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.RestorePoint]: """The operation to create the restore point. Updating properties of an existing restore point is @@ -26023,19 +23743,8 @@ def begin_create( :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str :param parameters: Parameters supplied to the Create restore point operation. Is either a - RestorePoint type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + RestorePoint type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint or IO[bytes] :return: An instance of LROPoller that returns either RestorePoint or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_09_01.models.RestorePoint] @@ -26068,7 +23777,7 @@ def begin_create( def get_long_running_output(pipeline_response): deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -26078,17 +23787,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.RestorePoint].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[_models.RestorePoint]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, restore_point_collection_name: str, restore_point_name: str, **kwargs: Any @@ -26107,22 +23814,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_restore_points_delete_request( + _request = build_restore_points_delete_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26132,11 +23838,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -26150,14 +23852,6 @@ def begin_delete( :type restore_point_collection_name: str :param restore_point_name: The name of the restore point. Required. :type restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -26185,7 +23879,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -26194,17 +23888,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -26227,7 +23917,6 @@ def get( :keyword expand: The expand expression to apply on the operation. 'InstanceView' retrieves information about the run-time state of a restore point. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.RestorePointExpandOptions - :keyword callable cls: A custom type or function that will be passed the direct response :return: RestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -26246,23 +23935,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RestorePoint] = kwargs.pop("cls", None) - request = build_restore_points_get_request( + _request = build_restore_points_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, restore_point_name=restore_point_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26274,13 +23962,9 @@ def get( deserialized = self._deserialize("RestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{restorePointName}" - } + return deserialized # type: ignore class CapacityReservationGroupsOperations: @@ -26326,7 +24010,6 @@ def create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26337,7 +24020,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26351,11 +24034,10 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26366,7 +24048,7 @@ def create_or_update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroup, IO], + parameters: Union[_models.CapacityReservationGroup, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to create or update a capacity reservation group. When updating a capacity @@ -26378,12 +24060,8 @@ def create_or_update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Create capacity reservation Group. Is either a - CapacityReservationGroup type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + CapacityReservationGroup type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup or IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26411,7 +24089,7 @@ def create_or_update( else: _json = self._serialize.body(parameters, "CapacityReservationGroup") - request = build_capacity_reservation_groups_create_or_update_request( + _request = build_capacity_reservation_groups_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -26419,16 +24097,15 @@ def create_or_update( content_type=content_type, json=_json, content=_content, - template_url=self.create_or_update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26448,10 +24125,6 @@ def create_or_update( return deserialized # type: ignore - create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } - @overload def update( self, @@ -26475,7 +24148,6 @@ def update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26486,7 +24158,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -26500,11 +24172,10 @@ def update( :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26515,7 +24186,7 @@ def update( self, resource_group_name: str, capacity_reservation_group_name: str, - parameters: Union[_models.CapacityReservationGroupUpdate, IO], + parameters: Union[_models.CapacityReservationGroupUpdate, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservationGroup: """The operation to update a capacity reservation group. When updating a capacity reservation @@ -26526,12 +24197,9 @@ def update( :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str :param parameters: Parameters supplied to the Update capacity reservation Group operation. Is - either a CapacityReservationGroupUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response + either a CapacityReservationGroupUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupUpdate or + IO[bytes] :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26559,7 +24227,7 @@ def update( else: _json = self._serialize.body(parameters, "CapacityReservationGroupUpdate") - request = build_capacity_reservation_groups_update_request( + _request = build_capacity_reservation_groups_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, @@ -26567,16 +24235,15 @@ def update( content_type=content_type, json=_json, content=_content, - template_url=self.update.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26588,13 +24255,9 @@ def update( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def delete( # pylint: disable=inconsistent-return-statements @@ -26609,7 +24272,6 @@ def delete( # pylint: disable=inconsistent-return-statements :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: None or the result of cls(response) :rtype: None :raises ~azure.core.exceptions.HttpResponseError: @@ -26628,21 +24290,20 @@ def delete( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_delete_request( + _request = build_capacity_reservation_groups_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.delete.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26652,11 +24313,7 @@ def delete( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def get( @@ -26680,7 +24337,6 @@ def get( None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroupInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservationGroup or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationGroup :raises ~azure.core.exceptions.HttpResponseError: @@ -26699,22 +24355,21 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.CapacityReservationGroup] = kwargs.pop("cls", None) - request = build_capacity_reservation_groups_get_request( + _request = build_capacity_reservation_groups_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26726,13 +24381,9 @@ def get( deserialized = self._deserialize("CapacityReservationGroup", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}" - } + return deserialized # type: ignore @distributed_trace def list_by_resource_group( @@ -26753,7 +24404,6 @@ def list_by_resource_group( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -26777,17 +24427,16 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_resource_group_request( + _request = build_capacity_reservation_groups_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26798,14 +24447,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -26815,11 +24464,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26831,10 +24480,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups" - } - @distributed_trace def list_by_subscription( self, *, expand: Optional[Union[str, _models.ExpandTypesForGetCapacityReservationGroups]] = None, **kwargs: Any @@ -26848,7 +24493,6 @@ def list_by_subscription( "virtualMachineScaleSetVMs/$ref" and "virtualMachines/$ref". Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.ExpandTypesForGetCapacityReservationGroups - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservationGroup or the result of cls(response) :rtype: @@ -26872,16 +24516,15 @@ def list_by_subscription( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservation_groups_list_by_subscription_request( + _request = build_capacity_reservation_groups_list_by_subscription_request( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_subscription.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -26892,14 +24535,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationGroupListResult", pipeline_response) @@ -26909,11 +24552,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -26925,10 +24568,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_subscription.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/capacityReservationGroups" - } - class CapacityReservationsOperations: """ @@ -26955,7 +24594,7 @@ def _create_or_update_initial( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> _models.CapacityReservation: error_map = { @@ -26981,7 +24620,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(parameters, "CapacityReservation") - request = build_capacity_reservations_create_or_update_request( + _request = build_capacity_reservations_create_or_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -26990,16 +24629,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27019,10 +24657,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } - @overload def begin_create_or_update( self, @@ -27049,14 +24683,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27070,7 +24696,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27086,18 +24712,10 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27111,7 +24729,7 @@ def begin_create_or_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservation, IO], + parameters: Union[_models.CapacityReservation, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to create or update a capacity reservation. Please note some properties can be @@ -27125,19 +24743,8 @@ def begin_create_or_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Create capacity reservation. Is either a - CapacityReservation type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + CapacityReservation type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27171,7 +24778,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27181,24 +24788,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> Optional[_models.CapacityReservation]: error_map = { @@ -27224,7 +24829,7 @@ def _update_initial( else: _json = self._serialize.body(parameters, "CapacityReservationUpdate") - request = build_capacity_reservations_update_request( + _request = build_capacity_reservations_update_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, @@ -27233,16 +24838,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27256,13 +24860,9 @@ def _update_initial( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -27288,14 +24888,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27309,7 +24901,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: IO, + parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -27323,18 +24915,10 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27348,7 +24932,7 @@ def begin_update( resource_group_name: str, capacity_reservation_group_name: str, capacity_reservation_name: str, - parameters: Union[_models.CapacityReservationUpdate, IO], + parameters: Union[_models.CapacityReservationUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.CapacityReservation]: """The operation to update a capacity reservation. @@ -27360,19 +24944,8 @@ def begin_update( :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str :param parameters: Parameters supplied to the Update capacity reservation operation. Is either - a CapacityReservationUpdate type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a CapacityReservationUpdate type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationUpdate or IO[bytes] :return: An instance of LROPoller that returns either CapacityReservation or the result of cls(response) :rtype: @@ -27406,7 +24979,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27416,17 +24989,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.CapacityReservation].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[_models.CapacityReservation]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, @@ -27449,22 +25020,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_capacity_reservations_delete_request( + _request = build_capacity_reservations_delete_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27474,11 +25044,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -27498,14 +25064,6 @@ def begin_delete( :type capacity_reservation_group_name: str :param capacity_reservation_name: The name of the capacity reservation. Required. :type capacity_reservation_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -27533,7 +25091,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -27542,17 +25100,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -27577,7 +25131,6 @@ def get( and can change outside of control plane operations. "instanceView" Default value is None. :paramtype expand: str or ~azure.mgmt.compute.v2023_09_01.models.CapacityReservationInstanceViewTypes - :keyword callable cls: A custom type or function that will be passed the direct response :return: CapacityReservation or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.CapacityReservation :raises ~azure.core.exceptions.HttpResponseError: @@ -27596,23 +25149,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.CapacityReservation] = kwargs.pop("cls", None) - request = build_capacity_reservations_get_request( + _request = build_capacity_reservations_get_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, capacity_reservation_name=capacity_reservation_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27624,13 +25176,9 @@ def get( deserialized = self._deserialize("CapacityReservation", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations/{capacityReservationName}" - } + return deserialized # type: ignore @distributed_trace def list_by_capacity_reservation_group( @@ -27643,7 +25191,6 @@ def list_by_capacity_reservation_group( :type resource_group_name: str :param capacity_reservation_group_name: The name of the capacity reservation group. Required. :type capacity_reservation_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either CapacityReservation or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_09_01.models.CapacityReservation] @@ -27666,17 +25213,16 @@ def list_by_capacity_reservation_group( def prepare_request(next_link=None): if not next_link: - request = build_capacity_reservations_list_by_capacity_reservation_group_request( + _request = build_capacity_reservations_list_by_capacity_reservation_group_request( resource_group_name=resource_group_name, capacity_reservation_group_name=capacity_reservation_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_capacity_reservation_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -27687,14 +25233,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("CapacityReservationListResult", pipeline_response) @@ -27704,11 +25250,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27720,10 +25266,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_capacity_reservation_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/capacityReservationGroups/{capacityReservationGroupName}/capacityReservations" - } - class LogAnalyticsOperations: """ @@ -27746,7 +25288,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _export_request_rate_by_interval_initial( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -27771,23 +25313,22 @@ def _export_request_rate_by_interval_initial( else: _json = self._serialize.body(parameters, "RequestRateByIntervalInput") - request = build_log_analytics_export_request_rate_by_interval_request( + _request = build_log_analytics_export_request_rate_by_interval_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_request_rate_by_interval_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -27801,13 +25342,9 @@ def _export_request_rate_by_interval_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_request_rate_by_interval_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return deserialized # type: ignore @overload def begin_export_request_rate_by_interval( @@ -27829,14 +25366,6 @@ def begin_export_request_rate_by_interval( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27846,7 +25375,7 @@ def begin_export_request_rate_by_interval( @overload def begin_export_request_rate_by_interval( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -27855,18 +25384,10 @@ def begin_export_request_rate_by_interval( :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27876,7 +25397,7 @@ def begin_export_request_rate_by_interval( @distributed_trace def begin_export_request_rate_by_interval( - self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.RequestRateByIntervalInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show Api requests made by this subscription in the given time window to show throttling activities. @@ -27884,19 +25405,9 @@ def begin_export_request_rate_by_interval( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getRequestRateByInterval Api. Is - either a RequestRateByIntervalInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RequestRateByIntervalInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a RequestRateByIntervalInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.RequestRateByIntervalInput or + IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -27928,7 +25439,7 @@ def begin_export_request_rate_by_interval( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -27940,20 +25451,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_request_rate_by_interval.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getRequestRateByInterval" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _export_throttled_requests_initial( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> Optional[_models.LogAnalyticsOperationResult]: error_map = { 401: ClientAuthenticationError, @@ -27978,23 +25487,22 @@ def _export_throttled_requests_initial( else: _json = self._serialize.body(parameters, "ThrottledRequestsInput") - request = build_log_analytics_export_throttled_requests_request( + _request = build_log_analytics_export_throttled_requests_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, content_type=content_type, json=_json, content=_content, - template_url=self._export_throttled_requests_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28008,13 +25516,9 @@ def _export_throttled_requests_initial( deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _export_throttled_requests_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return deserialized # type: ignore @overload def begin_export_throttled_requests( @@ -28035,14 +25539,6 @@ def begin_export_throttled_requests( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28052,7 +25548,7 @@ def begin_export_throttled_requests( @overload def begin_export_throttled_requests( - self, location: str, parameters: IO, *, content_type: str = "application/json", **kwargs: Any + self, location: str, parameters: IO[bytes], *, content_type: str = "application/json", **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28060,18 +25556,10 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Required. - :type parameters: IO + :type parameters: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28081,7 +25569,7 @@ def begin_export_throttled_requests( @distributed_trace def begin_export_throttled_requests( - self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO], **kwargs: Any + self, location: str, parameters: Union[_models.ThrottledRequestsInput, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.LogAnalyticsOperationResult]: """Export logs that show total throttled Api requests for this subscription in the given time window. @@ -28089,19 +25577,8 @@ def begin_export_throttled_requests( :param location: The location upon which virtual-machine-sizes is queried. Required. :type location: str :param parameters: Parameters supplied to the LogAnalytics getThrottledRequests Api. Is either - a ThrottledRequestsInput type or a IO type. Required. - :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ThrottledRequestsInput or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + a ThrottledRequestsInput type or a IO[bytes] type. Required. + :type parameters: ~azure.mgmt.compute.v2023_09_01.models.ThrottledRequestsInput or IO[bytes] :return: An instance of LROPoller that returns either LogAnalyticsOperationResult or the result of cls(response) :rtype: @@ -28133,7 +25610,7 @@ def begin_export_throttled_requests( def get_long_running_output(pipeline_response): deserialized = self._deserialize("LogAnalyticsOperationResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28145,17 +25622,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.LogAnalyticsOperationResult].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_export_throttled_requests.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/logAnalytics/apiAccess/getThrottledRequests" - } + return LROPoller[_models.LogAnalyticsOperationResult]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) class VirtualMachineRunCommandsOperations: @@ -28184,7 +25659,6 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu :param location: The location upon which run commands is queried. Required. :type location: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either RunCommandDocumentBase or the result of cls(response) :rtype: @@ -28208,16 +25682,15 @@ def list(self, location: str, **kwargs: Any) -> Iterable["_models.RunCommandDocu def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_request( + _request = build_virtual_machine_run_commands_list_request( location=location, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -28228,14 +25701,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("RunCommandListResult", pipeline_response) @@ -28245,11 +25718,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28261,10 +25734,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands" - } - @distributed_trace def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunCommandDocument: """Gets specific run command for a subscription in a location. @@ -28273,7 +25742,6 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma :type location: str :param command_id: The command id. Required. :type command_id: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: RunCommandDocument or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.RunCommandDocument :raises ~azure.core.exceptions.HttpResponseError: @@ -28292,21 +25760,20 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.RunCommandDocument] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_request( + _request = build_virtual_machine_run_commands_get_request( location=location, command_id=command_id, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28318,20 +25785,16 @@ def get(self, location: str, command_id: str, **kwargs: Any) -> _models.RunComma deserialized = self._deserialize("RunCommandDocument", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/locations/{location}/runCommands/{commandId}" - } + return deserialized # type: ignore def _create_or_update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28357,7 +25820,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_run_commands_create_or_update_request( + _request = build_virtual_machine_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -28366,16 +25829,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28395,10 +25857,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -28425,14 +25883,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28446,7 +25896,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28462,18 +25912,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28487,7 +25929,7 @@ def begin_create_or_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the run command. @@ -28500,19 +25942,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28546,7 +25977,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28556,24 +25987,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -28599,7 +26028,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_run_commands_update_request( + _request = build_virtual_machine_run_commands_update_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, @@ -28608,16 +26037,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28629,13 +26057,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -28663,14 +26087,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28684,7 +26100,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -28700,18 +26116,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28725,7 +26133,7 @@ def begin_update( resource_group_name: str, vm_name: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the run command. @@ -28738,19 +26146,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -28784,7 +26182,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -28794,17 +26192,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_name: str, run_command_name: str, **kwargs: Any @@ -28823,22 +26219,21 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_delete_request( + _request = build_virtual_machine_run_commands_delete_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28848,11 +26243,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -28867,14 +26258,6 @@ def begin_delete( :type vm_name: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -28902,7 +26285,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -28911,17 +26294,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get_by_virtual_machine( @@ -28943,7 +26322,6 @@ def get_by_virtual_machine( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -28962,23 +26340,22 @@ def get_by_virtual_machine( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_run_commands_get_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_get_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, run_command_name=run_command_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -28990,13 +26367,9 @@ def get_by_virtual_machine( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list_by_virtual_machine( @@ -29010,7 +26383,6 @@ def list_by_virtual_machine( :type vm_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29034,18 +26406,17 @@ def list_by_virtual_machine( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_run_commands_list_by_virtual_machine_request( + _request = build_virtual_machine_run_commands_list_by_virtual_machine_request( resource_group_name=resource_group_name, vm_name=vm_name, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list_by_virtual_machine.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29056,14 +26427,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -29073,11 +26444,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29089,12 +26460,8 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_virtual_machine.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachines/{vmName}/runCommands" - } - -class VirtualMachineScaleSetVMRunCommandsOperations: +class VirtualMachineScaleSetVMRunCommandsOperations: # pylint: disable=name-too-long """ .. warning:: **DO NOT** instantiate this class directly. @@ -29120,7 +26487,7 @@ def _create_or_update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29146,7 +26513,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommand") - request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_create_or_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29156,16 +26523,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29185,10 +26551,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } - @overload def begin_create_or_update( self, @@ -29217,14 +26579,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29239,7 +26593,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29256,18 +26610,10 @@ def begin_create_or_update( :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29282,7 +26628,7 @@ def begin_create_or_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommand, IO], + run_command: Union[_models.VirtualMachineRunCommand, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to create or update the VMSS VM run command. @@ -29296,19 +26642,8 @@ def begin_create_or_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Create Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommand type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommand type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand or IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29343,7 +26678,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29353,17 +26688,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, @@ -29371,7 +26704,7 @@ def _update_initial( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> _models.VirtualMachineRunCommand: error_map = { @@ -29397,7 +26730,7 @@ def _update_initial( else: _json = self._serialize.body(run_command, "VirtualMachineRunCommandUpdate") - request = build_virtual_machine_scale_set_vm_run_commands_update_request( + _request = build_virtual_machine_scale_set_vm_run_commands_update_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29407,16 +26740,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29428,13 +26760,9 @@ def _update_initial( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @overload def begin_update( @@ -29464,14 +26792,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29486,7 +26806,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: IO, + run_command: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -29503,18 +26823,10 @@ def begin_update( :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Required. - :type run_command: IO + :type run_command: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Known values are: 'application/json', 'text/json'. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29529,7 +26841,7 @@ def begin_update( vm_scale_set_name: str, instance_id: str, run_command_name: str, - run_command: Union[_models.VirtualMachineRunCommandUpdate, IO], + run_command: Union[_models.VirtualMachineRunCommandUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.VirtualMachineRunCommand]: """The operation to update the VMSS VM run command. @@ -29543,19 +26855,9 @@ def begin_update( :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str :param run_command: Parameters supplied to the Update Virtual Machine RunCommand operation. Is - either a VirtualMachineRunCommandUpdate type or a IO type. Required. - :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json', - 'text/json'. Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a VirtualMachineRunCommandUpdate type or a IO[bytes] type. Required. + :type run_command: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommandUpdate or + IO[bytes] :return: An instance of LROPoller that returns either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29590,7 +26892,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -29600,17 +26902,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.VirtualMachineRunCommand].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[_models.VirtualMachineRunCommand]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, vm_scale_set_name: str, instance_id: str, run_command_name: str, **kwargs: Any @@ -29629,23 +26929,22 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_delete_request( + _request = build_virtual_machine_scale_set_vm_run_commands_delete_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, run_command_name=run_command_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29655,11 +26954,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete( @@ -29675,14 +26970,6 @@ def begin_delete( :type instance_id: str :param run_command_name: The name of the virtual machine run command. Required. :type run_command_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -29711,7 +26998,7 @@ def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -29720,17 +27007,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def get( @@ -29755,7 +27038,6 @@ def get( :type run_command_name: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: VirtualMachineRunCommand or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_09_01.models.VirtualMachineRunCommand :raises ~azure.core.exceptions.HttpResponseError: @@ -29774,7 +27056,7 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-09-01")) cls: ClsType[_models.VirtualMachineRunCommand] = kwargs.pop("cls", None) - request = build_virtual_machine_scale_set_vm_run_commands_get_request( + _request = build_virtual_machine_scale_set_vm_run_commands_get_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, @@ -29782,16 +27064,15 @@ def get( subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29803,13 +27084,9 @@ def get( deserialized = self._deserialize("VirtualMachineRunCommand", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands/{runCommandName}" - } + return deserialized # type: ignore @distributed_trace def list( @@ -29831,7 +27108,6 @@ def list( :type instance_id: str :keyword expand: The expand expression to apply on the operation. Default value is None. :paramtype expand: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either VirtualMachineRunCommand or the result of cls(response) :rtype: @@ -29855,19 +27131,18 @@ def list( def prepare_request(next_link=None): if not next_link: - request = build_virtual_machine_scale_set_vm_run_commands_list_request( + _request = build_virtual_machine_scale_set_vm_run_commands_list_request( resource_group_name=resource_group_name, vm_scale_set_name=vm_scale_set_name, instance_id=instance_id, subscription_id=self._config.subscription_id, expand=expand, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -29878,14 +27153,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("VirtualMachineRunCommandsListResult", pipeline_response) @@ -29895,11 +27170,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -29910,7 +27185,3 @@ def get_next(next_link=None): return pipeline_response return ItemPaged(get_next, extract_data) - - list.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/virtualMachines/{instanceId}/runCommands" - } diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_compute_management_client.py index 93849cf80689..0b015c390c97 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, TYPE_CHECKING +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 .._serialization import Deserializer, Serializer @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: ARMPipelineClient = ARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,7 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-10-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: + 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 @@ -107,7 +127,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse: request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore def close(self) -> None: self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_configuration.py index 6a0e578f02a8..6e2d42f8ffc1 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMChallengeAuthenticationPolicy, ARMHttpLoggingPolicy @@ -19,7 +18,7 @@ from azure.core.credentials import TokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-10-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_metadata.json b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_metadata.json index ec2e30af6882..e07c10604bf0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_metadata.json +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/_metadata.json @@ -8,10 +8,10 @@ "host_value": "\"https://management.azure.com\"", "parameterized_host_template": null, "azure_arm": true, - "has_lro_operations": true, + "has_public_lro_operations": true, "client_side_validation": false, - "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"azurecore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"ARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \".._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"conditional\": {\"stdlib\": {\"typing\": [\"Any\", \"Optional\"]}}, \"regular\": {\"sdkcore\": {\"azure.mgmt.core\": [\"AsyncARMPipelineClient\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"AsyncARMAutoResourceProviderRegistrationPolicy\"], \"azure.profiles\": [\"KnownProfiles\", \"ProfileDefinition\"], \"azure.profiles.multiapiclient\": [\"MultiApiClientMixin\"]}, \"local\": {\"._configuration\": [\"ComputeManagementClientConfiguration\"], \"..._serialization\": [\"Deserializer\", \"Serializer\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "global_parameters": { "sync": { @@ -101,8 +101,8 @@ "credential_scopes": ["https://management.azure.com/.default"], "credential_call_sync": "ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", "credential_call_async": "AsyncARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)", - "sync_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", - "async_imports": "{\"regular\": {\"azurecore\": {\"azure.core.configuration\": [\"Configuration\"], \"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"azurecore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" + "sync_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMChallengeAuthenticationPolicy\", \"ARMHttpLoggingPolicy\"]}, \"local\": {\"._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials\": [\"TokenCredential\"]}}}", + "async_imports": "{\"regular\": {\"sdkcore\": {\"azure.core.pipeline\": [\"policies\"], \"azure.mgmt.core.policies\": [\"ARMHttpLoggingPolicy\", \"AsyncARMChallengeAuthenticationPolicy\"]}, \"local\": {\".._version\": [\"VERSION\"]}}, \"conditional\": {\"stdlib\": {\"typing\": [\"Any\"]}}, \"typing\": {\"sdkcore\": {\"azure.core.credentials_async\": [\"AsyncTokenCredential\"]}}}" }, "operation_groups": { "disks": "DisksOperations", diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_compute_management_client.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_compute_management_client.py index 8daecbd2ce63..e2a08b2e15d5 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_compute_management_client.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_compute_management_client.py @@ -9,8 +9,10 @@ from copy import deepcopy from typing import Any, Awaitable, TYPE_CHECKING +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 @@ -67,7 +69,25 @@ def __init__( self._config = ComputeManagementClientConfiguration( credential=credential, subscription_id=subscription_id, **kwargs ) - self._client: AsyncARMPipelineClient = AsyncARMPipelineClient(base_url=base_url, config=self._config, **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) @@ -87,7 +107,9 @@ def __init__( self._client, self._config, self._serialize, self._deserialize, "2023-10-02" ) - def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncHttpResponse]: + 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 @@ -107,7 +129,7 @@ def _send_request(self, request: HttpRequest, **kwargs: Any) -> Awaitable[AsyncH request_copy = deepcopy(request) request_copy.url = self._client.format_url(request_copy.url) - return self._client.send_request(request_copy, **kwargs) + return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore async def close(self) -> None: await self._client.close() diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_configuration.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_configuration.py index 97db33404d86..7f66a619234c 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_configuration.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/_configuration.py @@ -8,7 +8,6 @@ from typing import Any, TYPE_CHECKING -from azure.core.configuration import Configuration from azure.core.pipeline import policies from azure.mgmt.core.policies import ARMHttpLoggingPolicy, AsyncARMChallengeAuthenticationPolicy @@ -19,7 +18,7 @@ from azure.core.credentials_async import AsyncTokenCredential -class ComputeManagementClientConfiguration(Configuration): # pylint: disable=too-many-instance-attributes +class ComputeManagementClientConfiguration: # pylint: disable=too-many-instance-attributes,name-too-long """Configuration for ComputeManagementClient. Note that all parameters used to create this instance are saved as instance @@ -36,7 +35,6 @@ class ComputeManagementClientConfiguration(Configuration): # pylint: disable=to """ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None: - super(ComputeManagementClientConfiguration, self).__init__(**kwargs) api_version: str = kwargs.pop("api_version", "2023-10-02") if credential is None: @@ -49,6 +47,7 @@ def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **k self.api_version = api_version self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"]) kwargs.setdefault("sdk_moniker", "mgmt-compute/{}".format(VERSION)) + self.polling_interval = kwargs.get("polling_interval", 30) self._configure(**kwargs) def _configure(self, **kwargs: Any) -> None: @@ -57,9 +56,9 @@ def _configure(self, **kwargs: Any) -> None: 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.retry_policy = kwargs.get("retry_policy") or policies.AsyncRetryPolicy(**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( diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/operations/_operations.py index 5daa414f1cf4..ac27afd836b6 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/aio/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -97,7 +97,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -122,7 +122,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -130,16 +130,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -159,10 +158,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_create_or_update( self, @@ -186,14 +181,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -204,7 +191,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -218,18 +205,10 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -237,7 +216,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Creates or updates a disk. @@ -248,19 +227,8 @@ async def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_10_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_10_02.models.Disk or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -291,7 +259,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -301,20 +269,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -339,7 +305,7 @@ async def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -347,16 +313,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -376,10 +341,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload async def begin_update( self, @@ -403,14 +364,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -421,7 +374,7 @@ async def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -435,18 +388,10 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -454,7 +399,7 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Disk]: """Updates (patches) a disk. @@ -465,19 +410,8 @@ async def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_10_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_10_02.models.DiskUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -508,7 +442,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -518,17 +452,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -540,7 +472,6 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -559,21 +490,20 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -585,13 +515,9 @@ async def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -610,21 +536,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -634,11 +559,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -650,14 +571,6 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -684,7 +597,7 @@ async def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -693,17 +606,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Disk"]: @@ -711,7 +620,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -733,16 +641,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -753,14 +660,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -770,11 +677,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -786,15 +693,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -816,15 +718,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -835,14 +736,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -852,11 +753,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -868,13 +769,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - async def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -900,7 +799,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -908,16 +807,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -931,13 +829,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -963,14 +857,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -982,7 +868,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -997,18 +883,10 @@ async def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -1020,7 +898,7 @@ async def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a disk. @@ -1032,19 +910,8 @@ async def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -1076,7 +943,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1088,17 +955,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1117,21 +982,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1141,11 +1005,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -1159,14 +1019,6 @@ async def begin_revoke_access( after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1193,7 +1045,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -1204,17 +1056,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -1238,7 +1086,11 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -1263,7 +1115,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1271,16 +1123,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1300,10 +1151,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_create_or_update( self, @@ -1328,14 +1175,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1347,7 +1186,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1362,18 +1201,10 @@ async def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1382,7 +1213,11 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -1393,19 +1228,8 @@ async def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1437,7 +1261,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1447,23 +1271,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -1489,7 +1311,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -1497,16 +1319,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1526,10 +1347,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload async def begin_update( self, @@ -1554,14 +1371,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1573,7 +1382,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1588,18 +1397,10 @@ async def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1611,7 +1412,7 @@ async def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -1623,19 +1424,8 @@ async def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1667,7 +1457,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1677,17 +1467,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -1699,7 +1487,6 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -1718,21 +1505,20 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1744,13 +1530,9 @@ async def get(self, resource_group_name: str, disk_access_name: str, **kwargs: A deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -1769,21 +1551,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1793,11 +1574,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -1811,14 +1588,6 @@ async def begin_delete( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1845,7 +1614,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -1854,17 +1623,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: @@ -1872,7 +1637,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1895,16 +1659,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1915,14 +1678,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -1932,11 +1695,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1948,15 +1711,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -1979,15 +1737,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1998,14 +1755,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -2015,11 +1772,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2031,8 +1788,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace_async async def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2045,7 +1800,6 @@ async def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -2064,21 +1818,20 @@ async def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2090,20 +1843,16 @@ async def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - async def _update_a_private_endpoint_connection_initial( + async def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -2129,7 +1878,7 @@ async def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -2138,16 +1887,15 @@ async def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2167,12 +1915,8 @@ async def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -2200,14 +1944,6 @@ async def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2216,12 +1952,12 @@ async def begin_update_a_private_endpoint_connection( """ @overload - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2239,18 +1975,10 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2259,12 +1987,12 @@ async def begin_update_a_private_endpoint_connection( """ @distributed_trace_async - async def begin_update_a_private_endpoint_connection( + async def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -2280,20 +2008,9 @@ async def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of AsyncLROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2327,7 +2044,7 @@ async def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2337,17 +2054,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get_a_private_endpoint_connection( @@ -2363,7 +2078,6 @@ async def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -2382,22 +2096,21 @@ async def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2409,15 +2122,11 @@ async def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -2434,22 +2143,21 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2459,14 +2167,10 @@ async def _delete_a_private_endpoint_connection_initial( # pylint: disable=inco raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async - async def begin_delete_a_private_endpoint_connection( + async def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> AsyncLROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -2479,14 +2183,6 @@ async def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2514,7 +2210,7 @@ async def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -2523,17 +2219,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -2547,7 +2239,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -2571,17 +2262,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2592,14 +2282,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -2609,11 +2299,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2625,10 +2315,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -2654,7 +2340,7 @@ async def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2680,7 +2366,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2688,16 +2374,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2717,10 +2402,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_create_or_update( self, @@ -2745,14 +2426,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2765,7 +2438,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2780,18 +2453,10 @@ async def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2804,7 +2469,7 @@ async def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -2816,19 +2481,9 @@ async def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2861,7 +2516,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2871,23 +2526,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -2913,7 +2566,7 @@ async def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -2921,16 +2574,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2950,10 +2602,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload async def begin_update( self, @@ -2978,14 +2626,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -2998,7 +2638,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3013,18 +2653,10 @@ async def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3037,7 +2669,7 @@ async def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -3049,19 +2681,10 @@ async def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of AsyncLROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: @@ -3094,7 +2717,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3104,17 +2727,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get( @@ -3128,7 +2749,6 @@ async def get( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -3147,21 +2767,20 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3173,13 +2792,9 @@ async def get( deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3198,21 +2813,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3222,11 +2836,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete( @@ -3240,14 +2850,6 @@ async def begin_delete( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3274,7 +2876,7 @@ async def begin_delete( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -3283,17 +2885,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group( @@ -3303,7 +2901,6 @@ def list_by_resource_group( :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -3326,16 +2923,15 @@ def list_by_resource_group( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3346,14 +2942,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3363,11 +2959,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3379,15 +2975,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -3410,15 +3001,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3429,14 +3019,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -3446,11 +3036,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3462,8 +3052,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -3476,7 +3064,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -3498,17 +3085,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3519,14 +3105,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -3536,11 +3122,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3552,10 +3138,6 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -3598,7 +3180,6 @@ async def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -3617,23 +3198,22 @@ async def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3645,13 +3225,9 @@ async def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -3667,7 +3243,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskRestorePoint] @@ -3690,18 +3265,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3712,14 +3286,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -3729,11 +3303,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3745,17 +3319,13 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - async def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -3781,7 +3351,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -3791,16 +3361,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3814,13 +3383,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -3852,14 +3417,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -3873,7 +3430,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3892,18 +3449,10 @@ async def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -3917,7 +3466,7 @@ async def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -3933,19 +3482,8 @@ async def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -3979,7 +3517,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3991,17 +3529,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -4025,23 +3561,22 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4051,11 +3586,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -4078,14 +3609,6 @@ async def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4114,7 +3637,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -4125,17 +3648,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -4159,7 +3678,7 @@ def __init__(self, *args, **kwargs) -> None: self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") async def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4184,7 +3703,7 @@ async def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4192,16 +3711,15 @@ async def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4221,10 +3739,6 @@ async def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_create_or_update( self, @@ -4248,14 +3762,6 @@ async def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4267,7 +3773,7 @@ async def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4281,18 +3787,10 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4301,7 +3799,7 @@ async def begin_create_or_update( @distributed_trace_async async def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -4312,19 +3810,8 @@ async def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.Snapshot or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4356,7 +3843,7 @@ async def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4366,20 +3853,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -4404,7 +3893,7 @@ async def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4412,16 +3901,15 @@ async def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4441,10 +3929,6 @@ async def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload async def begin_update( self, @@ -4469,14 +3953,6 @@ async def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4488,7 +3964,7 @@ async def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4503,18 +3979,10 @@ async def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4523,7 +3991,11 @@ async def begin_update( @distributed_trace_async async def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> AsyncLROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -4534,19 +4006,8 @@ async def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.SnapshotUpdate or IO[bytes] :return: An instance of AsyncLROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4578,7 +4039,7 @@ async def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4588,17 +4049,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace_async async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -4610,7 +4069,6 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -4629,21 +4087,20 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4655,13 +4112,9 @@ async def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore async def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -4680,21 +4133,20 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4704,11 +4156,7 @@ async def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> AsyncLROPoller[None]: @@ -4720,14 +4168,6 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4754,7 +4194,7 @@ async def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwa def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast(AsyncPollingMethod, AsyncARMPolling(lro_delay, **kwargs)) @@ -4763,17 +4203,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: @@ -4781,7 +4217,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4804,16 +4239,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Asy def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4824,14 +4258,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4841,11 +4275,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4857,15 +4291,10 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2023_10_02.models.Snapshot] @@ -4888,15 +4317,14 @@ def list(self, **kwargs: Any) -> AsyncIterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4907,14 +4335,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request async def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -4924,11 +4352,11 @@ async def extract_data(pipeline_response): return deserialized.next_link or None, AsyncList(list_of_elem) async def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4940,13 +4368,11 @@ async def get_next(next_link=None): return AsyncItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - async def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4972,7 +4398,7 @@ async def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -4980,16 +4406,15 @@ async def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5003,13 +4428,9 @@ async def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload async def begin_grant_access( @@ -5035,14 +4456,6 @@ async def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -5054,7 +4467,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5069,18 +4482,10 @@ async def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -5092,7 +4497,7 @@ async def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> AsyncLROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -5104,19 +4509,8 @@ async def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of AsyncLROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] @@ -5148,7 +4542,7 @@ async def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5160,17 +4554,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_token( + return AsyncLROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return AsyncLROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) async def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5189,21 +4581,20 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = await self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5213,11 +4604,7 @@ async def _revoke_access_initial( # pylint: disable=inconsistent-return-stateme raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace_async async def begin_revoke_access( @@ -5231,14 +4618,6 @@ async def begin_revoke_access( after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for - this operation to not poll, or pass in your own initialized polling object for a personal - polling strategy. - :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5265,7 +4644,7 @@ async def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: AsyncPollingMethod = cast( @@ -5276,14 +4655,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return AsyncLROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_compute_management_client_enums.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_compute_management_client_enums.py index 58b8fcf9ae7b..7468ddbac5ea 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_compute_management_client_enums.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_compute_management_client_enums.py @@ -32,7 +32,7 @@ class CopyCompletionErrorReason(str, Enum, metaclass=CaseInsensitiveEnumMeta): COPY_SOURCE_NOT_FOUND = "CopySourceNotFound" """Indicates that the source snapshot was deleted while the background copy of the resource - #: created via CopyStart operation was in progress.""" + created via CopyStart operation was in progress.""" class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -40,8 +40,8 @@ class DataAccessAuthMode(str, Enum, metaclass=CaseInsensitiveEnumMeta): AZURE_ACTIVE_DIRECTORY = "AzureActiveDirectory" """When export/upload URL is used, the system checks if the user has an identity in Azure Active - #: Directory and has necessary permissions to export/upload the data. Please refer to - #: aka.ms/DisksAzureADAuth.""" + Directory and has necessary permissions to export/upload the data. Please refer to + aka.ms/DisksAzureADAuth.""" NONE = "None" """No additional authentication would be performed when accessing export/upload URL.""" @@ -55,33 +55,30 @@ class DiskCreateOption(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Disk will be attached to a VM.""" FROM_IMAGE = "FromImage" """Create a new disk from a platform image specified by the given imageReference or - #: galleryImageReference.""" - IMPORT = "Import" + galleryImageReference.""" + IMPORT_ENUM = "Import" """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" + by storageAccountId.""" COPY = "Copy" """Create a new disk or snapshot by copying from a disk or snapshot specified by the given - #: sourceResourceId.""" + sourceResourceId.""" RESTORE = "Restore" """Create a new disk by copying from a backup recovery point.""" UPLOAD = "Upload" """Create a new disk by obtaining a write token and using it to directly upload the contents of - #: the disk.""" + the disk.""" COPY_START = "CopyStart" """Create a new disk by using a deep copy process, where the resource creation is considered - #: complete only after all data has been copied from the source.""" + complete only after all data has been copied from the source.""" IMPORT_SECURE = "ImportSecure" """Similar to Import create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk by importing additional blob for VM guest state specified by securityDataUri in storage - #: account specified by storageAccountId""" + disk by importing additional blob for VM guest state specified by securityDataUri in storage + account specified by storageAccountId""" UPLOAD_PREPARED_SECURE = "UploadPreparedSecure" """Similar to Upload create option. Create a new Trusted Launch VM or Confidential VM supported - #: disk and upload using write token in both disk and VM guest state""" + disk and upload using write token in both disk and VM guest state""" COPY_FROM_SAN_SNAPSHOT = "CopyFromSanSnapshot" """Create a new disk by exporting from elastic san volume snapshot""" - IMPORT_ENUM = "Import" - """Create a disk by importing from a blob specified by a sourceUri in a storage account specified - #: by storageAccountId.""" class DiskEncryptionSetIdentityType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -102,10 +99,10 @@ class DiskEncryptionSetType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Resource using diskEncryptionSet would be encrypted at rest with Customer managed key that can - #: be changed and revoked by a customer.""" + be changed and revoked by a customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Resource using diskEncryptionSet would be encrypted at rest with two layers of encryption. One - #: of the keys is Customer managed and the other key is Platform managed.""" + of the keys is Customer managed and the other key is Platform managed.""" CONFIDENTIAL_VM_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVmEncryptedWithCustomerKey" """Confidential VM supported disk and VM guest state would be encrypted with customer managed key.""" @@ -115,20 +112,20 @@ class DiskSecurityTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): TRUSTED_LAUNCH = "TrustedLaunch" """Trusted Launch provides security features such as secure boot and virtual Trusted Platform - #: Module (vTPM)""" + Module (vTPM)""" CONFIDENTIAL_VM_VMGUEST_STATE_ONLY_ENCRYPTED_WITH_PLATFORM_KEY = ( "ConfidentialVM_VMGuestStateOnlyEncryptedWithPlatformKey" ) """Indicates Confidential VM disk with only VM guest state encrypted""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_PLATFORM_KEY = "ConfidentialVM_DiskEncryptedWithPlatformKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a platform - #: managed key""" + managed key""" CONFIDENTIAL_VM_DISK_ENCRYPTED_WITH_CUSTOMER_KEY = "ConfidentialVM_DiskEncryptedWithCustomerKey" """Indicates Confidential VM disk with both OS disk and VM guest state encrypted with a customer - #: managed key""" + managed key""" CONFIDENTIAL_VM_NON_PERSISTED_TPM = "ConfidentialVM_NonPersistedTPM" """Indicates Confidential VM disk with a ephemeral vTPM. vTPM state is not persisted across VM - #: reboots.""" + reboots.""" class DiskState(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -161,19 +158,19 @@ class DiskStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): """Premium SSD locally redundant storage. Best for production and performance sensitive workloads.""" STANDARD_SSD_LRS = "StandardSSD_LRS" """Standard SSD locally redundant storage. Best for web servers, lightly used enterprise - #: applications and dev/test.""" + applications and dev/test.""" ULTRA_SSD_LRS = "UltraSSD_LRS" """Ultra SSD locally redundant storage. Best for IO-intensive workloads such as SAP HANA, top tier - #: databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" + databases (for example, SQL, Oracle), and other transaction-heavy workloads.""" PREMIUM_ZRS = "Premium_ZRS" """Premium SSD zone redundant storage. Best for the production workloads that need storage - #: resiliency against zone failures.""" + resiliency against zone failures.""" STANDARD_SSD_ZRS = "StandardSSD_ZRS" """Standard SSD zone redundant storage. Best for web servers, lightly used enterprise applications - #: and dev/test that need storage resiliency against zone failures.""" + and dev/test that need storage resiliency against zone failures.""" PREMIUM_V2_LRS = "PremiumV2_LRS" """Premium SSD v2 locally redundant storage. Best for production and performance-sensitive - #: workloads that consistently require low latency and high IOPS and throughput.""" + workloads that consistently require low latency and high IOPS and throughput.""" class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -181,13 +178,13 @@ class EncryptionType(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENCRYPTION_AT_REST_WITH_PLATFORM_KEY = "EncryptionAtRestWithPlatformKey" """Disk is encrypted at rest with Platform managed key. It is the default encryption type. This is - #: not a valid encryption type for disk encryption sets.""" + not a valid encryption type for disk encryption sets.""" ENCRYPTION_AT_REST_WITH_CUSTOMER_KEY = "EncryptionAtRestWithCustomerKey" """Disk is encrypted at rest with Customer managed key that can be changed and revoked by a - #: customer.""" + customer.""" ENCRYPTION_AT_REST_WITH_PLATFORM_AND_CUSTOMER_KEYS = "EncryptionAtRestWithPlatformAndCustomerKeys" """Disk is encrypted at rest with 2 layers of encryption. One of the keys is Customer managed and - #: the other key is Platform managed.""" + the other key is Platform managed.""" class ExtendedLocationTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): @@ -261,12 +258,12 @@ class PublicNetworkAccess(str, Enum, metaclass=CaseInsensitiveEnumMeta): ENABLED = "Enabled" """You can generate a SAS URI to access the underlying data of the disk publicly on the internet - #: when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from - #: your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + when NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from + your trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" DISABLED = "Disabled" """You cannot access the underlying data of the disk publicly on the internet even when - #: NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your - #: trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" + NetworkAccessPolicy is set to AllowAll. You can access the data via the SAS URI only from your + trusted Azure VNET when NetworkAccessPolicy is set to AllowPrivate.""" class SnapshotStorageAccountTypes(str, Enum, metaclass=CaseInsensitiveEnumMeta): diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_models_py3.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_models_py3.py index 960fa87c1994..23f8715e2dc0 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_models_py3.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/models/_models_py3.py @@ -135,7 +135,7 @@ class CopyCompletionError(_serialization.Model): """Indicates the error details if the background copy of a resource created via the CopyStart operation fails. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar error_code: Indicates the error code if the background copy of a resource created via the CopyStart operation fails. Required. "CopySourceNotFound" @@ -176,11 +176,11 @@ class CreationData(_serialization.Model): # pylint: disable=too-many-instance-a Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", "CopyStart", - "ImportSecure", "UploadPreparedSecure", "CopyFromSanSnapshot", and "Import". + "ImportSecure", "UploadPreparedSecure", and "CopyFromSanSnapshot". :vartype create_option: str or ~azure.mgmt.compute.v2023_10_02.models.DiskCreateOption :ivar storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -265,7 +265,7 @@ def __init__( """ :keyword create_option: This enumerates the possible sources of a disk's creation. Required. Known values are: "Empty", "Attach", "FromImage", "Import", "Copy", "Restore", "Upload", - "CopyStart", "ImportSecure", "UploadPreparedSecure", "CopyFromSanSnapshot", and "Import". + "CopyStart", "ImportSecure", "UploadPreparedSecure", and "CopyFromSanSnapshot". :paramtype create_option: str or ~azure.mgmt.compute.v2023_10_02.models.DiskCreateOption :keyword storage_account_id: Required if createOption is Import. The Azure Resource Manager identifier of the storage account containing the blob to import as a disk. @@ -326,7 +326,7 @@ class Resource(_serialization.Model): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -375,7 +375,7 @@ class Disk(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -744,7 +744,7 @@ class DiskAccess(Resource): Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -821,7 +821,7 @@ def __init__( class DiskAccessList(_serialization.Model): """The List disk access operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk access resources. Required. :vartype value: list[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] @@ -877,7 +877,7 @@ class DiskEncryptionSet(Resource): # pylint: disable=too-many-instance-attribut Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -998,7 +998,7 @@ def __init__( class DiskEncryptionSetList(_serialization.Model): """The List disk encryption set operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk encryption sets. Required. :vartype value: list[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -1109,7 +1109,7 @@ def __init__( class DiskList(_serialization.Model): """The List Disks operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disks. Required. :vartype value: list[~azure.mgmt.compute.v2023_10_02.models.Disk] @@ -1334,7 +1334,7 @@ def __init__( class DiskRestorePointList(_serialization.Model): """The List Disk Restore Points operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of disk restore points. Required. :vartype value: list[~azure.mgmt.compute.v2023_10_02.models.DiskRestorePoint] @@ -1737,7 +1737,7 @@ class EncryptionSetIdentity(_serialization.Model): :vartype tenant_id: str :ivar user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :vartype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_10_02.models.UserAssignedIdentitiesValue] """ @@ -1771,7 +1771,7 @@ def __init__( :keyword user_assigned_identities: The list of user identities associated with the disk encryption set. The user identity dictionary key references will be ARM resource ids in the form: - '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. + '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.ManagedIdentity/userAssignedIdentities/{identityName}'. # pylint: disable=line-too-long :paramtype user_assigned_identities: dict[str, ~azure.mgmt.compute.v2023_10_02.models.UserAssignedIdentitiesValue] """ @@ -1785,7 +1785,7 @@ def __init__( class EncryptionSettingsCollection(_serialization.Model): """Encryption settings for disk or snapshot. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar enabled: Set this flag to true and provide DiskEncryptionKey and optional KeyEncryptionKey to enable encryption. Set this flag to false and remove DiskEncryptionKey and @@ -1909,7 +1909,7 @@ def __init__( class GrantAccessData(_serialization.Model): """Data used for requesting a SAS. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar access: Required. Known values are: "None", "Read", and "Write". :vartype access: str or ~azure.mgmt.compute.v2023_10_02.models.AccessLevel @@ -2048,7 +2048,7 @@ def __init__( class KeyForDiskEncryptionSet(_serialization.Model): """Key Vault Key Url to be used for server side encryption of Managed Disks and Snapshots. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. This property is optional and cannot be used if the KeyVault subscription is not the same as the Disk Encryption @@ -2087,7 +2087,7 @@ class KeyVaultAndKeyReference(_serialization.Model): """Key Vault Key Url and vault id of KeK, KeK is optional and when provided is used to unwrap the encryptionKey. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_10_02.models.SourceVault @@ -2120,7 +2120,7 @@ def __init__(self, *, source_vault: "_models.SourceVault", key_url: str, **kwarg class KeyVaultAndSecretReference(_serialization.Model): """Key Vault Secret Url and vault id of the encryption key. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar source_vault: Resource id of the KeyVault containing the key or secret. Required. :vartype source_vault: ~azure.mgmt.compute.v2023_10_02.models.SourceVault @@ -2412,7 +2412,7 @@ def __init__(self, *, target_tier: Optional[str] = None, **kwargs: Any) -> None: class PurchasePlan(_serialization.Model): """Used for establishing the purchase context of any 3rd Party artifact through MarketPlace. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar name: The plan ID. Required. :vartype name: str @@ -2462,7 +2462,7 @@ def __init__( class ResourceUriList(_serialization.Model): """The List resources which are encrypted with the disk encryption set. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of IDs or Owner IDs of resources which are encrypted with the disk encryption set. Required. @@ -2569,7 +2569,7 @@ class Snapshot(Resource): # pylint: disable=too-many-instance-attributes Variables are only populated by the server, and will be ignored when sending a request. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar id: Resource Id. :vartype id: str @@ -2836,7 +2836,7 @@ def __init__( # pylint: disable=too-many-locals class SnapshotList(_serialization.Model): """The List Snapshots operation response. - All required parameters must be populated in order to send to Azure. + All required parameters must be populated in order to send to server. :ivar value: A list of snapshots. Required. :vartype value: list[~azure.mgmt.compute.v2023_10_02.models.Snapshot] diff --git a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/operations/_operations.py b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/operations/_operations.py index cd388e51e7de..8ef5d7de553d 100644 --- a/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/operations/_operations.py +++ b/sdk/compute/azure-mgmt-compute/azure/mgmt/compute/v2023_10_02/operations/_operations.py @@ -1,4 +1,4 @@ -# pylint: disable=too-many-lines +# pylint: disable=too-many-lines,too-many-statements # coding=utf-8 # -------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. @@ -163,7 +163,7 @@ def build_disks_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_disks_list_by_resource_group_request( +def build_disks_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -276,7 +276,7 @@ def build_disks_revoke_access_request( return HttpRequest(method="POST", url=_url, params=_params, **kwargs) -def build_disk_accesses_create_or_update_request( +def build_disk_accesses_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -406,7 +406,7 @@ def build_disk_accesses_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_by_resource_group_request( +def build_disk_accesses_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -460,7 +460,7 @@ def build_disk_accesses_list_request(subscription_id: str, **kwargs: Any) -> Htt return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_private_link_resources_request( +def build_disk_accesses_get_private_link_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -491,7 +491,7 @@ def build_disk_accesses_get_private_link_resources_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_update_a_private_endpoint_connection_request( +def build_disk_accesses_update_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -532,7 +532,7 @@ def build_disk_accesses_update_a_private_endpoint_connection_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_get_a_private_endpoint_connection_request( +def build_disk_accesses_get_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -570,7 +570,7 @@ def build_disk_accesses_get_a_private_endpoint_connection_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_delete_a_private_endpoint_connection_request( +def build_disk_accesses_delete_a_private_endpoint_connection_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, @@ -608,7 +608,7 @@ def build_disk_accesses_delete_a_private_endpoint_connection_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_accesses_list_private_endpoint_connections_request( +def build_disk_accesses_list_private_endpoint_connections_request( # pylint: disable=name-too-long resource_group_name: str, disk_access_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -639,7 +639,7 @@ def build_disk_accesses_list_private_endpoint_connections_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_create_or_update_request( +def build_disk_encryption_sets_create_or_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -673,7 +673,7 @@ def build_disk_encryption_sets_create_or_update_request( return HttpRequest(method="PUT", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_update_request( +def build_disk_encryption_sets_update_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -738,7 +738,7 @@ def build_disk_encryption_sets_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_delete_request( +def build_disk_encryption_sets_delete_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -769,7 +769,7 @@ def build_disk_encryption_sets_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_by_resource_group_request( +def build_disk_encryption_sets_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -823,7 +823,7 @@ def build_disk_encryption_sets_list_request(subscription_id: str, **kwargs: Any) return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_encryption_sets_list_associated_resources_request( +def build_disk_encryption_sets_list_associated_resources_request( # pylint: disable=name-too-long resource_group_name: str, disk_encryption_set_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -894,7 +894,7 @@ def build_disk_restore_point_get_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_list_by_restore_point_request( +def build_disk_restore_point_list_by_restore_point_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -932,7 +932,7 @@ def build_disk_restore_point_list_by_restore_point_request( return HttpRequest(method="GET", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_grant_access_request( +def build_disk_restore_point_grant_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -975,7 +975,7 @@ def build_disk_restore_point_grant_access_request( return HttpRequest(method="POST", url=_url, params=_params, headers=_headers, **kwargs) -def build_disk_restore_point_revoke_access_request( +def build_disk_restore_point_revoke_access_request( # pylint: disable=name-too-long resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, @@ -1139,7 +1139,7 @@ def build_snapshots_delete_request( return HttpRequest(method="DELETE", url=_url, params=_params, **kwargs) -def build_snapshots_list_by_resource_group_request( +def build_snapshots_list_by_resource_group_request( # pylint: disable=name-too-long resource_group_name: str, subscription_id: str, **kwargs: Any ) -> HttpRequest: _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {}) @@ -1273,7 +1273,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1298,7 +1298,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk, "Disk") - request = build_disks_create_or_update_request( + _request = build_disks_create_or_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1306,16 +1306,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1335,10 +1334,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_create_or_update( self, @@ -1362,14 +1357,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1380,7 +1367,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1394,18 +1381,10 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1413,7 +1392,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.Disk, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Creates or updates a disk. @@ -1424,19 +1403,8 @@ def begin_create_or_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Put disk operation. Is either a Disk type - or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_10_02.models.Disk or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_10_02.models.Disk or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1467,7 +1435,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1477,20 +1445,18 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> _models.Disk: error_map = { 401: ClientAuthenticationError, @@ -1515,7 +1481,7 @@ def _update_initial( else: _json = self._serialize.body(disk, "DiskUpdate") - request = build_disks_update_request( + _request = build_disks_update_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -1523,16 +1489,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1552,10 +1517,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } - @overload def begin_update( self, @@ -1579,14 +1540,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1597,7 +1550,7 @@ def begin_update( self, resource_group_name: str, disk_name: str, - disk: IO, + disk: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -1611,18 +1564,10 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Required. - :type disk: IO + :type disk: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1630,7 +1575,7 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO], **kwargs: Any + self, resource_group_name: str, disk_name: str, disk: Union[_models.DiskUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Disk]: """Updates (patches) a disk. @@ -1641,19 +1586,8 @@ def begin_update( maximum name length is 80 characters. Required. :type disk_name: str :param disk: Disk object supplied in the body of the Patch disk operation. Is either a - DiskUpdate type or a IO type. Required. - :type disk: ~azure.mgmt.compute.v2023_10_02.models.DiskUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + DiskUpdate type or a IO[bytes] type. Required. + :type disk: ~azure.mgmt.compute.v2023_10_02.models.DiskUpdate or IO[bytes] :return: An instance of LROPoller that returns either Disk or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1684,7 +1618,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -1694,17 +1628,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Disk].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[_models.Disk]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _models.Disk: @@ -1716,7 +1648,6 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Disk or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.Disk :raises ~azure.core.exceptions.HttpResponseError: @@ -1735,21 +1666,20 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.Disk] = kwargs.pop("cls", None) - request = build_disks_get_request( + _request = build_disks_get_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1761,13 +1691,9 @@ def get(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> _model deserialized = self._deserialize("Disk", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -1786,21 +1712,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_delete_request( + _request = build_disks_delete_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1810,11 +1735,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -1826,14 +1747,6 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -1860,7 +1773,7 @@ def begin_delete(self, resource_group_name: str, disk_name: str, **kwargs: Any) def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -1869,17 +1782,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Disk"]: @@ -1887,7 +1796,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1909,16 +1817,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disks_list_by_resource_group_request( + _request = build_disks_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -1929,14 +1836,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -1946,11 +1853,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -1962,15 +1869,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: """Lists all the disks under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Disk or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.Disk] :raises ~azure.core.exceptions.HttpResponseError: @@ -1992,15 +1894,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Disk"]: def prepare_request(next_link=None): if not next_link: - request = build_disks_list_request( + _request = build_disks_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -2011,14 +1912,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskList", pipeline_response) @@ -2028,11 +1929,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2044,13 +1945,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks"} - def _grant_access_initial( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -2076,7 +1975,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disks_grant_access_request( + _request = build_disks_grant_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, @@ -2084,16 +1983,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2107,13 +2005,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -2139,14 +2033,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2157,7 +2043,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2172,18 +2058,10 @@ def begin_grant_access( :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2194,7 +2072,7 @@ def begin_grant_access( self, resource_group_name: str, disk_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a disk. @@ -2206,19 +2084,8 @@ def begin_grant_access( maximum name length is 80 characters. Required. :type disk_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -2249,7 +2116,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2261,17 +2128,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_name: str, **kwargs: Any @@ -2290,21 +2155,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disks_revoke_access_request( + _request = build_disks_revoke_access_request( resource_group_name=resource_group_name, disk_name=disk_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2314,11 +2178,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2330,14 +2190,6 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -2364,7 +2216,7 @@ def begin_revoke_access(self, resource_group_name: str, disk_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -2375,17 +2227,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class DiskAccessesOperations: @@ -2409,7 +2257,11 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> _models.DiskAccess: error_map = { 401: ClientAuthenticationError, @@ -2434,7 +2286,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_access, "DiskAccess") - request = build_disk_accesses_create_or_update_request( + _request = build_disk_accesses_create_or_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2442,16 +2294,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2471,10 +2322,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_create_or_update( self, @@ -2499,14 +2346,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2517,7 +2356,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2532,18 +2371,10 @@ def begin_create_or_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2551,7 +2382,11 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, disk_access_name: str, disk_access: Union[_models.DiskAccess, IO], **kwargs: Any + self, + resource_group_name: str, + disk_access_name: str, + disk_access: Union[_models.DiskAccess, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Creates or updates a disk access resource. @@ -2562,19 +2397,8 @@ def begin_create_or_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Put disk access operation. - Is either a DiskAccess type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccess type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2605,7 +2429,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2615,23 +2439,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskAccess: error_map = { @@ -2657,7 +2479,7 @@ def _update_initial( else: _json = self._serialize.body(disk_access, "DiskAccessUpdate") - request = build_disk_accesses_update_request( + _request = build_disk_accesses_update_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, @@ -2665,16 +2487,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2694,10 +2515,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } - @overload def begin_update( self, @@ -2722,14 +2539,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2740,7 +2549,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: IO, + disk_access: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -2755,18 +2564,10 @@ def begin_update( :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. Required. - :type disk_access: IO + :type disk_access: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2777,7 +2578,7 @@ def begin_update( self, resource_group_name: str, disk_access_name: str, - disk_access: Union[_models.DiskAccessUpdate, IO], + disk_access: Union[_models.DiskAccessUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskAccess]: """Updates (patches) a disk access resource. @@ -2789,19 +2590,8 @@ def begin_update( are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str :param disk_access: disk access object supplied in the body of the Patch disk access operation. - Is either a DiskAccessUpdate type or a IO type. Required. - :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccessUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Is either a DiskAccessUpdate type or a IO[bytes] type. Required. + :type disk_access: ~azure.mgmt.compute.v2023_10_02.models.DiskAccessUpdate or IO[bytes] :return: An instance of LROPoller that returns either DiskAccess or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -2832,7 +2622,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -2842,17 +2632,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskAccess].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[_models.DiskAccess]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> _models.DiskAccess: @@ -2864,7 +2652,6 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskAccess or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskAccess :raises ~azure.core.exceptions.HttpResponseError: @@ -2883,21 +2670,20 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskAccess] = kwargs.pop("cls", None) - request = build_disk_accesses_get_request( + _request = build_disk_accesses_get_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2909,13 +2695,9 @@ def get(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> deserialized = self._deserialize("DiskAccess", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -2934,21 +2716,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_request( + _request = build_disk_accesses_delete_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -2958,11 +2739,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs: Any) -> LROPoller[None]: @@ -2974,14 +2751,6 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3008,7 +2777,7 @@ def begin_delete(self, resource_group_name: str, disk_access_name: str, **kwargs def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3017,17 +2786,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskAccess"]: @@ -3035,7 +2800,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3057,16 +2821,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_by_resource_group_request( + _request = build_disk_accesses_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3077,14 +2840,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3094,11 +2857,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3110,15 +2873,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: """Lists all the disk access resources under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskAccess or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskAccess] :raises ~azure.core.exceptions.HttpResponseError: @@ -3140,15 +2898,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskAccess"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_request( + _request = build_disk_accesses_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3159,14 +2916,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskAccessList", pipeline_response) @@ -3176,11 +2933,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3192,8 +2949,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskAccesses"} - @distributed_trace def get_private_link_resources( self, resource_group_name: str, disk_access_name: str, **kwargs: Any @@ -3206,7 +2961,6 @@ def get_private_link_resources( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateLinkResourceListResult or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.PrivateLinkResourceListResult :raises ~azure.core.exceptions.HttpResponseError: @@ -3225,21 +2979,20 @@ def get_private_link_resources( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.PrivateLinkResourceListResult] = kwargs.pop("cls", None) - request = build_disk_accesses_get_private_link_resources_request( + _request = build_disk_accesses_get_private_link_resources_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_private_link_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3251,20 +3004,16 @@ def get_private_link_resources( deserialized = self._deserialize("PrivateLinkResourceListResult", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_private_link_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateLinkResources" - } + return deserialized # type: ignore - def _update_a_private_endpoint_connection_initial( + def _update_a_private_endpoint_connection_initial( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> _models.PrivateEndpointConnection: error_map = { @@ -3290,7 +3039,7 @@ def _update_a_private_endpoint_connection_initial( else: _json = self._serialize.body(private_endpoint_connection, "PrivateEndpointConnection") - request = build_disk_accesses_update_a_private_endpoint_connection_request( + _request = build_disk_accesses_update_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, @@ -3299,16 +3048,15 @@ def _update_a_private_endpoint_connection_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3328,12 +3076,8 @@ def _update_a_private_endpoint_connection_initial( return deserialized # type: ignore - _update_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } - @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, @@ -3361,14 +3105,6 @@ def begin_update_a_private_endpoint_connection( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3377,12 +3113,12 @@ def begin_update_a_private_endpoint_connection( """ @overload - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: IO, + private_endpoint_connection: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3400,18 +3136,10 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Required. - :type private_endpoint_connection: IO + :type private_endpoint_connection: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3420,12 +3148,12 @@ def begin_update_a_private_endpoint_connection( """ @distributed_trace - def begin_update_a_private_endpoint_connection( + def begin_update_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, - private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO], + private_endpoint_connection: Union[_models.PrivateEndpointConnection, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.PrivateEndpointConnection]: """Approve or reject a private endpoint connection under disk access resource, this can't be used @@ -3441,20 +3169,9 @@ def begin_update_a_private_endpoint_connection( :type private_endpoint_connection_name: str :param private_endpoint_connection: private endpoint connection object supplied in the body of the Put private endpoint connection operation. Is either a PrivateEndpointConnection type or a - IO type. Required. + IO[bytes] type. Required. :type private_endpoint_connection: - ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection or IO[bytes] :return: An instance of LROPoller that returns either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3488,7 +3205,7 @@ def begin_update_a_private_endpoint_connection( def get_long_running_output(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -3498,17 +3215,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.PrivateEndpointConnection].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[_models.PrivateEndpointConnection]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get_a_private_endpoint_connection( @@ -3524,7 +3239,6 @@ def get_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: PrivateEndpointConnection or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.PrivateEndpointConnection :raises ~azure.core.exceptions.HttpResponseError: @@ -3543,22 +3257,21 @@ def get_a_private_endpoint_connection( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.PrivateEndpointConnection] = kwargs.pop("cls", None) - request = build_disk_accesses_get_a_private_endpoint_connection_request( + _request = build_disk_accesses_get_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get_a_private_endpoint_connection.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3570,15 +3283,11 @@ def get_a_private_endpoint_connection( deserialized = self._deserialize("PrivateEndpointConnection", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return deserialized # type: ignore - def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements + def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsistent-return-statements,name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> None: error_map = { @@ -3595,22 +3304,21 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_accesses_delete_a_private_endpoint_connection_request( + _request = build_disk_accesses_delete_a_private_endpoint_connection_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, private_endpoint_connection_name=private_endpoint_connection_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_a_private_endpoint_connection_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3620,14 +3328,10 @@ def _delete_a_private_endpoint_connection_initial( # pylint: disable=inconsiste raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_a_private_endpoint_connection_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace - def begin_delete_a_private_endpoint_connection( + def begin_delete_a_private_endpoint_connection( # pylint: disable=name-too-long self, resource_group_name: str, disk_access_name: str, private_endpoint_connection_name: str, **kwargs: Any ) -> LROPoller[None]: """Deletes a private endpoint connection under a disk access resource. @@ -3640,14 +3344,6 @@ def begin_delete_a_private_endpoint_connection( :type disk_access_name: str :param private_endpoint_connection_name: The name of the private endpoint connection. Required. :type private_endpoint_connection_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -3675,7 +3371,7 @@ def begin_delete_a_private_endpoint_connection( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -3684,17 +3380,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete_a_private_endpoint_connection.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections/{privateEndpointConnectionName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_private_endpoint_connections( @@ -3708,7 +3400,6 @@ def list_private_endpoint_connections( can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_access_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either PrivateEndpointConnection or the result of cls(response) :rtype: @@ -3732,17 +3423,16 @@ def list_private_endpoint_connections( def prepare_request(next_link=None): if not next_link: - request = build_disk_accesses_list_private_endpoint_connections_request( + _request = build_disk_accesses_list_private_endpoint_connections_request( resource_group_name=resource_group_name, disk_access_name=disk_access_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_private_endpoint_connections.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -3753,14 +3443,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("PrivateEndpointConnectionListResult", pipeline_response) @@ -3770,11 +3460,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3786,10 +3476,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_private_endpoint_connections.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskAccesses/{diskAccessName}/privateEndpointConnections" - } - class DiskEncryptionSetsOperations: """ @@ -3815,7 +3501,7 @@ def _create_or_update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -3841,7 +3527,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSet") - request = build_disk_encryption_sets_create_or_update_request( + _request = build_disk_encryption_sets_create_or_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -3849,16 +3535,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -3878,10 +3563,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_create_or_update( self, @@ -3906,14 +3587,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -3925,7 +3598,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -3940,18 +3613,10 @@ def begin_create_or_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -3963,7 +3628,7 @@ def begin_create_or_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSet, IO], + disk_encryption_set: Union[_models.DiskEncryptionSet, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Creates or updates a disk encryption set. @@ -3975,19 +3640,9 @@ def begin_create_or_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Put disk - encryption set operation. Is either a DiskEncryptionSet type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSet type or a IO[bytes] type. Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -4019,7 +3674,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4029,23 +3684,21 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> _models.DiskEncryptionSet: error_map = { @@ -4071,7 +3724,7 @@ def _update_initial( else: _json = self._serialize.body(disk_encryption_set, "DiskEncryptionSetUpdate") - request = build_disk_encryption_sets_update_request( + _request = build_disk_encryption_sets_update_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, @@ -4079,16 +3732,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4108,10 +3760,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } - @overload def begin_update( self, @@ -4136,14 +3784,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -4155,7 +3795,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: IO, + disk_encryption_set: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -4170,18 +3810,10 @@ def begin_update( :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk encryption set operation. Required. - :type disk_encryption_set: IO + :type disk_encryption_set: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -4193,7 +3825,7 @@ def begin_update( self, resource_group_name: str, disk_encryption_set_name: str, - disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO], + disk_encryption_set: Union[_models.DiskEncryptionSetUpdate, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.DiskEncryptionSet]: """Updates (patches) a disk encryption set. @@ -4205,19 +3837,10 @@ def begin_update( name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str :param disk_encryption_set: disk encryption set object supplied in the body of the Patch disk - encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO type. Required. - :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSetUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + encryption set operation. Is either a DiskEncryptionSetUpdate type or a IO[bytes] type. + Required. + :type disk_encryption_set: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSetUpdate or + IO[bytes] :return: An instance of LROPoller that returns either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] @@ -4249,7 +3872,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -4259,17 +3882,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.DiskEncryptionSet].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[_models.DiskEncryptionSet]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> _models.DiskEncryptionSet: @@ -4281,7 +3902,6 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskEncryptionSet or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet :raises ~azure.core.exceptions.HttpResponseError: @@ -4300,21 +3920,20 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskEncryptionSet] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_get_request( + _request = build_disk_encryption_sets_get_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4326,13 +3945,9 @@ def get(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: deserialized = self._deserialize("DiskEncryptionSet", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4351,21 +3966,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_encryption_sets_delete_request( + _request = build_disk_encryption_sets_delete_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4375,11 +3989,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any) -> LROPoller[None]: @@ -4391,14 +4001,6 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -4425,7 +4027,7 @@ def begin_delete(self, resource_group_name: str, disk_encryption_set_name: str, def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -4434,17 +4036,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: @@ -4452,7 +4050,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4474,16 +4071,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_by_resource_group_request( + _request = build_disk_encryption_sets_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4494,14 +4090,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4511,11 +4107,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4527,15 +4123,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: """Lists all the disk encryption sets under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskEncryptionSet or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskEncryptionSet] :raises ~azure.core.exceptions.HttpResponseError: @@ -4557,15 +4148,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.DiskEncryptionSet"]: def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_request( + _request = build_disk_encryption_sets_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4576,14 +4166,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskEncryptionSetList", pipeline_response) @@ -4593,11 +4183,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4609,8 +4199,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/diskEncryptionSets"} - @distributed_trace def list_associated_resources( self, resource_group_name: str, disk_encryption_set_name: str, **kwargs: Any @@ -4623,7 +4211,6 @@ def list_associated_resources( name can't be changed after the disk encryption set is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name length is 80 characters. Required. :type disk_encryption_set_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either str or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[str] :raises ~azure.core.exceptions.HttpResponseError: @@ -4645,17 +4232,16 @@ def list_associated_resources( def prepare_request(next_link=None): if not next_link: - request = build_disk_encryption_sets_list_associated_resources_request( + _request = build_disk_encryption_sets_list_associated_resources_request( resource_group_name=resource_group_name, disk_encryption_set_name=disk_encryption_set_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_associated_resources.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4666,14 +4252,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("ResourceUriList", pipeline_response) @@ -4683,11 +4269,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4699,10 +4285,6 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_associated_resources.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/diskEncryptionSets/{diskEncryptionSetName}/associatedResources" - } - class DiskRestorePointOperations: """ @@ -4745,7 +4327,6 @@ def get( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: DiskRestorePoint or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.DiskRestorePoint :raises ~azure.core.exceptions.HttpResponseError: @@ -4764,23 +4345,22 @@ def get( api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.DiskRestorePoint] = kwargs.pop("cls", None) - request = build_disk_restore_point_get_request( + _request = build_disk_restore_point_get_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4792,13 +4372,9 @@ def get( deserialized = self._deserialize("DiskRestorePoint", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}" - } + return deserialized # type: ignore @distributed_trace def list_by_restore_point( @@ -4814,7 +4390,6 @@ def list_by_restore_point( :param vm_restore_point_name: The name of the vm restore point that the disk disk restore point belongs. Required. :type vm_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either DiskRestorePoint or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.DiskRestorePoint] :raises ~azure.core.exceptions.HttpResponseError: @@ -4836,18 +4411,17 @@ def list_by_restore_point( def prepare_request(next_link=None): if not next_link: - request = build_disk_restore_point_list_by_restore_point_request( + _request = build_disk_restore_point_list_by_restore_point_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_restore_point.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -4858,14 +4432,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("DiskRestorePointList", pipeline_response) @@ -4875,11 +4449,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4891,17 +4465,13 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_restore_point.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints" - } - def _grant_access_initial( self, resource_group_name: str, restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -4927,7 +4497,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_disk_restore_point_grant_access_request( + _request = build_disk_restore_point_grant_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, @@ -4937,16 +4507,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -4960,13 +4529,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -4998,14 +4563,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5018,7 +4575,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5037,18 +4594,10 @@ def begin_grant_access( :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5061,7 +4610,7 @@ def begin_grant_access( restore_point_collection_name: str, vm_restore_point_name: str, disk_restore_point_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a diskRestorePoint. @@ -5077,19 +4626,8 @@ def begin_grant_access( :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str :param grant_access_data: Access data object supplied in the body of the get disk access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -5122,7 +4660,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5134,17 +4672,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, @@ -5168,23 +4704,22 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_disk_restore_point_revoke_access_request( + _request = build_disk_restore_point_revoke_access_request( resource_group_name=resource_group_name, restore_point_collection_name=restore_point_collection_name, vm_restore_point_name=vm_restore_point_name, disk_restore_point_name=disk_restore_point_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5194,11 +4729,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access( @@ -5221,14 +4752,6 @@ def begin_revoke_access( :type vm_restore_point_name: str :param disk_restore_point_name: The name of the disk restore point created. Required. :type disk_restore_point_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5257,7 +4780,7 @@ def begin_revoke_access( def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -5268,17 +4791,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/restorePointCollections/{restorePointCollectionName}/restorePoints/{vmRestorePointName}/diskRestorePoints/{diskRestorePointName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore class SnapshotsOperations: @@ -5302,7 +4821,7 @@ def __init__(self, *args, **kwargs): self._api_version = input_args.pop(0) if input_args else kwargs.pop("api_version") def _create_or_update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5327,7 +4846,7 @@ def _create_or_update_initial( else: _json = self._serialize.body(snapshot, "Snapshot") - request = build_snapshots_create_or_update_request( + _request = build_snapshots_create_or_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5335,16 +4854,15 @@ def _create_or_update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._create_or_update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5364,10 +4882,6 @@ def _create_or_update_initial( return deserialized # type: ignore - _create_or_update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_create_or_update( self, @@ -5391,14 +4905,6 @@ def begin_create_or_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5409,7 +4915,7 @@ def begin_create_or_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5423,18 +4929,10 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5442,7 +4940,7 @@ def begin_create_or_update( @distributed_trace def begin_create_or_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO], **kwargs: Any + self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.Snapshot, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Creates or updates a snapshot. @@ -5453,19 +4951,8 @@ def begin_create_or_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Put disk operation. Is either a - Snapshot type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.Snapshot or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + Snapshot type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.Snapshot or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5496,7 +4983,7 @@ def begin_create_or_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5506,20 +4993,22 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_create_or_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _update_initial( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> _models.Snapshot: error_map = { 401: ClientAuthenticationError, @@ -5544,7 +5033,7 @@ def _update_initial( else: _json = self._serialize.body(snapshot, "SnapshotUpdate") - request = build_snapshots_update_request( + _request = build_snapshots_update_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -5552,16 +5041,15 @@ def _update_initial( content_type=content_type, json=_json, content=_content, - template_url=self._update_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5581,10 +5069,6 @@ def _update_initial( return deserialized # type: ignore - _update_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } - @overload def begin_update( self, @@ -5609,14 +5093,6 @@ def begin_update( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5627,7 +5103,7 @@ def begin_update( self, resource_group_name: str, snapshot_name: str, - snapshot: IO, + snapshot: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -5642,18 +5118,10 @@ def begin_update( :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Required. - :type snapshot: IO + :type snapshot: IO[bytes] :keyword content_type: Body Parameter content-type. Content type parameter for binary body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5661,7 +5129,11 @@ def begin_update( @distributed_trace def begin_update( - self, resource_group_name: str, snapshot_name: str, snapshot: Union[_models.SnapshotUpdate, IO], **kwargs: Any + self, + resource_group_name: str, + snapshot_name: str, + snapshot: Union[_models.SnapshotUpdate, IO[bytes]], + **kwargs: Any ) -> LROPoller[_models.Snapshot]: """Updates (patches) a snapshot. @@ -5672,19 +5144,8 @@ def begin_update( The max name length is 80 characters. Required. :type snapshot_name: str :param snapshot: Snapshot object supplied in the body of the Patch snapshot operation. Is - either a SnapshotUpdate type or a IO type. Required. - :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.SnapshotUpdate or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + either a SnapshotUpdate type or a IO[bytes] type. Required. + :type snapshot: ~azure.mgmt.compute.v2023_10_02.models.SnapshotUpdate or IO[bytes] :return: An instance of LROPoller that returns either Snapshot or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5715,7 +5176,7 @@ def begin_update( def get_long_running_output(pipeline_response): deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -5725,17 +5186,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.Snapshot].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_update.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[_models.Snapshot]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) @distributed_trace def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _models.Snapshot: @@ -5747,7 +5206,6 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: Snapshot or the result of cls(response) :rtype: ~azure.mgmt.compute.v2023_10_02.models.Snapshot :raises ~azure.core.exceptions.HttpResponseError: @@ -5766,21 +5224,20 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[_models.Snapshot] = kwargs.pop("cls", None) - request = build_snapshots_get_request( + _request = build_snapshots_get_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.get.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5792,13 +5249,9 @@ def get(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> _m deserialized = self._deserialize("Snapshot", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - get.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return deserialized # type: ignore def _delete_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -5817,21 +5270,20 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_delete_request( + _request = build_snapshots_delete_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._delete_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5841,11 +5293,7 @@ def _delete_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _delete_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -5857,14 +5305,6 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -5891,7 +5331,7 @@ def begin_delete(self, resource_group_name: str, snapshot_name: str, **kwargs: A def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast(PollingMethod, ARMPolling(lro_delay, **kwargs)) @@ -5900,17 +5340,13 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_delete.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore @distributed_trace def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Iterable["_models.Snapshot"]: @@ -5918,7 +5354,6 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite :param resource_group_name: The name of the resource group. Required. :type resource_group_name: str - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -5940,16 +5375,15 @@ def list_by_resource_group(self, resource_group_name: str, **kwargs: Any) -> Ite def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_by_resource_group_request( + _request = build_snapshots_list_by_resource_group_request( resource_group_name=resource_group_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list_by_resource_group.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -5960,14 +5394,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -5977,11 +5411,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -5993,15 +5427,10 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list_by_resource_group.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots" - } - @distributed_trace def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: """Lists snapshots under a subscription. - :keyword callable cls: A custom type or function that will be passed the direct response :return: An iterator like instance of either Snapshot or the result of cls(response) :rtype: ~azure.core.paging.ItemPaged[~azure.mgmt.compute.v2023_10_02.models.Snapshot] :raises ~azure.core.exceptions.HttpResponseError: @@ -6023,15 +5452,14 @@ def list(self, **kwargs: Any) -> Iterable["_models.Snapshot"]: def prepare_request(next_link=None): if not next_link: - request = build_snapshots_list_request( + _request = build_snapshots_list_request( subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self.list.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) else: # make call to next link with the client's api-version @@ -6042,14 +5470,14 @@ def prepare_request(next_link=None): for key, value in urllib.parse.parse_qs(_parsed_next_link.query).items() } ) - _next_request_params["api-version"] = self._config.api_version - request = HttpRequest( + _next_request_params["api-version"] = self._api_version + _request = HttpRequest( "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) - request.method = "GET" - return request + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) + _request.method = "GET" + return _request def extract_data(pipeline_response): deserialized = self._deserialize("SnapshotList", pipeline_response) @@ -6059,11 +5487,11 @@ def extract_data(pipeline_response): return deserialized.next_link or None, iter(list_of_elem) def get_next(next_link=None): - request = prepare_request(next_link) + _request = prepare_request(next_link) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6075,13 +5503,11 @@ def get_next(next_link=None): return ItemPaged(get_next, extract_data) - list.metadata = {"url": "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/snapshots"} - def _grant_access_initial( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> Optional[_models.AccessUri]: error_map = { @@ -6107,7 +5533,7 @@ def _grant_access_initial( else: _json = self._serialize.body(grant_access_data, "GrantAccessData") - request = build_snapshots_grant_access_request( + _request = build_snapshots_grant_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, @@ -6115,16 +5541,15 @@ def _grant_access_initial( content_type=content_type, json=_json, content=_content, - template_url=self._grant_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6138,13 +5563,9 @@ def _grant_access_initial( deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) - - return deserialized + return cls(pipeline_response, deserialized, {}) # type: ignore - _grant_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return deserialized # type: ignore @overload def begin_grant_access( @@ -6170,14 +5591,6 @@ def begin_grant_access( :keyword content_type: Body Parameter content-type. Content type parameter for JSON body. Default value is "application/json". :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6188,7 +5601,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: IO, + grant_access_data: IO[bytes], *, content_type: str = "application/json", **kwargs: Any @@ -6203,18 +5616,10 @@ def begin_grant_access( :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access operation. Required. - :type grant_access_data: IO + :type grant_access_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 - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6225,7 +5630,7 @@ def begin_grant_access( self, resource_group_name: str, snapshot_name: str, - grant_access_data: Union[_models.GrantAccessData, IO], + grant_access_data: Union[_models.GrantAccessData, IO[bytes]], **kwargs: Any ) -> LROPoller[_models.AccessUri]: """Grants access to a snapshot. @@ -6237,19 +5642,8 @@ def begin_grant_access( The max name length is 80 characters. Required. :type snapshot_name: str :param grant_access_data: Access data object supplied in the body of the get snapshot access - operation. Is either a GrantAccessData type or a IO type. Required. - :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO - :keyword content_type: Body Parameter content-type. Known values are: 'application/json'. - Default value is None. - :paramtype content_type: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. + operation. Is either a GrantAccessData type or a IO[bytes] type. Required. + :type grant_access_data: ~azure.mgmt.compute.v2023_10_02.models.GrantAccessData or IO[bytes] :return: An instance of LROPoller that returns either AccessUri or the result of cls(response) :rtype: ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2023_10_02.models.AccessUri] :raises ~azure.core.exceptions.HttpResponseError: @@ -6280,7 +5674,7 @@ def begin_grant_access( def get_long_running_output(pipeline_response): deserialized = self._deserialize("AccessUri", pipeline_response) if cls: - return cls(pipeline_response, deserialized, {}) + return cls(pipeline_response, deserialized, {}) # type: ignore return deserialized if polling is True: @@ -6292,17 +5686,15 @@ def get_long_running_output(pipeline_response): else: polling_method = polling if cont_token: - return LROPoller.from_continuation_token( + return LROPoller[_models.AccessUri].from_continuation_token( polling_method=polling_method, continuation_token=cont_token, client=self._client, deserialization_callback=get_long_running_output, ) - return LROPoller(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_grant_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/beginGetAccess" - } + return LROPoller[_models.AccessUri]( + self._client, raw_result, get_long_running_output, polling_method # type: ignore + ) def _revoke_access_initial( # pylint: disable=inconsistent-return-statements self, resource_group_name: str, snapshot_name: str, **kwargs: Any @@ -6321,21 +5713,20 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-10-02")) cls: ClsType[None] = kwargs.pop("cls", None) - request = build_snapshots_revoke_access_request( + _request = build_snapshots_revoke_access_request( resource_group_name=resource_group_name, snapshot_name=snapshot_name, subscription_id=self._config.subscription_id, api_version=api_version, - template_url=self._revoke_access_initial.metadata["url"], headers=_headers, params=_params, ) - request = _convert_request(request) - request.url = self._client.format_url(request.url) + _request = _convert_request(_request) + _request.url = self._client.format_url(_request.url) _stream = False pipeline_response: PipelineResponse = self._client._pipeline.run( # pylint: disable=protected-access - request, stream=_stream, **kwargs + _request, stream=_stream, **kwargs ) response = pipeline_response.http_response @@ -6345,11 +5736,7 @@ def _revoke_access_initial( # pylint: disable=inconsistent-return-statements raise HttpResponseError(response=response, error_format=ARMErrorFormat) if cls: - return cls(pipeline_response, None, {}) - - _revoke_access_initial.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return cls(pipeline_response, None, {}) # type: ignore @distributed_trace def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kwargs: Any) -> LROPoller[None]: @@ -6361,14 +5748,6 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw after the snapshot is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The max name length is 80 characters. Required. :type snapshot_name: str - :keyword callable cls: A custom type or function that will be passed the direct response - :keyword str continuation_token: A continuation token to restart a poller from a saved state. - :keyword polling: By default, your polling method will be ARMPolling. Pass in False for this - operation to not poll, or pass in your own initialized polling object for a personal polling - strategy. - :paramtype polling: bool or ~azure.core.polling.PollingMethod - :keyword int polling_interval: Default waiting time between two polls for LRO operations if no - Retry-After header is present. :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: @@ -6395,7 +5774,7 @@ def begin_revoke_access(self, resource_group_name: str, snapshot_name: str, **kw def get_long_running_output(pipeline_response): # pylint: disable=inconsistent-return-statements if cls: - return cls(pipeline_response, None, {}) + return cls(pipeline_response, None, {}) # type: ignore if polling is True: polling_method: PollingMethod = cast( @@ -6406,14 +5785,10 @@ def get_long_running_output(pipeline_response): # pylint: disable=inconsistent- else: polling_method = polling if cont_token: - return LROPoller.from_continuation_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(self._client, raw_result, get_long_running_output, polling_method) # type: ignore - - begin_revoke_access.metadata = { - "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/snapshots/{snapshotName}/endGetAccess" - } + return LROPoller[None](self._client, raw_result, get_long_running_output, polling_method) # type: ignore diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_create.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_create.py index da677d3f8d1f..2c46d622d809 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_create.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_create.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_maximum_set_gen.py index 7d113082924d..9c93a5ee4849 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_minimum_set_gen.py index 17975cba90eb..63e343c834d4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_maximum_set_gen.py index 84d5f5b614ea..049d96427d0a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_minimum_set_gen.py index 879c2b133e46..b957d05284ca 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_maximum_set_gen.py index 39d21aeb35e1..c3a05f2642e7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_minimum_set_gen.py index 94a8b80f3d43..16022b4592ba 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_available_sizes_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_by_subscription.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_by_subscription.py index 36d86eeb1a12..04cc35a6caac 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_by_subscription.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_by_subscription.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_maximum_set_gen.py index e96641ef17c1..f40c4491c5d2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_minimum_set_gen.py index 563c5431b521..b06cb02ffc84 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_update_maximum_set_gen.py index d1991964972d..27968d9edcd2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/availability_set_examples/availability_set_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_create_or_update.py index 1ef52c93300c..c84c7192bcf5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_maximum_set_gen.py index bb989df070a8..df6f861b7df4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_minimum_set_gen.py index 30dac5f69878..cee6b95abb03 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_get.py index cd439dc7e2df..5549fc7014f5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_create_or_update.py index ea360c1f83e1..75b6af2677f8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_maximum_set_gen.py index 1e8dcfde25f0..9f7997bbce3e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_minimum_set_gen.py index 94833e416023..c9ee3d309298 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_get.py index c256470f9465..cdac1bc2c313 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_resource_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_resource_group.py index 47db7a19ddaa..fe5b7737cd3c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_resource_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_resource_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_subscription.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_subscription.py index 1053ff0d2a27..ff0c96df706d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_subscription.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_list_by_subscription.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_update_maximum_set_gen.py index 86e0c856c14a..b8bf5d7ec948 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_group_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_list_by_reservation_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_list_by_reservation_group.py index 273c1a1485c9..f99125b35619 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_list_by_reservation_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_list_by_reservation_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_update_maximum_set_gen.py index b643edf77b3a..1a2508785dad 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/capacity_reservation_examples/capacity_reservation_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_maximum_set_gen.py index 59587f5ce7ad..2f4cbefbc222 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_minimum_set_gen.py index 6c130459faf8..abfd7a8eb5ee 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/operations_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_maximum_set_gen.py index 6d6cd431d126..5b43a0b4c05c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_minimum_set_gen.py index e279d24baa6e..7b5641f0d09c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/usage_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_maximum_set_gen.py index 3ffe647749ac..e3ea80fbe8bf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_minimum_set_gen.py index 0c247b4bf679..7fdc792d2766 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/compute_rp_common_examples/virtual_machine_sizes_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_create_or_update.py index 146947372d49..20aa8caa2592 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_maximum_set_gen.py index 9ef00046b047..80212426327e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_minimum_set_gen.py index fc0d0f867b11..a061148267c7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_get.py index 48bf75d93d56..9b2073201f67 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update.py index e0fc047a45b1..f88dfa2c5015 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update_with_ultra_ssd.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update_with_ultra_ssd.py index 027e4f731e6c..3acbae509ced 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update_with_ultra_ssd.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_create_or_update_with_ultra_ssd.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_maximum_set_gen.py index a16ad673e783..f555763fb022 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_minimum_set_gen.py index b99937169482..69f6c93aacfd 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get.py index 626e2e791d18..f9122d931c6b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get_ultra_ssd_enabled_dedicated_host_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get_ultra_ssd_enabled_dedicated_host_group.py index b2950de5fb17..d69226df02e2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get_ultra_ssd_enabled_dedicated_host_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_get_ultra_ssd_enabled_dedicated_host_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_maximum_set_gen.py index 0f194837e1f7..2be55a0fe61a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_minimum_set_gen.py index ac2ddfe84946..1e562eecd36a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_resource_group_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_maximum_set_gen.py index 5d24a7ad32c4..2a8860778c78 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_minimum_set_gen.py index c7700def1078..31ab1c61e38e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_list_by_subscription_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_update_maximum_set_gen.py index f32034bfb31e..d9f774540467 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_group_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_available_sizes.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_available_sizes.py index f9fe66ad42b0..fa692c36597c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_available_sizes.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_available_sizes.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_maximum_set_gen.py index 49c6aa3881fb..30436d785c35 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_minimum_set_gen.py index 3da64b01dbc7..ff35bb828935 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_list_by_host_group_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_redeploy.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_redeploy.py index 49e9205ca623..9570336d53f9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_redeploy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_redeploy.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_restart.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_restart.py index 0213ceb73a1b..ca1f994466dc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_restart.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_restart.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_maximum_set_gen.py index dc20a681ee23..e6d080cc54d7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_resize.py b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_resize.py index cb389b2a9c84..311ff38fec20 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_resize.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/dedicated_host_examples/dedicated_host_update_resize.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_ablob_included.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_ablob_included.py index 13b348a3e1b3..9cf520a9e108 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_ablob_included.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_ablob_included.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_amanaged_disk_included.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_amanaged_disk_included.py index 6789c4e1a2b7..052ee7b01181 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_amanaged_disk_included.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_amanaged_disk_included.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_asnapshot_included.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_asnapshot_included.py index f4407c60f360..2d11724009a1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_asnapshot_included.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_data_disk_from_asnapshot_included.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob.py index 8a21783465cb..880a2f8eda02 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob_with_disk_encryption_set.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob_with_disk_encryption_set.py index 627116aebe54..fbfc43a207a8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob_with_disk_encryption_set.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_ablob_with_disk_encryption_set.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk.py index 86a35e934e23..e9ef6aae3d92 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk_with_disk_encryption_set.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk_with_disk_encryption_set.py index 3e8f073cc163..f7ca6087f8d0 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk_with_disk_encryption_set.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_amanaged_disk_with_disk_encryption_set.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot.py index 098d58005b81..35c937679e4a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot_with_disk_encryption_set.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot_with_disk_encryption_set.py index c381b8b64011..88f4201731f0 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot_with_disk_encryption_set.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_asnapshot_with_disk_encryption_set.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_avm.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_avm.py index b3f668966414..7beeb3162aff 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_avm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_create_from_avm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_get.py index 93e33b0afca2..394c6b5302fd 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_resource_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_resource_group.py index 839c792bf431..44bcd5a36875 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_resource_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_resource_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_subscription.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_subscription.py index e0d337af3e50..83a1a00163b2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_subscription.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_list_by_subscription.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_update.py index 5c8da89e3cdb..8f0f43007f13 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/image_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_maximum_set_gen.py index 03dbcefea54a..4abcb6e95bbb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_minimum_set_gen.py index 68e9d8c86dd4..ef05e3afd042 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/image_examples/images_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_request_rate_by_interval.py b/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_request_rate_by_interval.py index 7eed14693859..133557fe611c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_request_rate_by_interval.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_request_rate_by_interval.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_throttled_requests.py b/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_throttled_requests.py index ea6c08877034..8e0e282abcea 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_throttled_requests.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/log_analytic_examples/log_analytics_throttled_requests.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_create_or_update.py index d99907d50f24..fd9a1a0a21e5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_delete.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_delete.py index 24fb9d5a9c2d..19f1a4cdfdf9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_delete.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_delete.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_get.py index f7f724ef2dea..e75a82d6122f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_resource_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_resource_group.py index ce4791c64c49..c0e25621bd3e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_resource_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_resource_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_subscription.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_subscription.py index 957f74f06118..8f558c4d97e3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_subscription.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_list_by_subscription.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_patch.py b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_patch.py index 6c55f1f7af34..cf7363825ff7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_patch.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/proximity_placement_group_examples/proximity_placement_group_patch.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update.py index 468055e019e1..f17c61e2226e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update_for_cross_region_copy.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update_for_cross_region_copy.py index 3f83a3433604..7323797150b7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update_for_cross_region_copy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_create_or_update_for_cross_region_copy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_maximum_set_gen.py index 0c22aa81fc59..66b632a57bbf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_minimum_set_gen.py index 7acc1a2daba3..e78d149093da 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get.py index 7b0f023097fd..26b4cf57ce9a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get_with_contained_restore_points.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get_with_contained_restore_points.py index 4f394c1224cd..d1c76f094de2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get_with_contained_restore_points.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_get_with_contained_restore_points.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_resource_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_resource_group.py index d56baeff652f..e0317197b285 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_resource_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_resource_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_subscription.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_subscription.py index ab821b58a44f..a0bf7bfb30be 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_subscription.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_list_by_subscription.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_update_maximum_set_gen.py index 090c49fd9fe4..aff575f05f1e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_collection_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_copy_between_regions.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_copy_between_regions.py index bd010d2f3a3d..f0e6ca588759 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_copy_between_regions.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_copy_between_regions.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_create.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_create.py index e09d64fd3ae5..bce4629106ef 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_create.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_create.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_maximum_set_gen.py index dd82e858f1d4..b28f4ccaffa3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_minimum_set_gen.py index 0a230ef86628..31da7d976014 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get.py index b8317d4c3e0d..2bf325a7003f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get_with_instance_view.py b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get_with_instance_view.py index d373804473e9..7a40a6c9cc66 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get_with_instance_view.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/restore_point_examples/restore_point_get_with_instance_view.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_get.py index 0d63cee427e8..67e6fe6b625b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_list.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_list.py index 3d829181a7de..a801a99b841f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_list.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/run_command_list.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command.py index fd5b38122274..bb86ecda3fd9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_create_or_update.py index fab90ad80060..c09737898339 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_delete.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_delete.py index 2d5b914c0047..9b4f8d775fb8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_delete.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_delete.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_get.py index a29a68cf0e02..594a09d00961 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_list.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_list.py index e4cba720b151..787e5583c142 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_list.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_list.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_update.py index e5bf0f424773..83be285a3f89 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_run_command_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command.py index 40a075e6b624..ce42df35d9f6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_create_or_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_create_or_update.py index 920a0f8d5973..c69fd987553f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_create_or_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_create_or_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_delete.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_delete.py index b9632814ea5f..6f2b8cd7f59e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_delete.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_delete.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_get.py index 03738b426f5f..f947762b62bc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_list.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_list.py index 8f8b8b586c54..b4338a9cbb62 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_list.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_list.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_update.py index d09e5cc9b0b1..cdae2a9f9f3e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/run_command_examples/virtual_machine_scale_set_vm_run_command_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_create.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_create.py index 4451350349e3..b3b5aaa28e14 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_create.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_create.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_maximum_set_gen.py index 1de65f849dcc..47ae1a3469f7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_minimum_set_gen.py index e160d13eee10..a5ba532c5ed7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair.py index 05bee20ef25f..e7951ba18870 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_ed25519.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_ed25519.py index cf808cd3ca2c..56b121578ad2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_ed25519.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_ed25519.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_rsa.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_rsa.py index 7337ea1210a2..d7efb289d1a4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_rsa.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_generate_key_pair_encryption_with_rsa.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_get.py index cf532a26bbb6..8518b60289ba 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_maximum_set_gen.py index a70ee812d9b2..1edc3c491dff 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_minimum_set_gen.py index 1f87a6e1bbb4..e5cab1f07395 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_resource_group_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_maximum_set_gen.py index be71d8ec41a7..cbfd54afe33a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_minimum_set_gen.py index eeed53f50b06..bc83c09b638f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_list_by_subscription_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_update_maximum_set_gen.py index 1d86a0854dda..5314aa4d60b6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/ssh_public_key_examples/ssh_public_key_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_assess_patches.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_assess_patches.py index dbe84b2809b9..7d6f70899d79 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_assess_patches.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_assess_patches.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_maximum_set_gen.py index ff2b06a728e3..fadc2e811916 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_minimum_set_gen.py index 8aaf1f88da63..4794a645a1ad 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_attach_detach_data_disks_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_maximum_set_gen.py index 7e1ec852677f..5094fa6d3800 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_minimum_set_gen.py index 1668e357dff9..e296a870ddc5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_capture_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_maximum_set_gen.py index 2ee07cbfa3a2..1420421b65a2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_minimum_set_gen.py index d80d970aa60b..f15e0607a34e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_convert_to_managed_disks_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_custom_image_vm_from_an_unmanaged_generalized_os_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_custom_image_vm_from_an_unmanaged_generalized_os_image.py index 217cb5901dee..1ee850bc02f9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_custom_image_vm_from_an_unmanaged_generalized_os_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_custom_image_vm_from_an_unmanaged_generalized_os_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acommunity_gallery_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acommunity_gallery_image.py index 6aad7a198948..9dab3e193139 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acommunity_gallery_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acommunity_gallery_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acustom_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acustom_image.py index 7e24f25709b6..d9f57c325716 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acustom_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_acustom_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ageneralized_shared_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ageneralized_shared_image.py index 842d420a6430..5a715e811c39 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ageneralized_shared_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ageneralized_shared_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ashared_gallery_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ashared_gallery_image.py index c77e2bf4c823..a8b89a063d6f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ashared_gallery_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_ashared_gallery_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_aspecialized_shared_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_aspecialized_shared_image.py index d3ad6c66d459..1a0730c47006 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_aspecialized_shared_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_from_aspecialized_shared_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_an_availability_set.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_an_availability_set.py index 3df54fac6de8..c4ba43bc5595 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_an_availability_set.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_an_availability_set.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_avmss_with_customer_assigned_platform_fault_domain.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_avmss_with_customer_assigned_platform_fault_domain.py index b77f35b6a2db..ba21def08af6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_avmss_with_customer_assigned_platform_fault_domain.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_in_avmss_with_customer_assigned_platform_fault_domain.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_automatic_by_platform_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_automatic_by_platform_settings.py index 0b19f1a81230..c4ff71415b4c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_automatic_by_platform_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_automatic_by_platform_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_assessment_mode_of_image_default.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_assessment_mode_of_image_default.py index d175333bccbd..1c3b1dbd83b3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_assessment_mode_of_image_default.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_assessment_mode_of_image_default.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_mode_of_image_default.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_mode_of_image_default.py index 73572a7a0188..d73f037114d4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_mode_of_image_default.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_mode_of_image_default.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_modes_of_automatic_by_platform.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_modes_of_automatic_by_platform.py index f3297f3483c1..5cb3b7950b18 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_modes_of_automatic_by_platform.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_linux_vm_with_patch_setting_modes_of_automatic_by_platform.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_platform_image_vm_with_unmanaged_os_and_data_disks.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_platform_image_vm_with_unmanaged_os_and_data_disks.py index 0ec1d1eda27c..0906b0669d6a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_platform_image_vm_with_unmanaged_os_and_data_disks.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_platform_image_vm_with_unmanaged_os_and_data_disks.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_automatic_by_platform_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_automatic_by_platform_settings.py index 649da227ebcc..869eeb98d9e8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_automatic_by_platform_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_automatic_by_platform_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_assessment_mode_of_image_default.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_assessment_mode_of_image_default.py index dac32af78c8c..d42ce601a086 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_assessment_mode_of_image_default.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_assessment_mode_of_image_default.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_os.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_os.py index d19007c551fe..90075922fdc5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_os.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_os.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_platform_and_enable_hot_patching_true.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_platform_and_enable_hot_patching_true.py index 0ebb71f94f76..ceba87fd081f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_platform_and_enable_hot_patching_true.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_automatic_by_platform_and_enable_hot_patching_true.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_manual.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_manual.py index 5e3b1c1750ff..9522b06eb16b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_manual.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_mode_of_manual.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_modes_of_automatic_by_platform.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_modes_of_automatic_by_platform.py index 32caaae5c699..33d849709b18 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_modes_of_automatic_by_platform.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_windows_vm_with_patch_setting_modes_of_automatic_by_platform.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk.py index 9c2c6edbfa61..b2b6d31e0b30 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_cache_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_cache_disk.py index 83802c7bcc16..7de5c6196c89 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_cache_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_cache_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_resource_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_resource_disk.py index b3624d7fd4e1..33115d6f3f2b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_resource_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_adiff_os_disk_using_diff_disk_placement_as_resource_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_amarketplace_image_plan.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_amarketplace_image_plan.py index 47c01744f4bf..71f1aa07a73f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_amarketplace_image_plan.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_amarketplace_image_plan.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_application_profile.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_application_profile.py index 18f35136c9e1..3ef7421e081d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_application_profile.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_application_profile.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_boot_diagnostics.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_boot_diagnostics.py index 8f4419b3a602..348d0a65686d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_boot_diagnostics.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_boot_diagnostics.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_capacity_reservation.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_capacity_reservation.py index aaecfb894489..2ff2b4aeab4a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_capacity_reservation.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_capacity_reservation.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_controller_type.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_controller_type.py index db98b4d3abc9..51e848140ca6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_controller_type.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_controller_type.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_encryption_set_resource.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_encryption_set_resource.py index 5ed72211ccbe..cde781ba4ed6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_encryption_set_resource.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_disk_encryption_set_resource.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_empty_data_disks.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_empty_data_disks.py index fc542ecba4a7..27e82848e3f7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_empty_data_disks.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_empty_data_disks.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_at_host.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_at_host.py index 5d32d919ba88..6bf74cdbcffd 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_at_host.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_at_host.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_identity.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_identity.py index 14ba5a23ec76..b59beda95b78 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_identity.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_encryption_identity.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_extensions_time_budget.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_extensions_time_budget.py index 5c41e0187672..e85e665903f4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_extensions_time_budget.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_extensions_time_budget.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_hibernation_enabled.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_hibernation_enabled.py index 504ff5b113dd..c353c0af96f5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_hibernation_enabled.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_hibernation_enabled.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_managed_boot_diagnostics.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_managed_boot_diagnostics.py index 428a1edfabb1..85d9abaa8cbf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_managed_boot_diagnostics.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_managed_boot_diagnostics.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration.py index 13e65c85d77c..3f7c49162dd2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration_dns_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration_dns_settings.py index f9eff7550c4e..b620b5286ef4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration_dns_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_network_interface_configuration_dns_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_password_authentication.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_password_authentication.py index 4dda0d601b1b..0c99663ea7e4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_password_authentication.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_password_authentication.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_premium_storage.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_premium_storage.py index f05de4cbdf60..870fa7e3c7d7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_premium_storage.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_premium_storage.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_proxy_agent_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_proxy_agent_settings.py index 68e7be0e0025..d5323f91cddf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_proxy_agent_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_proxy_agent_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_scheduled_events_profile.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_scheduled_events_profile.py index 159686c95ab5..29d2855ac3e4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_scheduled_events_profile.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_scheduled_events_profile.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm.py index b6af402ab2f5..5de98571a14b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_customer_managed_keys.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_customer_managed_keys.py index f28ba2e8b1b9..596ad594f392 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_customer_managed_keys.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_customer_managed_keys.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_non_persisted_tpm.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_non_persisted_tpm.py index fd363eabb95f..c1cfd373bf73 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_non_persisted_tpm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_security_type_confidential_vm_with_non_persisted_tpm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_ssh_authentication.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_ssh_authentication.py index 4bb358b73855..465b1dc5286e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_ssh_authentication.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_ssh_authentication.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_uefi_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_uefi_settings.py index febacaec6e3f..cceeaaae6fda 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_uefi_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_uefi_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_user_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_user_data.py index bc1a1b52d98f..ce8ac24c5379 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_user_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_user_data.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_vm_size_properties.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_vm_size_properties.py index 9813a94963e8..797e3991e1e6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_vm_size_properties.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_create_with_vm_size_properties.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_maximum_set_gen.py index 721dc6c812aa..47c265b26ec7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_minimum_set_gen.py index 1e620cc616c9..7ef521dd3acb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_deallocate_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_delete_force.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_delete_force.py index 618720fde447..a6cc23a6c3c5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_delete_force.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_delete_force.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_maximum_set_gen.py index cac9f4ddd3db..4d0ee8aae868 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_minimum_set_gen.py index 66b0c54a322b..0e2af3983cbe 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_create_or_update_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_maximum_set_gen.py index cd51d490e330..236ec323d1eb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_minimum_set_gen.py index 740c0d1a2f42..0819dadd6231 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_maximum_set_gen.py index a3f64aa4cb6c..67f09f102841 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_minimum_set_gen.py index 824f1576ce33..6fb44b3779e3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_maximum_set_gen.py index b326fcab523a..cac95a0e3da2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_minimum_set_gen.py index 2f9c2bd0b16a..7ba82a7c697e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_update.py index fe6f6365bc64..e727bb68e64d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_extension_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_generalize.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_generalize.py index dcd8f97d0684..62f3300fc513 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_generalize.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_generalize.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get.py index 3c46808d217c..7e9d9581c1e9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_auto_placed_on_dedicated_host_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_auto_placed_on_dedicated_host_group.py index 22555ec51eb5..9d4a36a22391 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_auto_placed_on_dedicated_host_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_auto_placed_on_dedicated_host_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view.py index 9ffdd102c377..d6bed888b71f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view_auto_placed_on_dedicated_host_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view_auto_placed_on_dedicated_host_group.py index e33399bcb581..ebf030109d63 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view_auto_placed_on_dedicated_host_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_instance_view_auto_placed_on_dedicated_host_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_disk_controller_type.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_disk_controller_type.py index d55afe4db3bb..03871036f57d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_disk_controller_type.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_disk_controller_type.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_vm_size_properties.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_vm_size_properties.py index 90060096e92a..5d478f31406f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_vm_size_properties.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_get_with_vm_size_properties.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_install_patches.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_install_patches.py index b39264b96dcc..eae45e0fecbc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_install_patches.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_install_patches.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_maximum_set_gen.py index d8ff9a1ae41f..8be5ab310f0d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_minimum_set_gen.py index acd6a85cfd73..0ee01f730bc6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_all_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_available_vm_sizes.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_available_vm_sizes.py index 78ac1d1f7b47..4f35c22a1d78 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_available_vm_sizes.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_available_vm_sizes.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_by_subscription_by_location.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_by_subscription_by_location.py index de90476c92ae..7ae17f8df6bf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_by_subscription_by_location.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_by_subscription_by_location.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_maximum_set_gen.py index b3dd10f8b5bd..ef5235cc3cb1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_minimum_set_gen.py index b212945e9f8c..fa0a48991f0f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_maximum_set_gen.py index dba545a17ba7..09af6fde8f08 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_minimum_set_gen.py index 1e84b3db6c87..318ea83d7c4c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_perform_maintenance_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_maximum_set_gen.py index f8283c54a03d..f5447646f804 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_minimum_set_gen.py index cb7c3f961c34..3e8ff7f208e3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_power_off_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reapply.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reapply.py index f92a3eaf2976..d5dc2d4032f4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reapply.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reapply.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_maximum_set_gen.py index 4b9549bea1cb..ad458bd38c20 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_minimum_set_gen.py index 149d205dd2ff..97c0a56d8e2b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_redeploy_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage.py index 75a7114f1808..a0664d1a825b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage_non_ephemeral_vms.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage_non_ephemeral_vms.py index 72350d3f56c3..b5bd5caa225b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage_non_ephemeral_vms.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_reimage_non_ephemeral_vms.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_maximum_set_gen.py index 88a41f2f7b97..7e9c5542a0f9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_minimum_set_gen.py index 40fe5c1f9695..e4414516704d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_restart_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_retrieve_boot_diagnostics_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_retrieve_boot_diagnostics_data.py index a97c6f161333..4da8ac85012c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_retrieve_boot_diagnostics_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_retrieve_boot_diagnostics_data.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_simulate_eviction.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_simulate_eviction.py index 1acab07c5a91..81a66bbcdb3a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_simulate_eviction.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_simulate_eviction.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_maximum_set_gen.py index 3d69aba440df..bd2a449ea317 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_minimum_set_gen.py index 530121ab81d5..b0c8be7db5a4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_start_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_detach_data_disk_using_to_be_detached_property.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_detach_data_disk_using_to_be_detached_property.py index 3df329457e6e..f9fde38db43d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_detach_data_disk_using_to_be_detached_property.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_detach_data_disk_using_to_be_detached_property.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_force_detach_data_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_force_detach_data_disk.py index 83830f6ebc94..340e27ebc715 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_force_detach_data_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_examples/virtual_machine_update_force_detach_data_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_maximum_set_gen.py index 03c2b045732e..56249d79b89f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_minimum_set_gen.py index 3c87be51815d..96a68d1c0f0b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_maximum_set_gen.py index 6f455c71aac8..25a13eb257d4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_minimum_set_gen.py index f0cc7164f43e..88a4089180ac 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_types_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_maximum_set_gen.py index ad30f12edf0b..7353fe98e975 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_minimum_set_gen.py index bba83916ad07..4b8f31abcbd2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_extension_image_examples/virtual_machine_extension_image_list_versions_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_maximum_set_gen.py index 689911ff6267..b74696dfb830 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_minimum_set_gen.py index e2bcc6869116..41b37b4896af 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_maximum_set_gen.py index 23672297d8a8..ab7c37d61be5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_minimum_set_gen.py index 3768f76c2313..de8a3e5fdb1f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_maximum_set_gen.py index 8f51d8c61189..2d0cc148c861 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_minimum_set_gen.py index 9700d68b2ab6..63ea7279478d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_offers_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_maximum_set_gen.py index 93af7e954ff1..1f91351a70b9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_minimum_set_gen.py index 8ff608a9dc9e..d4bc6cbffa89 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_publishers_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_maximum_set_gen.py index b0fa3d0a2dc8..dc437c69958a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_minimum_set_gen.py index 57a0853ec15b..13ef94f2479d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_image_list_skus_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_maximum_set_gen.py index 8a8286a66363..408bb9942ed0 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_minimum_set_gen.py index 0b27bdf8d37b..66ea17a57e54 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_maximum_set_gen.py index 24771a7f97ba..61e0b37f689f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_minimum_set_gen.py index 6ddb71384a67..a87f665388e5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_by_edge_zone_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_maximum_set_gen.py index e761b87d924c..e81765fcf96b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_minimum_set_gen.py index 7fa97bba3c16..751b8e913b9b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_maximum_set_gen.py index 176198ae7c63..1867b390fce0 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_minimum_set_gen.py index c1105bb1e5bf..c625c29c1d55 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_offers_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_maximum_set_gen.py index dfc09ac6d07b..1fc45900100d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_minimum_set_gen.py index a4ed3f6fdb98..2177b10fb5c5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_publishers_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_maximum_set_gen.py index e13f57e4d3f8..cd298d157316 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_minimum_set_gen.py index 5c53d14c85f6..9c779ba1a28e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_image_examples/virtual_machine_images_edge_zone_list_skus_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_approve_rolling_upgrade.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_approve_rolling_upgrade.py index 930f4b499618..db1618b9f297 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_approve_rolling_upgrade.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_approve_rolling_upgrade.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_convert_to_single_placement_group_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_convert_to_single_placement_group_maximum_set_gen.py index 88b88b83c99e..25e263e0f78e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_convert_to_single_placement_group_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_convert_to_single_placement_group_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_a_with_diff_os_disk_using_diff_disk_placement.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_a_with_diff_os_disk_using_diff_disk_placement.py index 6a23d6471c85..15958c5cd00d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_a_with_diff_os_disk_using_diff_disk_placement.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_a_with_diff_os_disk_using_diff_disk_placement.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_custom_image_from_an_unmanaged_generalized_os_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_custom_image_from_an_unmanaged_generalized_os_image.py index 40c2b232597e..e01e6d55fe4d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_custom_image_from_an_unmanaged_generalized_os_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_custom_image_from_an_unmanaged_generalized_os_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_acustom_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_acustom_image.py index 38bf533060dd..5560ac055152 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_acustom_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_acustom_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_ageneralized_shared_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_ageneralized_shared_image.py index 5fd04021b5bf..e021a7a35c75 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_ageneralized_shared_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_ageneralized_shared_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_aspecialized_shared_image.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_aspecialized_shared_image.py index 654e4a75d26a..0c0eed96f8ab 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_aspecialized_shared_image.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_aspecialized_shared_image.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_disable_tcp_state_tracking_network_interface.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_disable_tcp_state_tracking_network_interface.py index bca0314cb60e..f7684770cd0e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_disable_tcp_state_tracking_network_interface.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_disable_tcp_state_tracking_network_interface.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_fpga_network_interface.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_fpga_network_interface.py index 2c6aeff3d026..c7c1deac4cbe 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_fpga_network_interface.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_fpga_network_interface.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_network_interface_with_dns_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_network_interface_with_dns_settings.py index dd2925868d79..3ac3e68d25f8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_network_interface_with_dns_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_from_with_network_interface_with_dns_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_platform_image_with_unmanaged_os_disks.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_platform_image_with_unmanaged_os_disks.py index cf93f9c06cfd..e279be57904a 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_platform_image_with_unmanaged_os_disks.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_platform_image_with_unmanaged_os_disks.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_amarketplace_image_plan.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_amarketplace_image_plan.py index db174b328ae1..eb16e9e226f4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_amarketplace_image_plan.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_amarketplace_image_plan.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_application_profile.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_application_profile.py index 555f30e7e0ce..41078bd4f73c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_application_profile.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_application_profile.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_automatic_repairs.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_automatic_repairs.py index e165c3f504fd..9d9150f69ede 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_automatic_repairs.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_automatic_repairs.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_application_gateway.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_application_gateway.py index f34b49fd3795..4260e0ffe0f1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_application_gateway.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_application_gateway.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_load_balancer.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_load_balancer.py index 07fea9488d4f..130fd8409de6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_load_balancer.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_azure_load_balancer.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_boot_diagnostics.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_boot_diagnostics.py index cf179c4b14f1..9b868d51f043 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_boot_diagnostics.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_boot_diagnostics.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_capacity_reservation.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_capacity_reservation.py index 7c79c2ef8ace..1b6d2fcf0d24 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_capacity_reservation.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_capacity_reservation.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_diff_os_disk.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_diff_os_disk.py index 8aabfa0ce2ed..0301b0ea3c20 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_diff_os_disk.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_diff_os_disk.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_controller_type.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_controller_type.py index 045ed3e0a844..fc6fd297ebb9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_controller_type.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_controller_type.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_encryption_set_resource.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_encryption_set_resource.py index 962ce2eee977..c1f5880b5ca6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_encryption_set_resource.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_disk_encryption_set_resource.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_empty_data_disks_on_each_vm.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_empty_data_disks_on_each_vm.py index 7850b4caf3e8..4d817fb9b446 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_empty_data_disks_on_each_vm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_empty_data_disks_on_each_vm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_encryption_at_host.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_encryption_at_host.py index 8c3a754017c1..89810ab2e854 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_encryption_at_host.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_encryption_at_host.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_suppress_failures_enabled.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_suppress_failures_enabled.py index 119cb5d398c2..ef1f1fd5f04b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_suppress_failures_enabled.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_suppress_failures_enabled.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_time_budget.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_time_budget.py index 5bee03936239..045421392fbc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_time_budget.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_extensions_time_budget.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_managed_boot_diagnostics.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_managed_boot_diagnostics.py index db4ebc188173..5eda8e760d04 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_managed_boot_diagnostics.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_managed_boot_diagnostics.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_os_image_scheduled_event_enabled.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_os_image_scheduled_event_enabled.py index 6085599cfb45..0bef581f2796 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_os_image_scheduled_event_enabled.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_os_image_scheduled_event_enabled.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_password_authentication.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_password_authentication.py index 675d36c3365e..9dff3ca9f217 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_password_authentication.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_password_authentication.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_premium_storage.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_premium_storage.py index d433d58a2c96..0d8c0005e760 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_premium_storage.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_premium_storage.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_priority_mix_policy.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_priority_mix_policy.py index 26eb891aa29c..9bd8d3c897c6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_priority_mix_policy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_priority_mix_policy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ @@ -37,13 +40,11 @@ def main(): "properties": { "orchestrationMode": "Flexible", "priorityMixPolicy": {"baseRegularPriorityCount": 4, "regularPriorityPercentageAboveBase": 50}, - "platformFaultDomainCount": 1, "singlePlacementGroup": False, "virtualMachineProfile": { "billingProfile": {"maxPrice": -1}, "evictionPolicy": "Deallocate", "networkProfile": { - "networkApiVersion": "2020-11-01", "networkInterfaceConfigurations": [ { "name": "{vmss-name}", @@ -62,7 +63,7 @@ def main(): "primary": True, }, } - ], + ] }, "osProfile": { "adminPassword": "{your-password}", diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_protected_settings_from_key_vault.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_protected_settings_from_key_vault.py index 70bae3c60cf6..031ed21666dd 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_protected_settings_from_key_vault.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_protected_settings_from_key_vault.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_proxy_agent_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_proxy_agent_settings.py index fdb33dee59d9..2252d4cde965 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_proxy_agent_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_proxy_agent_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_creation_policy.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_creation_policy.py index 90ed28b3d8c7..0ba81bb92761 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_creation_policy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_creation_policy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_deletion_policy.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_deletion_policy.py index b056273912f2..bfbd703bfc12 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_deletion_policy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_resilient_vm_deletion_policy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_scale_in_policy.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_scale_in_policy.py index c74cc858a71f..748493d94014 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_scale_in_policy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_scale_in_policy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_posture_reference.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_posture_reference.py index 332a7e6c48bd..eaffc7af41bf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_posture_reference.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_posture_reference.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm.py index 7317771c8927..f99f84db7f83 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm_with_non_persisted_tpm.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm_with_non_persisted_tpm.py index ed340a41154f..1565a2505228 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm_with_non_persisted_tpm.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_security_type_confidential_vm_with_non_persisted_tpm.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_service_artifact_reference.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_service_artifact_reference.py index 591b795333c6..2f3f1c1f68d9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_service_artifact_reference.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_service_artifact_reference.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_spot_restore_policy.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_spot_restore_policy.py index 654df0b52c6d..abc477c693e0 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_spot_restore_policy.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_spot_restore_policy.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_ssh_authentication.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_ssh_authentication.py index 86bfc848e692..63446b15b56f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_ssh_authentication.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_ssh_authentication.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_terminate_scheduled_event_enabled.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_terminate_scheduled_event_enabled.py index ee67cf97576f..55f4046e6226 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_terminate_scheduled_event_enabled.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_terminate_scheduled_event_enabled.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_uefi_settings.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_uefi_settings.py index ace84dc5d03d..589120c8a8fe 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_uefi_settings.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_uefi_settings.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_user_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_user_data.py index 007dfbca982f..86863b932646 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_user_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_user_data.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vm_size_properties.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vm_size_properties.py index 14e29f04ff8e..54d21472f9c8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vm_size_properties.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vm_size_properties.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vms_in_different_zones.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vms_in_different_zones.py index 3c64e98e3546..0bb860ffd9ea 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vms_in_different_zones.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_create_with_vms_in_different_zones.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_maximum_set_gen.py index 224ae58b5d1d..5b61f7ca577d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_minimum_set_gen.py index 9921ff90f8f5..008739738123 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_deallocate_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_delete_force.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_delete_force.py index 55b726867d2c..abce963d1b8b 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_delete_force.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_delete_force.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_create_or_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_create_or_update_maximum_set_gen.py index 52b439ebb299..44ad9c19d9f7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_create_or_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_create_or_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_maximum_set_gen.py index 323731fbc6af..4e81a7ebcd8f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_minimum_set_gen.py index ebf6e1d3911b..36447cd0b769 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_delete_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_maximum_set_gen.py index 238deb7ac5bb..3a6723bea496 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_minimum_set_gen.py index f8485242711d..b5292e4845c5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_get_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_maximum_set_gen.py index 0090df6962c6..4dd8a02e9da1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_minimum_set_gen.py index d84567333ff6..cc4bfc7a57e3 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_rolling_upgrade.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_rolling_upgrade.py index 347f56cd10db..d29311ea3147 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_rolling_upgrade.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_rolling_upgrade.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_update_maximum_set_gen.py index edbc3bf5329b..8f385122fcaf 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_extension_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_maximum_set_gen.py index 1b393a60833e..1f7046c16e38 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_minimum_set_gen.py index 3187fdb98b79..c5a93f94501d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_force_recovery_service_fabric_platform_update_domain_walk_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get.py index 09bb548bfd71..2013df83a84e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_auto_placed_on_dedicated_host_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_auto_placed_on_dedicated_host_group.py index 3b7367eac192..3c5239c4a1dc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_auto_placed_on_dedicated_host_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_auto_placed_on_dedicated_host_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_maximum_set_gen.py index fe46b1bbc4ea..20b1e9fab7fa 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_minimum_set_gen.py index 6daa25a5c4e4..60d004953691 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_instance_view_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_maximum_set_gen.py index 9c42853028eb..5e6cf299345c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_minimum_set_gen.py index 295d12271d99..cceede4bc3a8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_os_upgrade_history_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_disk_controller_type.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_disk_controller_type.py index 38d392336139..4a0a2b9880e9 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_disk_controller_type.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_disk_controller_type.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_user_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_user_data.py index 8f6100354e5a..7323d8f965b2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_user_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_get_with_user_data.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_maximum_set_gen.py index dcbba25e8817..bed2378f8abc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_minimum_set_gen.py index 4839faaebce4..8627f33d95fc 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_all_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_by_subscription_by_location.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_by_subscription_by_location.py index 54e8d67960bb..f3bb9e3cca7e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_by_subscription_by_location.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_by_subscription_by_location.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_maximum_set_gen.py index a5befe1fbb06..154ee278ed6c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_minimum_set_gen.py index 29b7693247af..6e91fa646975 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_maximum_set_gen.py index fa21132cb952..900eafda37b6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_minimum_set_gen.py index f6782c80c345..65567489f23d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_list_skus_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_maximum_set_gen.py index 06226a38a838..28dc9597ecb4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_minimum_set_gen.py index c2147168c1e2..9ca7932af358 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_perform_maintenance_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_maximum_set_gen.py index 389d81c4a66c..fbc225733551 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_minimum_set_gen.py index 36bb2c85a777..d113c22e60ce 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_power_off_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_maximum_set_gen.py index 35687bb1d671..74b663c1e986 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_minimum_set_gen.py index 390f40cd7ed7..7e4e32ecd1ac 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reapply_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_maximum_set_gen.py index cd196e56278d..3465d71e744c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_minimum_set_gen.py index b8fdcf3ca1f8..f14e41044779 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_redeploy_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_maximum_set_gen.py index 182a1d3ed872..2b28095907b6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_minimum_set_gen.py index 371f4ce52ed3..8e9f88bd07f5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_all_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_maximum_set_gen.py index a3030ba453aa..878ba279b883 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_minimum_set_gen.py index 1c4ddb9bcbf1..bfc1b4e62ad7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_reimage_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_maximum_set_gen.py index 09993c1709ab..5922ae3e6559 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_minimum_set_gen.py index 1345033af010..c493b599d0a1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_restart_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_maximum_set_gen.py index dd71451a5b4f..1c52ab86734f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_minimum_set_gen.py index 5cbb5471c27d..cc410b2f67e7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_cancel_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_maximum_set_gen.py index 733f0d19e3ed..cbd731138828 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_minimum_set_gen.py index 7d363c4a3736..2406ad96e017 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_get_latest_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_maximum_set_gen.py index f07874489ae8..65ace5c8cbf1 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_minimum_set_gen.py index 65aed44bb878..ab7f66b962da 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_rolling_upgrade_start_os_upgrade_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_maximum_set_gen.py index ad8aa68ff395..2bef76096d5c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_minimum_set_gen.py index 0ef5f052c99a..bbae40bf9467 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_set_orchestration_service_state_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_maximum_set_gen.py index 6ef24f5cf670..d09c78f3e59e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_minimum_set_gen.py index 074d0af0d705..483621b9606e 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_start_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_update_maximum_set_gen.py index c8f017ec2621..2add278d2284 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_approve_rolling_upgrade.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_approve_rolling_upgrade.py index 699528362395..acf374120c3d 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_approve_rolling_upgrade.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_approve_rolling_upgrade.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_maximum_set_gen.py index c2cd49983cff..9dc355ff8d89 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_minimum_set_gen.py index 0bad828c102e..0671a1b1b9c8 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_attach_detach_data_disks_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_maximum_set_gen.py index d0a88a3dc026..9674c93522d6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_minimum_set_gen.py index 7ded7f7940a5..4f4b526ce8aa 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_deallocate_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_delete_force.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_delete_force.py index 4d3d55915a9a..878fcd3d48a7 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_delete_force.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_delete_force.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_create.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_create.py index 86b5233f817b..998187e90754 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_create.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_create.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_delete.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_delete.py index 43183dffb6ba..8c175cf7983f 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_delete.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_delete.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_get.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_get.py index b8a86595445e..f0b47d82857c 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_get.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_get.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_list.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_list.py index 5249c77885f4..41502fca53fa 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_list.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_list.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_update.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_update.py index f88e27fe96b0..4ef2177d2fb5 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_update.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_extension_update.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_instance_view_auto_placed_on_dedicated_host_group.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_instance_view_auto_placed_on_dedicated_host_group.py index dbf710c88614..c71e8a5cffeb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_instance_view_auto_placed_on_dedicated_host_group.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_instance_view_auto_placed_on_dedicated_host_group.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_user_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_user_data.py index e4176c7e82d0..ea91409eb347 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_user_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_user_data.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_vm_size_properties.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_vm_size_properties.py index c609814482cf..67f75e882fdb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_vm_size_properties.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_get_with_vm_size_properties.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_maximum_set_gen.py index e89b526a64a2..97e3390ff8e4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_minimum_set_gen.py index 7f729131d51a..ce8dcf499491 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_list_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_maximum_set_gen.py index c2a32a84357d..1d68d3dcd086 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_minimum_set_gen.py index a564eb6efd1f..8cbaa60cf1eb 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_perform_maintenance_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_maximum_set_gen.py index a1498de8a676..f361c825b8b2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_minimum_set_gen.py index 224ac3a77c17..a9c3b3d5f157 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_power_off_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_maximum_set_gen.py index b07c7f6e51a0..c5f844e38406 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_minimum_set_gen.py index 7747ebc58af7..1ca34e9d8994 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_redeploy_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_maximum_set_gen.py index 2cf1a44b8b70..b8d5d98027ed 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_minimum_set_gen.py index d84631de7236..114b885cffa2 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_all_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_maximum_set_gen.py index d6083e35f2d3..458c260d7324 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_minimum_set_gen.py index fcea999de2ab..a12ab40bcfee 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_reimage_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_maximum_set_gen.py index 93a98a570b72..9640fcb65724 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_minimum_set_gen.py index 33b7a4317c58..40341cfe5217 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_restart_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_retrieve_boot_diagnostics_data.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_retrieve_boot_diagnostics_data.py index 457d32155cc1..ae6e2eefad08 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_retrieve_boot_diagnostics_data.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_retrieve_boot_diagnostics_data.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_simulate_eviction.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_simulate_eviction.py index 007a5cef7b93..4ce05cdb4de4 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_simulate_eviction.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_simulate_eviction.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_maximum_set_gen.py index 51115dee612d..1267227f79f6 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_maximum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_minimum_set_gen.py index 4ae4a9654fc9..2a98fc3f4981 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_start_minimum_set_gen.py @@ -7,6 +7,7 @@ # -------------------------------------------------------------------------- from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_maximum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_maximum_set_gen.py index a49356a92a10..973ead1ee797 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_maximum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_maximum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_minimum_set_gen.py b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_minimum_set_gen.py index 0a9e8714e327..6b752d135501 100644 --- a/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_minimum_set_gen.py +++ b/sdk/compute/azure-mgmt-compute/generated_samples/virtual_machine_scale_set_examples/virtual_machine_scale_set_vm_update_minimum_set_gen.py @@ -6,7 +6,10 @@ # Changes may cause incorrect behavior and will be lost if the code is regenerated. # -------------------------------------------------------------------------- +from typing import Any, IO, Union + from azure.identity import DefaultAzureCredential + from azure.mgmt.compute import ComputeManagementClient """ diff --git a/sdk/compute/azure-mgmt-compute/setup.py b/sdk/compute/azure-mgmt-compute/setup.py index f92bb8ab57d2..f259470da1a7 100644 --- a/sdk/compute/azure-mgmt-compute/setup.py +++ b/sdk/compute/azure-mgmt-compute/setup.py @@ -74,9 +74,9 @@ "pytyped": ["py.typed"], }, install_requires=[ - "isodate<1.0.0,>=0.6.1", - "azure-common~=1.1", - "azure-mgmt-core>=1.3.2,<2.0.0", + "isodate>=0.6.1", + "azure-common>=1.1", + "azure-mgmt-core>=1.3.2", ], python_requires=">=3.8", ) diff --git a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_disks.py b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_disks.py index 5134ba1f63ea..85429d9f8658 100644 --- a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_disks.py +++ b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_disks.py @@ -62,7 +62,6 @@ def setup_method(self, method): self.mgmt_client._PROFILE_TAG + " test" ) - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_disks_multi(self, resource_group): @@ -257,7 +256,6 @@ def test_compute_disk_encryption(self, resource_group): result = self.mgmt_client.disk_encryption_sets.begin_delete(resource_group.name, DISK_ENCRYPTION_SET_NAME) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_shot(self, resource_group): @@ -370,7 +368,6 @@ def test_compute_shot(self, resource_group): result = self.mgmt_client.snapshots.begin_delete(resource_group.name, SNAPSHOT_NAME) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_disks(self, resource_group): diff --git a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_galleries.py b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_galleries.py index 10445fa39734..dd8e5a283bb8 100644 --- a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_galleries.py +++ b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_galleries.py @@ -69,7 +69,6 @@ def delete_snapshot(self, group_name, snapshot_name): result = self.mgmt_client.snapshots.begin_delete(group_name, snapshot_name) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_galleries(self, resource_group): diff --git a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vm.py b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vm.py index 5593a9c2c845..030f7c723a69 100644 --- a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vm.py +++ b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vm.py @@ -80,7 +80,6 @@ def create_network_interface(self, group_name, location, nic_name, subnet): return nic_info.id - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vm(self, resource_group): @@ -389,7 +388,6 @@ def test_compute_vm_2(self, resource_group): result = self.mgmt_client.virtual_machines.begin_delete(resource_group.name, VIRTUAL_MACHINE_NAME) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vm_image(self, resource_group): @@ -413,7 +411,6 @@ def test_compute_vm_image(self, resource_group): # List Virtual Machine image skus (TODO: need swagger file) result = self.mgmt_client.virtual_machine_images.list_skus(AZURE_LOCATION, PUBLISHER_NAME, OFFER) - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vm_extension_image(self, resource_group): diff --git a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vmss.py b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vmss.py index 0d74e6a70f8f..e75a8b5754c9 100644 --- a/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vmss.py +++ b/sdk/compute/azure-mgmt-compute/tests/test_mgmt_compute_vmss.py @@ -393,7 +393,6 @@ def test_compute_vmss_extension(self, resource_group): result = self.mgmt_client.virtual_machine_scale_sets.begin_delete(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_vm(self, resource_group): @@ -550,7 +549,6 @@ def test_compute_vmss_vm(self, resource_group): result = self.mgmt_client.virtual_machine_scale_sets.begin_delete(resource_group.name, VIRTUAL_MACHINE_SCALE_SET_NAME) result = result.result() - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_vm_2(self, resource_group): @@ -986,7 +984,6 @@ def test_compute_vmss_base_2(self, resource_group): # can not test it, see: # https://docs.microsoft.com/en-us/azure/virtual-machine-scale-sets/virtual-machine-scale-sets-maintenance-notifications # """) - @pytest.mark.skipif(os.getenv('AZURE_TEST_RUN_LIVE') not in ('true', 'yes'), reason='only run live test') @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) @recorded_by_proxy def test_compute_vmss_perform_maintenance(self, resource_group):